test_frequencies.py 821 B

1234567891011121314151617181920212223242526272829
  1. import pytest
  2. from pandas._libs.tslibs import offsets
  3. from pandas.tseries.frequencies import (
  4. is_subperiod,
  5. is_superperiod,
  6. )
  7. @pytest.mark.parametrize(
  8. "p1,p2,expected",
  9. [
  10. # Input validation.
  11. (offsets.MonthEnd(), None, False),
  12. (offsets.YearEnd(), None, False),
  13. (None, offsets.YearEnd(), False),
  14. (None, offsets.MonthEnd(), False),
  15. (None, None, False),
  16. (offsets.YearEnd(), offsets.MonthEnd(), True),
  17. (offsets.Hour(), offsets.Minute(), True),
  18. (offsets.Second(), offsets.Milli(), True),
  19. (offsets.Milli(), offsets.Micro(), True),
  20. (offsets.Micro(), offsets.Nano(), True),
  21. ],
  22. )
  23. def test_super_sub_symmetry(p1, p2, expected):
  24. assert is_superperiod(p1, p2) is expected
  25. assert is_subperiod(p2, p1) is expected