conftest.py 706 B

123456789101112131415161718192021222324252627282930313233
  1. import pytest
  2. # The various methods we support
  3. downsample_methods = [
  4. "min",
  5. "max",
  6. "first",
  7. "last",
  8. "sum",
  9. "mean",
  10. "sem",
  11. "median",
  12. "prod",
  13. "var",
  14. "std",
  15. "ohlc",
  16. "quantile",
  17. ]
  18. upsample_methods = ["count", "size"]
  19. series_methods = ["nunique"]
  20. resample_methods = downsample_methods + upsample_methods + series_methods
  21. @pytest.fixture(params=downsample_methods)
  22. def downsample_method(request):
  23. """Fixture for parametrization of Grouper downsample methods."""
  24. return request.param
  25. @pytest.fixture(params=resample_methods)
  26. def resample_method(request):
  27. """Fixture for parametrization of Grouper resample methods."""
  28. return request.param