def test_pandas_parquet_roundtrip1(tmpdir): df = DataFrame({"a": [1, 2, 3], "b": ["a", "b", "c"]}) filename = os.path.join(str(tmpdir), "test.pq") df.to_parquet(filename) pq_df = pd_read_parquet(filename) assert_frame_equal(df, pq_df)
def test_pandas_parquet_roundtrip2(test_dataset, tmpdir): test_dataset = "naturalearth_lowres" df = DataFrame(read_file(get_path(test_dataset)).drop(columns=["geometry"])) filename = os.path.join(str(tmpdir), "test.pq") df.to_parquet(filename) pq_df = pd_read_parquet(filename) assert_frame_equal(df, pq_df)