From 6b089453c4b178f87faa6c11f8e86d5b0dac94c5 Mon Sep 17 00:00:00 2001 From: Ben Gyori Date: Thu, 6 Jul 2023 11:54:06 -0400 Subject: [PATCH] Update test to match new id/name scheme --- tests/test_askenet_source.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/test_askenet_source.py b/tests/test_askenet_source.py index e24ed28a5..dda00ee82 100644 --- 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():