test_values.py 741 B

1234567891011121314151617181920212223242526272829
  1. import numpy as np
  2. import pytest
  3. from pandas import (
  4. IntervalIndex,
  5. Series,
  6. period_range,
  7. )
  8. import pandas._testing as tm
  9. class TestValues:
  10. @pytest.mark.parametrize(
  11. "data",
  12. [
  13. period_range("2000", periods=4),
  14. IntervalIndex.from_breaks([1, 2, 3, 4]),
  15. ],
  16. )
  17. def test_values_object_extension_dtypes(self, data):
  18. # https://github.com/pandas-dev/pandas/issues/23995
  19. result = Series(data).values
  20. expected = np.array(data.astype(object))
  21. tm.assert_numpy_array_equal(result, expected)
  22. def test_values(self, datetime_series):
  23. tm.assert_almost_equal(
  24. datetime_series.values, datetime_series, check_dtype=False
  25. )