From 3daf2ef2537e6ce8a64e9db23f87584f0c1d8085 Mon Sep 17 00:00:00 2001
From: Mikhail Sveshnikov <mike0sv@gmail.com>
Date: Wed, 15 Dec 2021 22:13:41 +0700
Subject: [PATCH] add tests (#136)

---
 tests/core/test_requirements.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/core/test_requirements.py b/tests/core/test_requirements.py
index fe58d535..254779d3 100644
--- a/tests/core/test_requirements.py
+++ b/tests/core/test_requirements.py
@@ -1,5 +1,8 @@
 import pytest
+from pydantic import parse_obj_as
 
+from mlem.contrib.sklearn import SklearnModel
+from mlem.core.objects import ModelMeta
 from mlem.core.requirements import (
     CustomRequirement,
     InstallableRequirement,
@@ -108,6 +111,17 @@ def test_resolve_unique_req():
     assert reqs.installable[0] == req
 
 
+def test_serialize_empty():
+    mt = SklearnModel(methods={}, model="")
+    obj = ModelMeta(model_type=mt)
+    payload = obj.dict()
+    obj2 = ModelMeta(model_type=mt)
+    obj2.requirements.__root__.append(InstallableRequirement(module="sklearn"))
+    assert obj.requirements.__root__ == []
+    new_obj = parse_obj_as(ModelMeta, payload)
+    assert new_obj == obj
+
+
 # Copyright 2019 Zyfra
 # Copyright 2021 Iterative
 #