Fix invocation of reload_modules fixture

dagardner-nv · Apr 5, 2023 · f460f66 · f460f66
1 parent c8992cd
commit f460f66
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/tests/test_dfp.py b/tests/test_dfp.py
@@ -47,8 +47,7 @@
 
 @pytest.mark.slow
 @pytest.mark.use_python
-@pytest.mark.reload_modules(preprocess_ae_stage)
-@pytest.mark.reload_modules(train_ae_stage)
+@pytest.mark.reload_modules([preprocess_ae_stage, train_ae_stage])
 @pytest.mark.usefixtures("reload_modules")
 @mock.patch('morpheus.stages.preprocess.train_ae_stage.AutoEncoder')
 def test_dfp_roleg(mock_ae, config, tmp_path):
@@ -131,8 +130,7 @@ def test_dfp_roleg(mock_ae, config, tmp_path):
 
 @pytest.mark.slow
 @pytest.mark.use_python
-@pytest.mark.reload_modules(preprocess_ae_stage)
-@pytest.mark.reload_modules(train_ae_stage)
+@pytest.mark.reload_modules([preprocess_ae_stage, train_ae_stage])
 @pytest.mark.usefixtures("reload_modules")
 @mock.patch('morpheus.stages.preprocess.train_ae_stage.AutoEncoder')
 def test_dfp_user123(mock_ae, config, tmp_path):
@@ -213,8 +211,7 @@ def test_dfp_user123(mock_ae, config, tmp_path):
 
 @pytest.mark.slow
 @pytest.mark.use_python
-@pytest.mark.reload_modules(preprocess_ae_stage)
-@pytest.mark.reload_modules(train_ae_stage)
+@pytest.mark.reload_modules([preprocess_ae_stage, train_ae_stage])
 @pytest.mark.usefixtures("reload_modules")
 @mock.patch('morpheus.stages.preprocess.train_ae_stage.AutoEncoder')
 def test_dfp_user123_multi_segment(mock_ae, config, tmp_path):

diff --git a/tests/test_dfp_kafka.py b/tests/test_dfp_kafka.py
@@ -55,9 +55,7 @@
 @pytest.mark.kafka
 @pytest.mark.slow
 @pytest.mark.use_python
-@pytest.mark.reload_modules(commands)
-@pytest.mark.reload_modules(preprocess_ae_stage)
-@pytest.mark.reload_modules(train_ae_stage)
+@pytest.mark.reload_modules([commands, preprocess_ae_stage, train_ae_stage])
 @pytest.mark.usefixtures("reload_modules")
 @mock.patch('morpheus.stages.preprocess.train_ae_stage.AutoEncoder')
 def test_dfp_roleg(mock_ae,
@@ -159,8 +157,7 @@ def test_dfp_roleg(mock_ae,
 @pytest.mark.kafka
 @pytest.mark.slow
 @pytest.mark.use_python
-@pytest.mark.reload_modules(preprocess_ae_stage)
-@pytest.mark.reload_modules(train_ae_stage)
+@pytest.mark.reload_modules([preprocess_ae_stage, train_ae_stage])
 @pytest.mark.usefixtures("reload_modules")
 @mock.patch('morpheus.stages.preprocess.train_ae_stage.AutoEncoder')
 def test_dfp_user123(mock_ae,