Skip to content

Commit 244d985

Browse files
committed
type names too
1 parent 4487a36 commit 244d985

File tree

2 files changed

+19
-5
lines changed

2 files changed

+19
-5
lines changed

pandas-stubs/core/indexes/multi.pyi

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,21 +30,29 @@ class MultiIndex(Index[Any]):
3030
levels=...,
3131
codes=...,
3232
sortorder=...,
33-
names=...,
33+
names: SequenceNotStr[Hashable] = ...,
3434
dtype=...,
3535
copy=...,
3636
name: SequenceNotStr[Hashable] = ...,
3737
verify_integrity: bool = ...,
3838
_set_identity: bool = ...,
3939
) -> None: ...
4040
@classmethod
41-
def from_arrays(cls, arrays, sortorder=..., names=...) -> Self: ...
41+
def from_arrays(
42+
cls, arrays, sortorder=..., names: SequenceNotStr[Hashable] = ...
43+
) -> Self: ...
4244
@classmethod
43-
def from_tuples(cls, tuples, sortorder=..., names=...) -> Self: ...
45+
def from_tuples(
46+
cls, tuples, sortorder=..., names: SequenceNotStr[Hashable] = ...
47+
) -> Self: ...
4448
@classmethod
45-
def from_product(cls, iterables, sortorder=..., names=...) -> Self: ...
49+
def from_product(
50+
cls, iterables, sortorder=..., names: SequenceNotStr[Hashable] = ...
51+
) -> Self: ...
4652
@classmethod
47-
def from_frame(cls, df, sortorder=..., names=...) -> Self: ...
53+
def from_frame(
54+
cls, df, sortorder=..., names: SequenceNotStr[Hashable] = ...
55+
) -> Self: ...
4856
@property
4957
def shape(self): ...
5058
@property # Should be read-only

tests/test_indexes.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,12 @@ def test_multiindex_initialization() -> None:
7171
),
7272
pd.MultiIndex,
7373
)
74+
check(
75+
assert_type(
76+
pd.MultiIndex([[1], [4]], codes=[[0], [0]], names=["a", "b"]), pd.MultiIndex
77+
),
78+
pd.MultiIndex,
79+
)
7480

7581

7682
def test_index_tolist() -> None:

0 commit comments

Comments
 (0)