@@ -1528,13 +1528,22 @@ 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 .integer )
1531+ check (
1532+ assert_type (idx .set_names ("chinchilla" ), "pd.Index[int]" ), pd .Index , np .integer
1533+ )
15321534 check (assert_type (idx .set_names ((0 ,)), "pd.Index[int]" ), pd .Index , np .integer )
15331535 check (
1534- assert_type (idx .set_names (["chinchilla" ]), "pd.Index[int]" ), pd .Index , np .integer
1536+ assert_type (idx .set_names (["chinchilla" ]), "pd.Index[int]" ),
1537+ pd .Index ,
1538+ np .integer ,
15351539 )
15361540
15371541 mi = pd .MultiIndex .from_arrays ([[1 , 2 , 3 ], [4 , 5 , 6 ]], names = ["elk" , "owl" ])
15381542 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 )
1543+ check (
1544+ assert_type (mi .set_names ({"elk" : "beluga" , "owl" : "pig" }), pd .MultiIndex ),
1545+ pd .MultiIndex ,
1546+ )
1547+ check (
1548+ assert_type (pd .Index ([(1 ,)]).set_names (1 ), pd .MultiIndex ), pd .MultiIndex , tuple
1549+ )
0 commit comments