Skip to content

Commit eafca1c

Browse files
Revert "test_pivot_columns_is_none"
This reverts commit 2cad97c.
1 parent 2e1d33c commit eafca1c

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

pandas/tests/reshape/test_pivot.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2664,21 +2664,18 @@ def test_pivot_columns_not_given(self):
26642664
with pytest.raises(TypeError, match="missing 1 required keyword-only argument"):
26652665
df.pivot()
26662666

2667+
@pytest.mark.xfail(
2668+
using_string_dtype(), reason="TODO(infer_string) None is cast to NaN"
2669+
)
26672670
def test_pivot_columns_is_none(self):
26682671
# GH#48293
2669-
df = DataFrame([[1, 2, 3]], columns=Index([None, "b", "c"], dtype="object"))
2672+
df = DataFrame({None: [1], "b": 2, "c": 3})
26702673
result = df.pivot(columns=None)
26712674
expected = DataFrame({("b", 1): [2], ("c", 1): 3})
2672-
expected.columns = expected.columns.set_levels(
2673-
expected.columns.levels[0].astype(object), level=0
2674-
)
26752675
tm.assert_frame_equal(result, expected)
26762676

26772677
result = df.pivot(columns=None, index="b")
26782678
expected = DataFrame({("c", 1): 3}, index=Index([2], name="b"))
2679-
expected.columns = expected.columns.set_levels(
2680-
expected.columns.levels[0].astype(object), level=0
2681-
)
26822679
tm.assert_frame_equal(result, expected)
26832680

26842681
result = df.pivot(columns=None, index="b", values="c")

0 commit comments

Comments
 (0)