@@ -1528,12 +1528,13 @@ def test_datetimeindex_where() -> None:
15281528def 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