diff --git a/src/caosadvancedtools/models/data_model.py b/src/caosadvancedtools/models/data_model.py index 1390cf2e45b83c7fd715e674706bc98173db4ebf..e0582365c0106eba08bd0ea6211fb447ca65eb02 100644 --- a/src/caosadvancedtools/models/data_model.py +++ b/src/caosadvancedtools/models/data_model.py @@ -148,8 +148,8 @@ class DataModel(dict): else: query = db.Query(f"FIND ENTITY with id={ent.id}") ref = query.execute(unique=True) - diff = (describe_diff(*compare_entities(ent, ref - ), name=ent.name)) + diff = (describe_diff(*compare_entities(ent, ref), + name=ent.name)) if diff != "": if verbose: diff --git a/unittests/test_yaml_model_parser.py b/unittests/test_yaml_model_parser.py index 56c8672b0e073101a38329037d7006e99b88a359..a80ef7131cb0454faba5a7385ea4d0e26c1395ec 100644 --- a/unittests/test_yaml_model_parser.py +++ b/unittests/test_yaml_model_parser.py @@ -22,6 +22,8 @@ from tempfile import NamedTemporaryFile from pytest import raises, mark import linkahead as db + +import caosadvancedtools from caosadvancedtools.models.parser import (TwiceDefinedException, YamlDefinitionError, parse_model_from_string, @@ -714,6 +716,7 @@ test_reference: assert "id" in cs[1] mq = Mock() + def mq_init(self, query): self.query = query @@ -729,4 +732,22 @@ test_reference: stdout, stderr = capfd.readouterr() - # TODO: test that there were no changes required + # TODO: test that there were no changes required + + +def test_setting_values(): + model = parse_model_from_string(""" + parameter: + datatype: INTEGER + + Simulation: + role: Record + obligatory_properties: + parameter: 26 + """) + + assert len(model) == 2 + assert str(model["parameter"])[:-1] == '<Property name="parameter" datatype="INTEGER"/>' + assert model["Simulation"].role == "Record" + assert model["Simulation"].name == "Simulation" + assert model["Simulation"].get_property("parameter").value == 26