WIP package version upgrade

- fix unit test (thx SC)
blobquiet · May 23, 2023 · 427bfb6 · 427bfb6
1 parent ca42dd0
commit 427bfb6
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/tests/test_job.py b/tests/test_job.py
@@ -26,7 +26,7 @@
 from fedbiomed.researcher.job import Job
 from fedbiomed.researcher.requests import Requests
 from fedbiomed.researcher.responses import Responses
-
+import fedbiomed.researcher.job # needed for specific mocking
 
 
 class TestJob(ResearcherTestCase):
@@ -189,18 +189,17 @@ def test_job_init_05_build_wrongly_saved_model(self, mock_logger_critical):
             mock_logger_critical.assert_called_once()
 
     @patch('fedbiomed.common.logger.logger.critical')
-    @patch('inspect.isclass')
     def test_job_06_init_isclass_raises_error(self,
-                                              mock_isclass,
                                               mock_logger_critical):
         """ Test initialization when inspect.isclass raises NameError"""
 
-        mock_isclass.side_effect = NameError
-        with self.assertRaises(NameError):
-            _ = Job(training_plan_class='FakeModel',
-                    training_args=TrainingArgs({"batch_size": 12}, only_required=False),
-                    data=self.fds)
-            mock_logger_critical.assert_called_once()
+        with patch.object(fedbiomed.researcher.job, 'inspect') as mock_inspect:
+            mock_inspect.isclass.side_effect = NameError
+            with self.assertRaises(NameError):
+                _ = Job(training_plan_class='FakeModel',
+                        training_args=TrainingArgs({"batch_size": 12}, only_required=False),
+                        data=self.fds)
+                mock_logger_critical.assert_called_once()
 
     @patch('fedbiomed.common.logger.logger.error')
     def test_job_07_initialization_raising_exception_save_and_save_code(self,