Skip to content

Commit c9a08f8

Browse files
committed
Fix mocking in test_python
1 parent d17b9d5 commit c9a08f8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/tests/engine/test_python.py

+3
Original file line numberDiff line numberDiff line change
@@ -2978,6 +2978,7 @@ def test_write_training_dataset(self, mocker):
29782978

29792979
def test_write_training_dataset_query_td(self, mocker, backend_fixtures):
29802980
# Arrange
2981+
mocker.patch("hsfs.client.get_instance")
29812982
mocker.patch("hsfs.engine.get_type")
29822983
mocker.patch("hsfs.core.training_dataset_job_conf.TrainingDatasetJobConf")
29832984
mock_job = mocker.patch("hsfs.core.job.Job")
@@ -3022,6 +3023,7 @@ def test_write_training_dataset_query_td(self, mocker, backend_fixtures):
30223023

30233024
def test_write_training_dataset_query_fv(self, mocker, backend_fixtures):
30243025
# Arrange
3026+
mocker.patch("hsfs.client.get_instance")
30253027
mocker.patch("hsfs.engine.get_type")
30263028
mocker.patch("hsfs.core.training_dataset_job_conf.TrainingDatasetJobConf")
30273029
mock_job = mocker.patch("hsfs.core.job.Job")
@@ -3073,6 +3075,7 @@ def test_write_training_dataset_query_fv(self, mocker, backend_fixtures):
30733075

30743076
def test_create_hive_connection(self, mocker):
30753077
# Arrange
3078+
mocker.patch("hopsworks.client.get_instance")
30763079
mocker.patch("hsfs.client.get_instance")
30773080
mock_pyhive_conn = mocker.patch("pyhive.hive.Connection")
30783081

0 commit comments

Comments
 (0)