remove assert before assert_called_once_with

Author: mollyheamazon

tests/unit/sagemaker/feature_store/feature_processor/test_config_uploader.py

@@ -100,7 +100,7 @@ def remote_decorator_config_with_filter(sagemaker_session):
 def test_prepare_and_upload_callable(mock_stored_function, config_uploader, wrapped_func):
     mock_stored_function.save(wrapped_func).return_value = None
     config_uploader._prepare_and_upload_callable(wrapped_func, "s3_base_uri", sagemaker_session)
-    assert mock_stored_function.assert_called_once_with(
+    mock_stored_function.assert_called_once_with(
         s3_base_uri="s3_base_uri",
         s3_kms_key=config_uploader.remote_decorator_config.s3_kms_key,
         hmac_key="some_secret_key",
@@ -244,7 +244,7 @@ def test_prepare_step_input_channel(
     )
     remote_decorator_config = config_uploader.remote_decorator_config
 
-    assert mock_upload_callable.assert_called_once_with(wrapped_func)
+    mock_upload_callable.assert_called_once_with(wrapped_func)
 
     mock_script_upload.assert_called_once_with(
         spark_config=config_uploader.remote_decorator_config.spark_config,

tests/unit/sagemaker/feature_store/feature_processor/test_feature_scheduler.py

@@ -269,7 +269,7 @@ def test_to_pipeline(
     )
     assert pipeline_arn == PIPELINE_ARN
 
-    assert mock_upload_callable.assert_called_once_with(job_function)
+    mock_upload_callable.assert_called_once_with(job_function)
     local_dependencies_path = mock_runtime_manager().snapshot()
     mock_python_version = mock_runtime_manager()._current_python_version()
     container_args.extend(["--client_python_version", mock_python_version])

tests/unit/sagemaker/modules/train/test_model_trainer.py

@@ -1048,7 +1048,7 @@ def mock_upload_data(path, bucket, key_prefix):
 
     model_trainer.train()
 
-    assert mock_local_container.train.assert_called_once_with(
+    mock_local_container.train.assert_called_once_with(
         training_job_name=unique_name,
         instance_type=compute.instance_type,
         instance_count=compute.instance_count,

tests/unit/sagemaker/remote_function/runtime_environment/test_runtime_environment_manager.py

@@ -450,7 +450,7 @@ def test_run_pre_exec_script_cmd_error(isfile):
 def test_change_dir_permission(mock_subprocess_run):
     RuntimeEnvironmentManager().change_dir_permission(dirs=["a", "b", "c"], new_permission="777")
     expected_command = ["sudo", "chmod", "-R", "777", "a", "b", "c"]
-    assert mock_subprocess_run.assert_called_once_with(
+    mock_subprocess_run.assert_called_once_with(
         expected_command, check=True, stderr=subprocess.PIPE
     )
 

tests/unit/sagemaker/remote_function/test_job.py

@@ -1190,7 +1190,7 @@ def test_start_with_spark(
 
     assert job.job_name.startswith("job-function")
 
-    assert mock_stored_function.assert_called_once_with(
+    mock_stored_function.assert_called_once_with(
         sagemaker_session=session(), s3_base_uri=f"{S3_URI}/{job.job_name}", s3_kms_key=None
     )
 

tests/unit/test_session.py

@@ -5781,7 +5781,7 @@ def test_start_query_execution(sagemaker_session):
         query_string="query",
         output_location="s3://results",
     )
-    assert athena_mock.start_query_execution.assert_called_once_with(
+    athena_mock.start_query_execution.assert_called_once_with(
         QueryString="query",
         QueryExecutionContext={"Catalog": "catalog", "Database": "database"},
         OutputLocation="s3://results",