Skip to content

Commit c94cbd5

Browse files
committed
TST: use new class in test_rolling_forward_window
1 parent fff110c commit c94cbd5

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

pandas/tests/window/test_base_indexer.py

+2-11
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
from pandas import DataFrame, Series
55
import pandas._testing as tm
6-
from pandas.api.indexers import BaseIndexer
6+
from pandas.api.indexers import BaseIndexer, FixedForwardWindowIndexer
77
from pandas.core.window.indexers import ExpandingIndexer
88

99

@@ -105,19 +105,10 @@ def get_window_bounds(self, num_values, min_periods, center, closed):
105105
)
106106
def test_rolling_forward_window(constructor, func, alt_func, expected):
107107
# GH 32865
108-
class ForwardIndexer(BaseIndexer):
109-
def get_window_bounds(self, num_values, min_periods, center, closed):
110-
start = np.arange(num_values, dtype="int64")
111-
end_s = start[: -self.window_size] + self.window_size
112-
end_e = np.full(self.window_size, num_values, dtype="int64")
113-
end = np.concatenate([end_s, end_e])
114-
115-
return start, end
116-
117108
values = np.arange(10)
118109
values[5] = 100.0
119110

120-
indexer = ForwardIndexer(window_size=3)
111+
indexer = FixedForwardWindowIndexer(window_size=3)
121112
rolling = constructor(values).rolling(window=indexer, min_periods=2)
122113
result = getattr(rolling, func)()
123114
expected = constructor(expected)

0 commit comments

Comments
 (0)