diff --git a/tests/common/beamlines/test_beamline_utils.py b/tests/common/beamlines/test_beamline_utils.py index 584742b9bdc..21ba692aaf5 100644 --- a/tests/common/beamlines/test_beamline_utils.py +++ b/tests/common/beamlines/test_beamline_utils.py @@ -19,7 +19,8 @@ @pytest.fixture(autouse=True) -def flush_event_loop_on_finish(event_loop): +def flush_event_loop_on_finish(): + event_loop = asyncio.get_event_loop() # wait for the test function to complete yield None diff --git a/tests/devices/unit_tests/i03/test_undulator_dcm.py b/tests/devices/unit_tests/i03/test_undulator_dcm.py index 16f4d7b745b..553b00ea426 100644 --- a/tests/devices/unit_tests/i03/test_undulator_dcm.py +++ b/tests/devices/unit_tests/i03/test_undulator_dcm.py @@ -16,7 +16,8 @@ @pytest.fixture(autouse=True) -def flush_event_loop_on_finish(event_loop): +def flush_event_loop_on_finish(): + event_loop = asyncio.get_event_loop() # wait for the test function to complete yield None