Skip to content

Commit 041df3f

Browse files
committed
move and simplify test
1 parent 7852918 commit 041df3f

File tree

2 files changed

+12
-14
lines changed

2 files changed

+12
-14
lines changed

pandas/tests/window/test_moments.py

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1891,20 +1891,6 @@ def test_rolling_corr_with_zero_variance(self, window):
18911891

18921892
assert s.rolling(window=window).corr(other=other).isna().all()
18931893

1894-
def test_rolling_max_datetimeindex(self):
1895-
# GH 21096
1896-
n = 10
1897-
index = pd.date_range(start="2018-1-1 01:00:00", freq="1min", periods=n)
1898-
s = Series(data=0, index=index)
1899-
1900-
s.iloc[1] = np.nan
1901-
s.iloc[-1] = 2
1902-
maxes = s.rolling(window=f"{n}min").max()
1903-
result = maxes.value_counts(dropna=False)
1904-
expected = Series(data={0.0: n - 1, 2.0: 1})
1905-
1906-
tm.assert_series_equal(result, expected)
1907-
19081894
def _check_pairwise_moment(self, dispatch, name, **kwargs):
19091895
def get_result(obj, obj2=None):
19101896
return getattr(getattr(obj, dispatch)(**kwargs), name)(obj2)

pandas/tests/window/test_timeseries_window.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -535,6 +535,18 @@ def test_ragged_max(self):
535535
expected["B"] = [0.0, 1, 2, 3, 4]
536536
tm.assert_frame_equal(result, expected)
537537

538+
def test_minutes_freq_max(self):
539+
# GH 21096
540+
n = 10
541+
index = date_range(start="2018-1-1 01:00:00", freq="1min", periods=n)
542+
s = Series(data=0, index=index)
543+
s.iloc[1] = np.nan
544+
s.iloc[-1] = 2
545+
result = s.rolling(window=f"{n}min").max()
546+
expected = Series(data=[0] * (n - 1) + [2.0], index=index)
547+
548+
tm.assert_series_equal(result, expected)
549+
538550
def test_ragged_apply(self, raw):
539551

540552
df = self.ragged

0 commit comments

Comments
 (0)