diff --git a/src/caosadvancedtools/models/parser.py b/src/caosadvancedtools/models/parser.py index f0371cc922d7a747cb73114b098501dc31f7500a..62d444618c5fee94aabc18ef11eae841b6941900 100644 --- a/src/caosadvancedtools/models/parser.py +++ b/src/caosadvancedtools/models/parser.py @@ -301,7 +301,7 @@ class Parser(object): self.model[ent_name].add_property( name=n, importance=importance, - datatype=_get_listdatatype(e["datatype"])) + datatype=db.LIST(_get_listdatatype(e["datatype"]))) else: self.model[ent_name].add_property(name=n, importance=importance) diff --git a/unittests/test_parser.py b/unittests/test_parser.py index 64c98e9b531f20733a9a38143d9371e1efb58eac..cff71be85d6ca4f06efe09c5fbdcc32a59a524a8 100644 --- a/unittests/test_parser.py +++ b/unittests/test_parser.py @@ -224,7 +224,10 @@ B: c: datatype: INTEGER """ - parse_model_from_yaml(to_file(string_list)) + model = parse_model_from_yaml(to_file(string_list)) + print(model) + self.assertTrue(isinstance(model['A'], db.RecordType)) + self.assertEqual(model['A'].properties[0].datatype, db.LIST("B")) class ParserTest(unittest.TestCase):