diff --git a/unittests/test_json_schema_model_parser.py b/unittests/test_json_schema_model_parser.py index 7f47890f413dce5511cd498fe802e03a1af3be70..a136f9ba2ae9965978c7f1234acb16289a3ca305 100644 --- a/unittests/test_json_schema_model_parser.py +++ b/unittests/test_json_schema_model_parser.py @@ -164,7 +164,7 @@ def test_enum(): assert isinstance(model[name], db.Record) assert model[name].name == name assert len(model[name].parents) == 1 - assert model[name].has_parent(model["license"]) + assert model[name].has_parent(model["license"], retrieve=False) # Also allow enums with non-string types number_enums = ["1.1", "2.2", "3.3"] @@ -181,7 +181,7 @@ def test_enum(): assert isinstance(model[name], db.Record) assert model[name].name == name assert len(model[name].parents) == 1 - assert model[name].has_parent(model["number_enum"]) + assert model[name].has_parent(model["number_enum"], retrieve=False) @pytest.mark.xfail(reason="Don't allow integer enums until https://gitlab.indiscale.com/caosdb/src/caosdb-server/-/issues/224 has been fixed") @@ -207,7 +207,7 @@ def test_int_enum(): assert isinstance(model[name], db.Record) assert model[name].name == name assert len(model[name].parents) == 1 - assert model[name].has_parent(model["int_enum"]) + assert model[name].has_parent(model["int_enum"], retrieve=False) def test_references(): @@ -339,7 +339,7 @@ def test_list(): assert isinstance(model[name], db.Record) assert model[name].name == name assert len(model[name].parents) == 1 - assert model[name].has_parent(model["license"]) + assert model[name].has_parent(model["license"], retrieve=False) def test_name_property(): diff --git a/unittests/test_yaml_model_parser.py b/unittests/test_yaml_model_parser.py index 6cdea7922a8503be082e8947edecd7e8c849730b..17c382285a47b75b9408ad5e4793c2df35b18a82 100644 --- a/unittests/test_yaml_model_parser.py +++ b/unittests/test_yaml_model_parser.py @@ -32,14 +32,6 @@ def has_property(el, name): return False -def has_parent(el, name): - for p in el.get_parents(): - if p.name == name: - return True - - return False - - class TwiceTest(unittest.TestCase): def test_defined_once(self): string = """ @@ -148,16 +140,16 @@ RT4: RT5: """ model = parse_model_from_yaml(to_file(string)) - assert has_parent(model["RT1"], "RT2") + assert model["RT1"].has_parent("RT2", retrieve=False) assert (model["RT1"].get_parent( "RT2")._flags["inheritance"] == db.SUGGESTED) - assert has_parent(model["RT1"], "RT3") + assert model["RT1"].has_parent("RT3", retrieve=False) assert (model["RT1"].get_parent( "RT3")._flags["inheritance"] == db.RECOMMENDED) - assert has_parent(model["RT1"], "RT4") + assert model["RT1"].has_parent("RT4", retrieve=False) assert (model["RT1"].get_parent( "RT4")._flags["inheritance"] == db.OBLIGATORY) - assert has_parent(model["RT1"], "RT5") + assert model["RT1"].has_parent("RT5", retrieve=False) assert (model["RT1"].get_parent( "RT5")._flags["inheritance"] == db.OBLIGATORY) @@ -509,5 +501,5 @@ R3: assert isinstance(r3, db.RecordType) for par in ["R1", "R2"]: # Until removal, both do the same - assert has_parent(r3, par) + assert r3.has_parent(par, retrieve=False) assert r3.get_parent(par)._flags["inheritance"] == db.OBLIGATORY