Skip to content

Commit 181c7e5

Browse files
test_pivot_index_is_none
1 parent 888d5cb commit 181c7e5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pandas/tests/reshape/test_pivot.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -2682,15 +2682,15 @@ def test_pivot_columns_is_none(self):
26822682
expected = DataFrame({1: 3}, index=Index([2], name="b"))
26832683
tm.assert_frame_equal(result, expected)
26842684

2685-
@pytest.mark.xfail(
2686-
using_string_dtype(), reason="TODO(infer_string) None is cast to NaN"
2687-
)
26882685
def test_pivot_index_is_none(self):
26892686
# GH#48293
2690-
df = DataFrame({None: [1], "b": 2, "c": 3})
2687+
df = DataFrame([[1, 2, 3]], columns=Index([None, "b", "c"], dtype="object"))
26912688

26922689
result = df.pivot(columns="b", index=None)
26932690
expected = DataFrame({("c", 2): 3}, index=[1])
2691+
expected.columns = expected.columns.set_levels(
2692+
expected.columns.levels[0].astype(object), level=0
2693+
)
26942694
expected.columns.names = [None, "b"]
26952695
tm.assert_frame_equal(result, expected)
26962696

0 commit comments

Comments
 (0)