test_resolution.py 641 B

123456789101112131415161718192021222324
  1. import numpy as np
  2. import pytz
  3. from pandas._libs.tslibs import (
  4. Resolution,
  5. get_resolution,
  6. )
  7. from pandas._libs.tslibs.dtypes import NpyDatetimeUnit
  8. def test_get_resolution_nano():
  9. # don't return the fallback RESO_DAY
  10. arr = np.array([1], dtype=np.int64)
  11. res = get_resolution(arr)
  12. assert res == Resolution.RESO_NS
  13. def test_get_resolution_non_nano_data():
  14. arr = np.array([1], dtype=np.int64)
  15. res = get_resolution(arr, None, NpyDatetimeUnit.NPY_FR_us.value)
  16. assert res == Resolution.RESO_US
  17. res = get_resolution(arr, pytz.UTC, NpyDatetimeUnit.NPY_FR_us.value)
  18. assert res == Resolution.RESO_US