| 123456789101112131415161718192021222324252627 |
- 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"),
- ("Y-DEC", "DEC"),
- (offsets.YearEnd().freqstr, "DEC"),
- ("Y-MAY", "MAY"),
- (offsets.YearEnd(month=5).freqstr, "MAY"),
- ],
- )
- def test_get_rule_month(obj, expected):
- result = get_rule_month(obj)
- assert result == expected
|
