Skip to content

Commit 6267fe0

Browse files
committed
TST: Add test for ordered flag
Closes #27955
1 parent f6ae749 commit 6267fe0

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

pandas/tests/io/test_parquet.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -460,11 +460,18 @@ def test_categorical(self, pa):
460460
dtype = pd.CategoricalDtype(["foo", "bar", "baz"])
461461
df["b"] = pd.Categorical.from_codes(codes=[1, 0, 0, 1, -1, 1], dtype=dtype)
462462

463+
# test for ordered flag
464+
df["c"] = pd.Categorical(
465+
["a", "b", "c", "a"], categories=["b", "c", "d"], ordered=True
466+
)
467+
463468
if LooseVersion(pyarrow.__version__) >= LooseVersion("0.15.0"):
464469
check_round_trip(df, pa)
465470
else:
466471
# de-serialized as object for pyarrow < 0.15
467-
expected = df.assign(a=df.a.astype(object), b=df.b.astype(object))
472+
expected = df.assign(
473+
a=df.a.astype(object), b=df.b.astype(object), c=df.c.astype(object)
474+
)
468475
check_round_trip(df, pa, expected=expected)
469476

470477
def test_s3_roundtrip(self, df_compat, s3_resource, pa):

0 commit comments

Comments
 (0)