|
1 | 1 | from __future__ import annotations
|
2 | 2 |
|
| 3 | +from collections.abc import Hashable |
3 | 4 | import datetime as dt
|
4 | 5 | from typing import (
|
5 | 6 | TYPE_CHECKING,
|
@@ -1439,15 +1440,17 @@ def test_index_naming() -> None:
|
1439 | 1440 | """
|
1440 | 1441 | Test index names type both for the getter and the setter.
|
1441 | 1442 | The names of an index should be settable with a sequence (not str) and names
|
1442 |
| - property is a list[str | None] (FrozenList). |
| 1443 | + property is a list[Hashable | None] (FrozenList). |
1443 | 1444 | """
|
1444 | 1445 | df = pd.DataFrame({"a": ["a", "b", "c"], "i": [10, 11, 12]})
|
1445 | 1446 |
|
1446 | 1447 | df.index.names = ["idx"]
|
1447 |
| - check(assert_type(df.index.names, list[str | None]), list) |
| 1448 | + check(assert_type(df.index.names, list[Hashable | None]), list) |
| 1449 | + df.index.names = [3] |
| 1450 | + check(assert_type(df.index.names, list[Hashable | None]), list) |
1448 | 1451 | df.index.names = ("idx2",)
|
1449 |
| - check(assert_type(df.index.names, list[str | None]), list) |
| 1452 | + check(assert_type(df.index.names, list[Hashable | None]), list) |
1450 | 1453 | df.index.names = [None]
|
1451 |
| - check(assert_type(df.index.names, list[str | None]), list) |
| 1454 | + check(assert_type(df.index.names, list[Hashable | None]), list) |
1452 | 1455 | df.index.names = (None,)
|
1453 |
| - check(assert_type(df.index.names, list[str | None]), list) |
| 1456 | + check(assert_type(df.index.names, list[Hashable | None]), list) |
0 commit comments