Skip to content

Commit 6b5e114

Browse files
committed
move test to TestParquetPyarrow
1 parent b945aad commit 6b5e114

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

pandas/tests/io/test_parquet.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -376,19 +376,6 @@ def test_columns_dtypes_invalid(self, engine):
376376
]
377377
self.check_error_on_write(df, engine, ValueError)
378378

379-
def test_to_bytes_without_path_or_buf_provided(self, engine, df_full):
380-
# GH 37105
381-
382-
buf = df_full.to_parquet(engine=engine)
383-
assert isinstance(buf, bytes)
384-
385-
with tm.ensure_clean() as path:
386-
with open(path, "wb") as f:
387-
f.write(buf)
388-
res = pd.read_parquet(path)
389-
390-
tm.assert_frame_equal(df_full, res)
391-
392379
@pytest.mark.parametrize("compression", [None, "gzip", "snappy", "brotli"])
393380
def test_compression(self, engine, compression):
394381

@@ -525,6 +512,19 @@ def test_basic_subset_columns(self, pa, df_full):
525512
read_kwargs={"columns": ["string", "int"]},
526513
)
527514

515+
def test_to_bytes_without_path_or_buf_provided(self, pa, df_full):
516+
# GH 37105
517+
518+
buf = df_full.to_parquet(engine=pa)
519+
assert isinstance(buf, bytes)
520+
521+
with tm.ensure_clean() as path:
522+
with open(path, "wb") as f:
523+
f.write(buf)
524+
res = pd.read_parquet(path)
525+
526+
tm.assert_frame_equal(df_full, res)
527+
528528
def test_duplicate_columns(self, pa):
529529
# not currently able to handle duplicate columns
530530
df = pd.DataFrame(np.arange(12).reshape(4, 3), columns=list("aaa")).copy()

0 commit comments

Comments
 (0)