From e30c6e1ad0da3c97ce84e185c89fa31b22cb8a81 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sami=20M=C3=A4kinen?= Date: Thu, 6 Feb 2025 10:46:33 +0200 Subject: [PATCH] Fix assignment tests --- Scripts/tests/emme_only/test_assignment.py | 3 ++- Scripts/tests/unit/test_assignment.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/Scripts/tests/emme_only/test_assignment.py b/Scripts/tests/emme_only/test_assignment.py index aab0cd22..8b9ba5ad 100644 --- a/Scripts/tests/emme_only/test_assignment.py +++ b/Scripts/tests/emme_only/test_assignment.py @@ -3,6 +3,7 @@ import os import logging +from events.model_system_event_listener import EventHandler import utils.log as log import numpy @@ -70,7 +71,7 @@ def __init__(self): emme_context.import_scenario( os.path.join(project_dir, "..", "Network"), scenario_num, "test", overwrite=True) - self.ass_model = ass.EmmeAssignmentModel(emme_context, scenario_num) + self.ass_model = ass.EmmeAssignmentModel(emme_context, scenario_num, EventHandler()) self.ass_model.prepare_network() def test_assignment(self): diff --git a/Scripts/tests/unit/test_assignment.py b/Scripts/tests/unit/test_assignment.py index 03851e31..e0237c05 100644 --- a/Scripts/tests/unit/test_assignment.py +++ b/Scripts/tests/unit/test_assignment.py @@ -5,6 +5,7 @@ import pandas import os +from events.model_system_event_listener import EventHandler from utils.validate_network import validate from assignment.emme_bindings.mock_project import MockProject from assignment.emme_assignment import EmmeAssignmentModel @@ -32,7 +33,7 @@ def test_assignment(self): context.modeller.emmebank.scenario(scenario_id).get_network(), fares) ass_model = EmmeAssignmentModel( - context, scenario_id) + context, scenario_id, EventHandler()) ass_model.prepare_network() peripheral_cost = numpy.arange(10).reshape((1, 10)) ass_model.calc_transit_cost(fares, peripheral_cost)