Update test to match new id/name scheme

gyorilab · Jul 6, 2023 · 6b08945 · 6b08945
1 parent 28a9303
commit 6b08945
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/test_askenet_source.py b/tests/test_askenet_source.py
@@ -14,11 +14,12 @@ def test_petrinet_model_from_url():
     assert len(template_model.templates) == 2
     assert isinstance(template_model.templates[0], ControlledConversion)
     assert isinstance(template_model.templates[1], NaturalConversion)
-    assert template_model.templates[0].controller.name == 'Infected'
-    assert template_model.templates[0].subject.name == 'Susceptible'
-    assert template_model.templates[0].outcome.name == 'Infected'
-    assert template_model.templates[1].subject.name == 'Infected'
-    assert template_model.templates[1].outcome.name == 'Recovered'
+    assert template_model.templates[0].controller.display_name == 'Infected'
+    assert template_model.templates[0].controller.name == 'I'
+    assert template_model.templates[0].subject.display_name == 'Susceptible'
+    assert template_model.templates[0].outcome.display_name == 'Infected'
+    assert template_model.templates[1].subject.display_name == 'Infected'
+    assert template_model.templates[1].outcome.display_name == 'Recovered'
 
 
 def test_regnet_model_from_url():