diff --git a/tests/test_gcloud.py b/tests/test_gcloud.py index 758c66e4e..eec346097 100644 --- a/tests/test_gcloud.py +++ b/tests/test_gcloud.py @@ -40,7 +40,7 @@ def test_open_read(self): self.storage._client.bucket.assert_called_with(self.bucket_name) self.storage._bucket.get_blob.assert_called_with(self.filename, timeout=60) - f.blob.download_to_file = lambda tmpfile: tmpfile.write(data) + f.blob.download_to_file = lambda tmpfile, **kwargs: tmpfile.write(data) self.assertEqual(f.read(), data) def test_open_read_num_bytes(self): @@ -51,7 +51,7 @@ def test_open_read_num_bytes(self): self.storage._client.bucket.assert_called_with(self.bucket_name) self.storage._bucket.get_blob.assert_called_with(self.filename, timeout=60) - f.blob.download_to_file = lambda tmpfile: tmpfile.write(data) + f.blob.download_to_file = lambda tmpfile, **kwargs: tmpfile.write(data) self.assertEqual(f.read(num_bytes), data[0:num_bytes]) def test_open_read_nonexistent(self):