GH1285 Allow range in pd.MultiIndex.from_product (#1296)

Author: loicdiridollou

pandas-stubs/core/indexes/multi.pyi

@@ -57,7 +57,7 @@ class MultiIndex(Index):
     @classmethod
     def from_product(
         cls,
-        iterables: Sequence[SequenceNotStr[Hashable] | pd.Series | pd.Index],
+        iterables: Sequence[SequenceNotStr[Hashable] | pd.Series | pd.Index | range],
         sortorder: int | None = ...,
         names: SequenceNotStr[Hashable] = ...,
     ) -> Self: ...

tests/test_indexes.py

@@ -1384,3 +1384,16 @@ def test_index_infer_objects() -> None:
     df = pd.DataFrame({"A": ["a", 1, 2, 3]})
     idx = df.set_index("A").index[1:]
     check(assert_type(idx.infer_objects(), pd.Index), pd.Index)
+
+
+def test_multiindex_range() -> None:
+    """Test using range in `MultiIndex.from_product` GH1285."""
+    midx = pd.MultiIndex.from_product(
+        [range(3), range(5)],
+    )
+    check(assert_type(midx, pd.MultiIndex), pd.MultiIndex)
+
+    midx_mixed_types = pd.MultiIndex.from_product(
+        [range(3), pd.Series([2, 3, 5])],
+    )
+    check(assert_type(midx_mixed_types, pd.MultiIndex), pd.MultiIndex)