test_easter.py 1.1 KB

123456789101112131415161718192021222324252627282930313233
  1. """
  2. Tests for the following offsets:
  3. - Easter
  4. """
  5. from __future__ import annotations
  6. from datetime import datetime
  7. import pytest
  8. from pandas.tests.tseries.offsets.common import assert_offset_equal
  9. from pandas.tseries.offsets import Easter
  10. class TestEaster:
  11. @pytest.mark.parametrize(
  12. "offset,date,expected",
  13. [
  14. (Easter(), datetime(2010, 1, 1), datetime(2010, 4, 4)),
  15. (Easter(), datetime(2010, 4, 5), datetime(2011, 4, 24)),
  16. (Easter(2), datetime(2010, 1, 1), datetime(2011, 4, 24)),
  17. (Easter(), datetime(2010, 4, 4), datetime(2011, 4, 24)),
  18. (Easter(2), datetime(2010, 4, 4), datetime(2012, 4, 8)),
  19. (-Easter(), datetime(2011, 1, 1), datetime(2010, 4, 4)),
  20. (-Easter(), datetime(2010, 4, 5), datetime(2010, 4, 4)),
  21. (-Easter(2), datetime(2011, 1, 1), datetime(2009, 4, 12)),
  22. (-Easter(), datetime(2010, 4, 4), datetime(2009, 4, 12)),
  23. (-Easter(2), datetime(2010, 4, 4), datetime(2008, 3, 23)),
  24. ],
  25. )
  26. def test_offset(self, offset, date, expected):
  27. assert_offset_equal(offset, date, expected)