12
12
from pandas .tests .tseries .offsets .common import assert_offset_equal
13
13
14
14
from pandas .tseries .offsets import Easter
15
+ from dateutil .easter import EASTER_WESTERN , EASTER_ORTHODOX
15
16
16
17
17
18
class TestEaster :
@@ -32,3 +33,39 @@ class TestEaster:
32
33
)
33
34
def test_offset (self , offset , date , expected ):
34
35
assert_offset_equal (offset , date , expected )
36
+
37
+ @pytest .mark .parametrize (
38
+ "offset,date,expected" ,
39
+ [
40
+ (Easter (method = EASTER_WESTERN ), datetime (2010 , 1 , 1 ), datetime (2010 , 4 , 4 )),
41
+ (Easter (method = EASTER_WESTERN ), datetime (2010 , 4 , 5 ), datetime (2011 , 4 , 24 )),
42
+ (Easter (2 , method = EASTER_WESTERN ), datetime (2010 , 1 , 1 ), datetime (2011 , 4 , 24 )),
43
+ (Easter (method = EASTER_WESTERN ), datetime (2010 , 4 , 4 ), datetime (2011 , 4 , 24 )),
44
+ (Easter (2 , method = EASTER_WESTERN ), datetime (2010 , 4 , 4 ), datetime (2012 , 4 , 8 )),
45
+ (- Easter (method = EASTER_WESTERN ), datetime (2011 , 1 , 1 ), datetime (2010 , 4 , 4 )),
46
+ (- Easter (method = EASTER_WESTERN ), datetime (2010 , 4 , 5 ), datetime (2010 , 4 , 4 )),
47
+ (- Easter (2 , method = EASTER_WESTERN ), datetime (2011 , 1 , 1 ), datetime (2009 , 4 , 12 )),
48
+ (- Easter (method = EASTER_WESTERN ), datetime (2010 , 4 , 4 ), datetime (2009 , 4 , 12 )),
49
+ (- Easter (2 , method = EASTER_WESTERN ), datetime (2010 , 4 , 4 ), datetime (2008 , 3 , 23 )),
50
+ ],
51
+ )
52
+ def test_western_easter_offset (self , offset , date , expected ):
53
+ assert_offset_equal (offset , date , expected )
54
+
55
+ @pytest .mark .parametrize (
56
+ "offset,date,expected" ,
57
+ [
58
+ (Easter (method = EASTER_ORTHODOX ), datetime (2010 , 1 , 1 ), datetime (2010 , 4 , 4 )),
59
+ (Easter (method = EASTER_ORTHODOX ), datetime (2010 , 4 , 5 ), datetime (2011 , 4 , 24 )),
60
+ (Easter (2 , method = EASTER_ORTHODOX ), datetime (2010 , 1 , 1 ), datetime (2011 , 4 , 24 )),
61
+ (Easter (method = EASTER_ORTHODOX ), datetime (2010 , 4 , 4 ), datetime (2011 , 4 , 24 )),
62
+ (Easter (2 , method = EASTER_ORTHODOX ), datetime (2010 , 4 , 4 ), datetime (2012 , 4 , 15 )),
63
+ (- Easter (method = EASTER_ORTHODOX ), datetime (2011 , 1 , 1 ), datetime (2010 , 4 , 4 )),
64
+ (- Easter (method = EASTER_ORTHODOX ), datetime (2010 , 4 , 5 ), datetime (2010 , 4 , 4 )),
65
+ (- Easter (2 , method = EASTER_ORTHODOX ), datetime (2011 , 1 , 1 ), datetime (2009 , 4 , 19 )),
66
+ (- Easter (method = EASTER_ORTHODOX ), datetime (2010 , 4 , 4 ), datetime (2009 , 4 , 19 )),
67
+ (- Easter (2 , method = EASTER_ORTHODOX ), datetime (2010 , 4 , 4 ), datetime (2008 , 4 , 27 )),
68
+ ],
69
+ )
70
+ def test_orthodox_easter_offset (self , offset , date , expected ):
71
+ assert_offset_equal (offset , date , expected )
0 commit comments