diff --git a/tests/test_google_cloud_storage.py b/tests/test_google_cloud_storage.py index eb95323d..e7714344 100644 --- a/tests/test_google_cloud_storage.py +++ b/tests/test_google_cloud_storage.py @@ -11,22 +11,32 @@ def patch_gcs_client(mocker): def test_upload_files(patch_gcs_client): gcs_storage = GoogleCloudStorage(bucket_name="benchmark_bucket") - gcs_storage.upload_files(output_files=["mock_output_stats.csv"], directory="benchmark_outputs") + gcs_storage.upload_files( + output_files=["mock_output_stats.csv"], directory="benchmark_outputs" + ) bucket = patch_gcs_client.return_value.get_bucket.return_value blob = bucket.blob.return_value assert blob.upload_from_filename.call_count == 1 - assert blob.upload_from_filename.call_args[1] == {"filename": "mock_output_stats.csv"} + assert blob.upload_from_filename.call_args[1] == { + "filename": "mock_output_stats.csv" + } def test_upload_files_with_prefix(patch_gcs_client): gcs_storage = GoogleCloudStorage(bucket_name="benchmark_bucket") - gcs_storage.upload_files(output_files=["mock_output_stats.csv"], directory="benchmark_outputs", output_filename_prefix="test") + gcs_storage.upload_files( + output_files=["mock_output_stats.csv"], + directory="benchmark_outputs", + output_filename_prefix="test", + ) bucket = patch_gcs_client.return_value.get_bucket.return_value blob = bucket.blob.return_value assert blob.upload_from_filename.call_count == 1 - assert blob.upload_from_filename.call_args[1] == {"filename": "mock_output_stats.csv"} + assert blob.upload_from_filename.call_args[1] == { + "filename": "mock_output_stats.csv" + }