From ceec1827c5d92548f4ca5ff3b1b310ff72838add Mon Sep 17 00:00:00 2001 From: Marco Gorelli <33491632+MarcoGorelli@users.noreply.github.com> Date: Mon, 10 Feb 2025 17:37:19 +0000 Subject: [PATCH] type RangeIndex properties --- pandas-stubs/core/indexes/range.pyi | 9 ++++++--- tests/test_indexes.py | 7 +++++++ 2 files changed, 13 insertions(+), 3 deletions(-) diff --git a/pandas-stubs/core/indexes/range.pyi b/pandas-stubs/core/indexes/range.pyi index 2ed55f3d4..b7569fd23 100644 --- a/pandas-stubs/core/indexes/range.pyi +++ b/pandas-stubs/core/indexes/range.pyi @@ -24,9 +24,12 @@ class RangeIndex(Index[int]): @classmethod def from_range(cls, data, name=..., dtype=...): ... def __reduce__(self): ... - def start(self): ... - def stop(self): ... - def step(self): ... + @property + def start(self) -> int: ... + @property + def stop(self) -> int: ... + @property + def step(self) -> int: ... @property def nbytes(self) -> int: ... def memory_usage(self, deep: bool = ...) -> int: ... diff --git a/tests/test_indexes.py b/tests/test_indexes.py index e77371cfb..25285d954 100644 --- a/tests/test_indexes.py +++ b/tests/test_indexes.py @@ -252,6 +252,13 @@ def test_range_index_union(): ) +def test_range_index_start_stop_step(): + idx = pd.RangeIndex(3) + check(assert_type(idx.start, int), int) + check(assert_type(idx.stop, int), int) + check(assert_type(idx.step, int), int) + + def test_interval_range(): check( assert_type(pd.interval_range(0, 10), "pd.IntervalIndex[pd.Interval[int]]"),