| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556 |
- import pytest
- from pandas import (
- DatetimeIndex,
- Series,
- Timestamp,
- date_range,
- )
- import pandas._testing as tm
- @pytest.mark.parametrize(
- "cons",
- [
- lambda x: DatetimeIndex(x),
- lambda x: DatetimeIndex(DatetimeIndex(x)),
- ],
- )
- def test_datetimeindex(using_copy_on_write, cons):
- dt = date_range("2019-12-31", periods=3, freq="D")
- ser = Series(dt)
- idx = cons(ser)
- expected = idx.copy(deep=True)
- ser.iloc[0] = Timestamp("2020-12-31")
- if using_copy_on_write:
- tm.assert_index_equal(idx, expected)
- def test_datetimeindex_tz_convert(using_copy_on_write):
- dt = date_range("2019-12-31", periods=3, freq="D", tz="Europe/Berlin")
- ser = Series(dt)
- idx = DatetimeIndex(ser).tz_convert("US/Eastern")
- expected = idx.copy(deep=True)
- ser.iloc[0] = Timestamp("2020-12-31", tz="Europe/Berlin")
- if using_copy_on_write:
- tm.assert_index_equal(idx, expected)
- def test_datetimeindex_tz_localize(using_copy_on_write):
- dt = date_range("2019-12-31", periods=3, freq="D")
- ser = Series(dt)
- idx = DatetimeIndex(ser).tz_localize("Europe/Berlin")
- expected = idx.copy(deep=True)
- ser.iloc[0] = Timestamp("2020-12-31")
- if using_copy_on_write:
- tm.assert_index_equal(idx, expected)
- def test_datetimeindex_isocalendar(using_copy_on_write):
- dt = date_range("2019-12-31", periods=3, freq="D")
- ser = Series(dt)
- df = DatetimeIndex(ser).isocalendar()
- expected = df.index.copy(deep=True)
- ser.iloc[0] = Timestamp("2020-12-31")
- if using_copy_on_write:
- tm.assert_index_equal(df.index, expected)
|
