1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
| """
| Tests for offset behavior with indices.
| """
| import pytest
|
| from pandas import (
| Series,
| date_range,
| )
|
| from pandas.tseries.offsets import (
| BMonthBegin,
| BMonthEnd,
| BQuarterBegin,
| BQuarterEnd,
| BYearBegin,
| BYearEnd,
| MonthBegin,
| MonthEnd,
| QuarterBegin,
| QuarterEnd,
| YearBegin,
| YearEnd,
| )
|
|
| @pytest.mark.parametrize("n", [-2, 1])
| @pytest.mark.parametrize(
| "cls",
| [
| MonthBegin,
| MonthEnd,
| BMonthBegin,
| BMonthEnd,
| QuarterBegin,
| QuarterEnd,
| BQuarterBegin,
| BQuarterEnd,
| YearBegin,
| YearEnd,
| BYearBegin,
| BYearEnd,
| ],
| )
| def test_apply_index(cls, n):
| offset = cls(n=n)
| rng = date_range(start="1/1/2000", periods=100000, freq="T")
| ser = Series(rng)
|
| res = rng + offset
| assert res.freq is None # not retained
| assert res[0] == rng[0] + offset
| assert res[-1] == rng[-1] + offset
| res2 = ser + offset
| # apply_index is only for indexes, not series, so no res2_v2
| assert res2.iloc[0] == ser.iloc[0] + offset
| assert res2.iloc[-1] == ser.iloc[-1] + offset
|
|
