Skip to content

Commit 860ff03

Browse files
authored
TST: move easter to own file and parameterize (#27085) (#42990)
1 parent 50fdd86 commit 860ff03

File tree

2 files changed

+36
-17
lines changed

2 files changed

+36
-17
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
"""
2+
Tests for the following offsets:
3+
- Easter
4+
"""
5+
from __future__ import annotations
6+
7+
from datetime import datetime
8+
9+
import pytest
10+
11+
from pandas.tests.tseries.offsets.common import (
12+
Base,
13+
assert_offset_equal,
14+
)
15+
16+
from pandas.tseries.offsets import Easter
17+
18+
19+
class TestEaster(Base):
20+
@pytest.mark.parametrize(
21+
"offset,date,expected",
22+
[
23+
(Easter(), datetime(2010, 1, 1), datetime(2010, 4, 4)),
24+
(Easter(), datetime(2010, 4, 5), datetime(2011, 4, 24)),
25+
(Easter(2), datetime(2010, 1, 1), datetime(2011, 4, 24)),
26+
(Easter(), datetime(2010, 4, 4), datetime(2011, 4, 24)),
27+
(Easter(2), datetime(2010, 4, 4), datetime(2012, 4, 8)),
28+
(-Easter(), datetime(2011, 1, 1), datetime(2010, 4, 4)),
29+
(-Easter(), datetime(2010, 4, 5), datetime(2010, 4, 4)),
30+
(-Easter(2), datetime(2011, 1, 1), datetime(2009, 4, 12)),
31+
(-Easter(), datetime(2010, 4, 4), datetime(2009, 4, 12)),
32+
(-Easter(2), datetime(2010, 4, 4), datetime(2008, 3, 23)),
33+
],
34+
)
35+
def test_offset(self, offset, date, expected):
36+
assert_offset_equal(offset, date, expected)

pandas/tests/tseries/offsets/test_offsets.py

-17
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@
3636
from pandas.tests.tseries.offsets.common import (
3737
Base,
3838
WeekDay,
39-
assert_offset_equal,
4039
)
4140

4241
import pandas.tseries.offsets as offsets
@@ -564,22 +563,6 @@ def test_eq(self):
564563
assert offset1 != offset2
565564

566565

567-
def test_Easter():
568-
assert_offset_equal(Easter(), datetime(2010, 1, 1), datetime(2010, 4, 4))
569-
assert_offset_equal(Easter(), datetime(2010, 4, 5), datetime(2011, 4, 24))
570-
assert_offset_equal(Easter(2), datetime(2010, 1, 1), datetime(2011, 4, 24))
571-
572-
assert_offset_equal(Easter(), datetime(2010, 4, 4), datetime(2011, 4, 24))
573-
assert_offset_equal(Easter(2), datetime(2010, 4, 4), datetime(2012, 4, 8))
574-
575-
assert_offset_equal(-Easter(), datetime(2011, 1, 1), datetime(2010, 4, 4))
576-
assert_offset_equal(-Easter(), datetime(2010, 4, 5), datetime(2010, 4, 4))
577-
assert_offset_equal(-Easter(2), datetime(2011, 1, 1), datetime(2009, 4, 12))
578-
579-
assert_offset_equal(-Easter(), datetime(2010, 4, 4), datetime(2009, 4, 12))
580-
assert_offset_equal(-Easter(2), datetime(2010, 4, 4), datetime(2008, 3, 23))
581-
582-
583566
class TestOffsetNames:
584567
def test_get_offset_name(self):
585568
assert BDay().freqstr == "B"

0 commit comments

Comments
 (0)