Fix mocking in test_python

logicalclocks · Jun 28, 2024 · c9a08f8 · c9a08f8
1 parent d17b9d5
commit c9a08f8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/tests/engine/test_python.py b/python/tests/engine/test_python.py
@@ -2978,6 +2978,7 @@ def test_write_training_dataset(self, mocker):
 
     def test_write_training_dataset_query_td(self, mocker, backend_fixtures):
         # Arrange
+        mocker.patch("hsfs.client.get_instance")
         mocker.patch("hsfs.engine.get_type")
         mocker.patch("hsfs.core.training_dataset_job_conf.TrainingDatasetJobConf")
         mock_job = mocker.patch("hsfs.core.job.Job")
@@ -3022,6 +3023,7 @@ def test_write_training_dataset_query_td(self, mocker, backend_fixtures):
 
     def test_write_training_dataset_query_fv(self, mocker, backend_fixtures):
         # Arrange
+        mocker.patch("hsfs.client.get_instance")
         mocker.patch("hsfs.engine.get_type")
         mocker.patch("hsfs.core.training_dataset_job_conf.TrainingDatasetJobConf")
         mock_job = mocker.patch("hsfs.core.job.Job")
@@ -3073,6 +3075,7 @@ def test_write_training_dataset_query_fv(self, mocker, backend_fixtures):
 
     def test_create_hive_connection(self, mocker):
         # Arrange
+        mocker.patch("hopsworks.client.get_instance")
         mocker.patch("hsfs.client.get_instance")
         mock_pyhive_conn = mocker.patch("pyhive.hive.Connection")