30 lines
769 B
Python
30 lines
769 B
Python
|
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
|