File tree 1 file changed +3
-9
lines changed
1 file changed +3
-9
lines changed Original file line number Diff line number Diff line change @@ -4240,9 +4240,7 @@ def test_model_evaluate_with_gcs_input_uris(
4240
4240
4241
4241
assert isinstance (eval_job , model_evaluation_job ._ModelEvaluationJob )
4242
4242
4243
- assert mock_pipeline_service_create .called_once
4244
-
4245
- assert mock_pipeline_service_get .called_once
4243
+ mock_pipeline_service_create .assert_called_once ()
4246
4244
4247
4245
eval_job .wait ()
4248
4246
@@ -4283,9 +4281,7 @@ def test_model_evaluate_with_bigquery_input(
4283
4281
4284
4282
assert isinstance (eval_job , model_evaluation_job ._ModelEvaluationJob )
4285
4283
4286
- assert mock_pipeline_service_create .called_once
4287
-
4288
- assert mock_pipeline_service_get .called_once
4284
+ mock_pipeline_service_create .assert_called_once ()
4289
4285
4290
4286
@pytest .mark .parametrize (
4291
4287
"job_spec_json" ,
@@ -4315,9 +4311,7 @@ def test_model_evaluate_using_initialized_staging_bucket(
4315
4311
4316
4312
assert isinstance (eval_job , model_evaluation_job ._ModelEvaluationJob )
4317
4313
4318
- assert mock_pipeline_service_create .called_once
4319
-
4320
- assert mock_pipeline_service_get .called_once
4314
+ mock_pipeline_service_create .assert_called_once ()
4321
4315
4322
4316
def test_model_evaluate_with_no_staging_path_or_initialized_staging_bucket_raises (
4323
4317
self ,
You can’t perform that action at this time.
0 commit comments