Skip to content

Commit 5dd2bb4

Browse files
assert result after assignment
1 parent c30cf1d commit 5dd2bb4

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

pandas/tests/indexing/multiindex/test_multiindex.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -274,10 +274,21 @@ def test_multiindex_assign_aligns_as_implicit_tuple(self):
274274

275275
def test_multiindex_assign_alignment_with_non_string_dtype(self):
276276
# GH 62518
277-
meta = DataFrame(
278-
columns=MultiIndex.from_arrays(
279-
[["a", "a", "z", "z"], pd.Categorical([1, 2, 1, 2])]
280-
),
281-
dtype=object,
277+
columns = MultiIndex.from_arrays(
278+
[["a", "a", "z", "z"], pd.Categorical([1, 2, 1, 2])]
282279
)
280+
281+
meta = DataFrame(columns=columns, dtype=object)
283282
meta["z"] = meta["z"].astype("int64")
283+
284+
result = DataFrame(
285+
data={
286+
("a", 1): Series([], dtype=object),
287+
("a", 2): Series([], dtype=object),
288+
("z", 1): Series([], dtype="int64"),
289+
("z", 2): Series([], dtype="int64"),
290+
},
291+
columns=columns,
292+
)
293+
294+
tm.assert_frame_equal(meta, result)

0 commit comments

Comments
 (0)