(before, after)
| 426 | |
| 427 | @pytest.mark.parametrize("before, after", [("6s", "6s"), ("2s", "2s"), ("6s", "2s")]) |
| 428 | def test_time_rolling(before, after): |
| 429 | window = before |
| 430 | expected = dts.compute().rolling(window).count() |
| 431 | |
| 432 | # String timedelta |
| 433 | result = dts.map_overlap(lambda x: x.rolling(window).count(), before, after) |
| 434 | assert_eq(result, expected) |
| 435 | |
| 436 | # Timedelta |
| 437 | before = pd.Timedelta(before) |
| 438 | after = pd.Timedelta(after) |
| 439 | result = dts.map_overlap(lambda x: x.rolling(window).count(), before, after) |
| 440 | assert_eq(result, expected) |
| 441 | |
| 442 | |
| 443 | def test_rolling_agg_aggregate(): |
nothing calls this directly
no test coverage detected