Fix mocking in test_transformer

aversey · aversey · commit b30f8b08c010 · 2024-09-10T09:43:02.000+02:00
diff --git a/python/tests/test_transformer.py b/python/tests/test_transformer.py
@@ -298,12 +298,14 @@ def test_extract_fields_from_json(self, mocker, backend_fixtures):
 
     def _mock_serving_variables(self, mocker, num_instances, force_scale_to_zero=False):
         mocker.patch(
-            "hsml.client.get_serving_resource_limits",
+            "hopsworks_common.client.get_serving_resource_limits",
             return_value=SERVING_RESOURCE_LIMITS,
         )
         mocker.patch(
-            "hsml.client.get_serving_num_instances_limits", return_value=num_instances
+            "hopsworks_common.client.get_serving_num_instances_limits",
+            return_value=num_instances,
         )
         mocker.patch(
-            "hsml.client.is_scale_to_zero_required", return_value=force_scale_to_zero
+            "hopsworks_common.client.is_scale_to_zero_required",
+            return_value=force_scale_to_zero,
         )

Original file line number	Diff line number	Diff line change
`@@ -298,12 +298,14 @@ def test_extract_fields_from_json(self, mocker, backend_fixtures):`
`298`	`298`
`299`	`299`	`def _mock_serving_variables(self, mocker, num_instances, force_scale_to_zero=False):`
`300`	`300`	`mocker.patch(`
`301`		`- "hsml.client.get_serving_resource_limits",`
	`301`	`+ "hopsworks_common.client.get_serving_resource_limits",`
`302`	`302`	`return_value=SERVING_RESOURCE_LIMITS,`
`303`	`303`	`)`
`304`	`304`	`mocker.patch(`
`305`		`- "hsml.client.get_serving_num_instances_limits", return_value=num_instances`
	`305`	`+ "hopsworks_common.client.get_serving_num_instances_limits",`
	`306`	`+ return_value=num_instances,`
`306`	`307`	`)`
`307`	`308`	`mocker.patch(`
`308`		`- "hsml.client.is_scale_to_zero_required", return_value=force_scale_to_zero`
	`309`	`+ "hopsworks_common.client.is_scale_to_zero_required",`
	`310`	`+ return_value=force_scale_to_zero,`
`309`	`311`	`)`