Skip to content

Commit 68fc429

Browse files
Apply suggestions from code review
Co-authored-by: Yi-Fan Wang <[email protected]>
1 parent 96f67d9 commit 68fc429

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

tests/indexes/test_indexes.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1528,12 +1528,13 @@ def test_datetimeindex_where() -> None:
15281528
def test_index_set_names() -> None:
15291529
"""Test Index.where with multiple types of other GH1419."""
15301530
idx = pd.Index([1, 2])
1531-
check(assert_type(idx.set_names("chinchilla"), "pd.Index[int]"), pd.Index, np.int64)
1532-
check(assert_type(idx.set_names((0,)), "pd.Index[int]"), pd.Index, np.int64)
1531+
check(assert_type(idx.set_names("chinchilla"), "pd.Index[int]"), pd.Index, np.integer)
1532+
check(assert_type(idx.set_names((0,)), "pd.Index[int]"), pd.Index, np.integer)
15331533
check(
1534-
assert_type(idx.set_names(["chinchilla"]), "pd.Index[int]"), pd.Index, np.int64
1534+
assert_type(idx.set_names(["chinchilla"]), "pd.Index[int]"), pd.Index, np.integer
15351535
)
15361536

15371537
mi = pd.MultiIndex.from_arrays([[1, 2, 3], [4, 5, 6]], names=["elk", "owl"])
1538-
mi.set_names(["beluga", "pig"])
1539-
mi.set_names({"elk": "beluga", "owl": "pig"})
1538+
check(assert_type(mi.set_names(["beluga", "pig"]), pd.MultiIndex), pd.MultiIndex)
1539+
check(assert_type(mi.set_names({"elk": "beluga", "owl": "pig"}), pd.MultiIndex), pd.MultiIndex)
1540+
check(assert_type(pd.Index([(1,)]).set_names(1), pd.MultiIndex), pd.MultiIndex, tuple)

0 commit comments

Comments
 (0)