Skip to content

Commit 5fb15b6

Browse files
author
wdyy20041223
committed
TST: Add regression tests for enlarging MultiIndex with None keys (GH#59153)
1 parent 7cebec6 commit 5fb15b6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pandas/tests/indexing/multiindex/test_setitem.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -506,15 +506,15 @@ def test_setitem_enlargement_multiindex_with_none(self):
506506
[("A", "a1"), ("A", "a2"), ("B", "b1"), ("B", None), ("A", None)]
507507
)
508508
expected = DataFrame(
509-
[[0, 6], [1, 5], [2, 4], [3, 7], [12, 13]],
509+
[[0.0, 6.0], [1.0, 5.0], [2.0, 4.0], [3.0, 7.0], [12.0, 13.0]],
510510
index=expected_index,
511511
columns=[0, 1],
512512
)
513513
tm.assert_frame_equal(df, expected)
514514

515515
# Test retrieval of the newly added row
516516
result = df.loc[("A", None), :]
517-
expected_row = Series([12, 13], index=[0, 1], name=("A", np.nan))
517+
expected_row = Series([12.0, 13.0], index=[0, 1], name=("A", np.nan))
518518
tm.assert_series_equal(result, expected_row)
519519

520520
def test_setitem_enlargement_multiindex_multiple_none(self):
@@ -538,7 +538,7 @@ def test_setitem_enlargement_multiindex_multiple_none(self):
538538
]
539539
)
540540
expected = DataFrame(
541-
[[1, 2], [3, 4], [5, 6], [7, 8]],
541+
[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]],
542542
index=expected_index,
543543
columns=["x", "y"],
544544
)

0 commit comments

Comments
 (0)