def test_move(teardown):
local_file = get_local_filename(BLOB_FILE)
temp_file = get_temp_local_filename(BLOB_FILE)
temp_file2 = get_temp_local_filename(BLOB_FILE)
teardown.delete(temp_file)
teardown.delete(temp_file2)
storage.copy(local_file, temp_file)
storage.move(temp_file, temp_file2)
assert_that(storage.get(temp_file2)).is_equal_to(BLOB_DATA)
assert_that(storage.exists(temp_file)).is_false()
def test_get_blob_from_gcs():
remote_file = get_gcs_filename(BLOB_FILE)
assert_that(storage.get(remote_file)).is_equal_to(BLOB_DATA)
def test_get_tsv_from_gcs():
remote_file = get_gcs_filename(TSV_FILE)
assert_that(storage.get(remote_file)).is_equal_to(TSV_DATA)
def test_get_json_from_gcs():
remote_file = get_gcs_filename(JSON_FILE)
assert_that(storage.get(remote_file)).is_equal_to(JSON_DATA)
def test_copy_remote_to_remote(teardown):
remote_file = get_gcs_filename(BLOB_FILE)
remote_temp_file = get_temp_gcs_filename(BLOB_FILE)
teardown.delete(remote_temp_file)
storage.copy(remote_file, remote_temp_file)
assert_that(storage.get(remote_temp_file)).is_equal_to(BLOB_DATA)
def test_copy_remote_to_remote_with_filetype_change(teardown):
remote_file = get_gcs_filename(CSV_FILE)
remote_temp_file = get_temp_gcs_filename(JSON_FILE)
teardown.delete(remote_temp_file)
storage.copy(remote_file, remote_temp_file)
assert_that(storage.get(remote_temp_file)).is_equal_to(CSV_DATA)
def test_save_blob_to_local(teardown):
local_file = get_temp_local_filename(BLOB_FILE)
teardown.delete(local_file)
storage.save(BLOB_DATA, local_file)
assert_that(storage.get(local_file)).is_equal_to(BLOB_DATA)
def test_copy_local_to_local(teardown):
local_file = get_local_filename(BLOB_FILE)
temp_file = get_temp_local_filename(BLOB_FILE)
teardown.delete(temp_file)
storage.copy(local_file, temp_file)
assert_that(storage.get(temp_file)).is_equal_to(BLOB_DATA)
def test_save_pandas_to_csv(teardown):
local_file = get_temp_local_filename(CSV_FILE)
teardown.delete(local_file)
storage.save(pd.DataFrame(CSV_DATA), local_file)
assert_that(storage.get(local_file)).is_equal_to(CSV_DATA)
def test_save_blob_to_gcs(teardown):
remote_file = get_temp_gcs_filename(BLOB_FILE)
teardown.delete(remote_file)
storage.save(BLOB_DATA, remote_file)
assert_that(storage.get(remote_file)).is_equal_to(BLOB_DATA)
def test_save_raw_csv_to_local(teardown):
local_file = get_temp_local_filename(CSV_FILE)
teardown.delete(local_file)
storage.save(CSV_DATA_RAW, local_file, asBlob=True)
assert_that(storage.get(local_file)).is_equal_to(CSV_DATA)
def test_get_blob_from_local():
local_file = get_local_filename(BLOB_FILE)
assert_that(storage.get(local_file)).is_equal_to(BLOB_DATA)
def test_get_tsv_from_local():
local_file = get_local_filename(TSV_FILE)
assert_that(storage.get(local_file)).is_equal_to(TSV_DATA)
def test_get_json_from_local_as_blob():
local_file = get_local_filename(JSON_FILE)
assert_that(storage.get(local_file, asBlob=True)).is_equal_to(JSON_DATA_RAW)
def test_get_json_from_local():
local_file = get_local_filename(JSON_FILE)
assert_that(storage.get(local_file)).is_equal_to(JSON_DATA)