From 2839be098d8adf94cc908fd7a9a387cdbc94c841 Mon Sep 17 00:00:00 2001 From: Lauri Himanen Date: Fri, 3 May 2024 08:44:52 +0300 Subject: [PATCH] Fixed test calls. --- simulationworkflownormalizer/normalizer.py | 2 +- tests/test_simulationworkflownormalizer.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/simulationworkflownormalizer/normalizer.py b/simulationworkflownormalizer/normalizer.py index 4a04afb..fcab4af 100644 --- a/simulationworkflownormalizer/normalizer.py +++ b/simulationworkflownormalizer/normalizer.py @@ -75,7 +75,7 @@ def _resolve_workflow(self, archive: EntryArchive): def normalize(self, archive: EntryArchive, logger=None) -> None: logger = logger if logger is not None else get_logger(__name__) - super().normalize(logger) + super().normalize(archive, logger) # Do nothing if run section is not present if not archive.run: diff --git a/tests/test_simulationworkflownormalizer.py b/tests/test_simulationworkflownormalizer.py index 32729f0..9733d21 100644 --- a/tests/test_simulationworkflownormalizer.py +++ b/tests/test_simulationworkflownormalizer.py @@ -46,7 +46,7 @@ def test_resolve_workflow_from_program_name( ): run = Run(program=Program(name=program_name)) entry_archive.run.append(run) - SimulationWorkflowNormalizer(entry_archive).normalize(get_logger(__name__)) + SimulationWorkflowNormalizer().normalize(entry_archive, get_logger(__name__)) assert isinstance(entry_archive.workflow2, workflow_class) @@ -58,5 +58,5 @@ def test_resolve_workflow_from_calculation( ): run = Run(calculation=[Calculation() for _ in range(n_calculations)]) entry_archive.run.append(run) - SimulationWorkflowNormalizer(entry_archive).normalize(get_logger(__name__)) + SimulationWorkflowNormalizer().normalize(entry_archive, get_logger(__name__)) assert isinstance(entry_archive.workflow2, workflow_class)