conftest.py 881 B

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. import datetime
  2. import pytest
  3. from pandas._libs.tslibs import Timestamp
  4. from pandas._libs.tslibs.offsets import MonthOffset
  5. from pandas.tseries import offsets
  6. @pytest.fixture(
  7. params=[
  8. getattr(offsets, o) for o in offsets.__all__ if o not in ("Tick", "BaseOffset")
  9. ]
  10. )
  11. def offset_types(request):
  12. """
  13. Fixture for all the datetime offsets available for a time series.
  14. """
  15. return request.param
  16. @pytest.fixture(
  17. params=[
  18. getattr(offsets, o)
  19. for o in offsets.__all__
  20. if issubclass(getattr(offsets, o), MonthOffset) and o != "MonthOffset"
  21. ]
  22. )
  23. def month_classes(request):
  24. """
  25. Fixture for month based datetime offsets available for a time series.
  26. """
  27. return request.param
  28. @pytest.fixture
  29. def dt():
  30. """
  31. Fixture for common Timestamp.
  32. """
  33. return Timestamp(datetime.datetime(2008, 1, 2))