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
| import pytest
|
| from pandas import (
| DatetimeIndex,
| date_range,
| )
| import pandas._testing as tm
|
|
| @pytest.mark.parametrize("tz", [None, "Asia/Shanghai", "Europe/Berlin"])
| @pytest.mark.parametrize("name", [None, "my_dti"])
| @pytest.mark.parametrize("unit", ["ns", "us", "ms", "s"])
| def test_dti_snap(name, tz, unit):
| dti = DatetimeIndex(
| [
| "1/1/2002",
| "1/2/2002",
| "1/3/2002",
| "1/4/2002",
| "1/5/2002",
| "1/6/2002",
| "1/7/2002",
| ],
| name=name,
| tz=tz,
| freq="D",
| )
| dti = dti.as_unit(unit)
|
| result = dti.snap(freq="W-MON")
| expected = date_range("12/31/2001", "1/7/2002", name=name, tz=tz, freq="w-mon")
| expected = expected.repeat([3, 4])
| expected = expected.as_unit(unit)
| tm.assert_index_equal(result, expected)
| assert result.tz == expected.tz
| assert result.freq is None
| assert expected.freq is None
|
| result = dti.snap(freq="B")
|
| expected = date_range("1/1/2002", "1/7/2002", name=name, tz=tz, freq="b")
| expected = expected.repeat([1, 1, 1, 2, 2])
| expected = expected.as_unit(unit)
| tm.assert_index_equal(result, expected)
| assert result.tz == expected.tz
| assert result.freq is None
| assert expected.freq is None
|
|
