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
| import pytest
|
| from pandas._libs.tslibs.parsing import get_rule_month
|
| from pandas.tseries import offsets
|
|
| @pytest.mark.parametrize(
| "obj,expected",
| [
| ("W", "DEC"),
| (offsets.Week().freqstr, "DEC"),
| ("D", "DEC"),
| (offsets.Day().freqstr, "DEC"),
| ("Q", "DEC"),
| (offsets.QuarterEnd(startingMonth=12).freqstr, "DEC"),
| ("Q-JAN", "JAN"),
| (offsets.QuarterEnd(startingMonth=1).freqstr, "JAN"),
| ("A-DEC", "DEC"),
| ("Y-DEC", "DEC"),
| (offsets.YearEnd().freqstr, "DEC"),
| ("A-MAY", "MAY"),
| ("Y-MAY", "MAY"),
| (offsets.YearEnd(month=5).freqstr, "MAY"),
| ],
| )
| def test_get_rule_month(obj, expected):
| result = get_rule_month(obj)
| assert result == expected
|
|
