Skip to content

Commit 1cf968c

Browse files
kotamatsuokarhshadrach
authored andcommitted
TST: add pattern for nearest (pandas-dev#33939)
1 parent f132c90 commit 1cf968c

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

pandas/tests/resample/test_datetime_index.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -540,11 +540,14 @@ def test_upsample_with_limit():
540540
tm.assert_series_equal(result, expected)
541541

542542

543-
def test_nearest_upsample_with_limit():
544-
rng = date_range("1/1/2000", periods=3, freq="5t")
543+
@pytest.mark.parametrize("freq", ["Y", "10M", "5D", "10H", "5Min", "10S"])
544+
@pytest.mark.parametrize("rule", ["Y", "3M", "15D", "30H", "15Min", "30S"])
545+
def test_nearest_upsample_with_limit(tz_aware_fixture, freq, rule):
546+
# GH 33939
547+
rng = date_range("1/1/2000", periods=3, freq=freq, tz=tz_aware_fixture)
545548
ts = Series(np.random.randn(len(rng)), rng)
546549

547-
result = ts.resample("t").nearest(limit=2)
550+
result = ts.resample(rule).nearest(limit=2)
548551
expected = ts.reindex(result.index, method="nearest", limit=2)
549552
tm.assert_series_equal(result, expected)
550553

0 commit comments

Comments
 (0)