def test_read_psv_xz(setup_bucket_w_dfs, test_bucket, test_df, test_df_keys):
"""
Tests that reading files stored as an xz-compressed PSV works properly
"""
for key in test_df_keys['psv.xz']:
df = read(key, test_bucket)
assert df.equals(test_df)
def test_read_csv_bz2(setup_bucket_w_dfs, test_bucket, test_df, test_df_keys):
"""
Tests that reading files stored as a bzip2-compressed CSV works properly
"""
for key in test_df_keys['csv.bz2']:
df = read(key, test_bucket)
assert df.equals(test_df)
def test_read_json(setup_bucket_w_dfs, test_bucket, test_df, test_df_keys):
"""Tests that reading files stored as JSON works properly"""
for key in test_df_keys['json']:
df = read(key, test_bucket)
assert df.equals(test_df)
def test_read_feather(setup_bucket_w_dfs, test_bucket, test_df, test_df_keys):
"""Tests that reading files stored as feather works properly"""
for key in test_df_keys['feather']:
df = read(key, test_bucket)
assert df.equals(test_df)
def test_read_pq(setup_bucket_w_dfs, test_bucket, test_df, test_df_keys):
"""Tests that reading files stored as Parquet works properly"""
for key in test_df_keys['pq']:
df = read(key, test_bucket)
assert df.equals(test_df)
def test_read_pkl(setup_bucket_w_dfs, test_bucket, test_df, test_df_keys):
"""Tests that reading pickled files works properly"""
for key in test_df_keys['pkl']:
df = read(key, test_bucket)
assert df.equals(test_df)
