From a754bd9f25e492bd40aabe59ebc361804c071167 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20tom=20W=C3=B6rden?= <h.tomwoerden@indiscale.com>
Date: Sat, 17 Jul 2021 17:48:14 +0200
Subject: [PATCH] fix and test
---
src/caosadvancedtools/models/parser.py | 2 +-
unittests/test_parser.py | 5 ++++-
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/src/caosadvancedtools/models/parser.py b/src/caosadvancedtools/models/parser.py
index f0371cc9..62d44461 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 64c98e9b..cff71be8 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):
--
GitLab