Skip to content

Commit 18a6b0d

Browse files
fixup test for python storage
1 parent 42b77db commit 18a6b0d

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pandas/tests/frame/test_reductions.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -681,7 +681,12 @@ def test_mode_sortwarning(self, using_infer_string):
681681
df = DataFrame({"A": [np.nan, np.nan, "a", "a"]})
682682
expected = DataFrame({"A": ["a", np.nan]})
683683

684-
warning = None if using_infer_string else UserWarning
684+
# TODO(infer_string) avoid this UserWarning for python storage
685+
warning = (
686+
None
687+
if using_infer_string and df.A.dtype.storage == "pyarrow"
688+
else UserWarning
689+
)
685690
with tm.assert_produces_warning(warning, match="Unable to sort modes"):
686691
result = df.mode(dropna=False)
687692
result = result.sort_values(by="A").reset_index(drop=True)

0 commit comments

Comments
 (0)