98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas._libs.tslibs import (
|
||
|
|
Period,
|
||
|
|
Resolution,
|
||
|
|
to_offset,
|
||
|
|
)
|
||
|
|
from pandas._libs.tslibs.dtypes import _attrname_to_abbrevs
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"freqstr,exp_freqstr",
|
||
|
|
[("D", "D"), ("W", "D"), ("M", "D"), ("S", "S"), ("T", "S"), ("H", "S")],
|
||
|
|
)
|
||
|
|
def test_get_to_timestamp_base(freqstr, exp_freqstr):
|
||
|
|
off = to_offset(freqstr)
|
||
|
|
per = Period._from_ordinal(1, off)
|
||
|
|
exp_code = to_offset(exp_freqstr)._period_dtype_code
|
||
|
|
|
||
|
|
result_code = per._dtype._get_to_timestamp_base()
|
||
|
|
assert result_code == exp_code
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"freqstr,expected",
|
||
|
|
[
|
||
|
|
("A", "year"),
|
||
|
|
("Q", "quarter"),
|
||
|
|
("M", "month"),
|
||
|
|
("D", "day"),
|
||
|
|
("H", "hour"),
|
||
|
|
("T", "minute"),
|
||
|
|
("S", "second"),
|
||
|
|
("L", "millisecond"),
|
||
|
|
("U", "microsecond"),
|
||
|
|
("N", "nanosecond"),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_get_attrname_from_abbrev(freqstr, expected):
|
||
|
|
assert Resolution.get_reso_from_freqstr(freqstr).attrname == expected
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("freq", ["D", "H", "T", "S", "L", "U", "N"])
|
||
|
|
def test_get_freq_roundtrip2(freq):
|
||
|
|
obj = Resolution.get_reso_from_freqstr(freq)
|
||
|
|
result = _attrname_to_abbrevs[obj.attrname]
|
||
|
|
assert freq == result
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"args,expected",
|
||
|
|
[
|
||
|
|
((1.5, "T"), (90, "S")),
|
||
|
|
((62.4, "T"), (3744, "S")),
|
||
|
|
((1.04, "H"), (3744, "S")),
|
||
|
|
((1, "D"), (1, "D")),
|
||
|
|
((0.342931, "H"), (1234551600, "U")),
|
||
|
|
((1.2345, "D"), (106660800, "L")),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_resolution_bumping(args, expected):
|
||
|
|
# see gh-14378
|
||
|
|
off = to_offset(str(args[0]) + args[1])
|
||
|
|
assert off.n == expected[0]
|
||
|
|
assert off._prefix == expected[1]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"args",
|
||
|
|
[
|
||
|
|
(0.5, "N"),
|
||
|
|
# Too much precision in the input can prevent.
|
||
|
|
(0.3429324798798269273987982, "H"),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_cat(args):
|
||
|
|
msg = "Invalid frequency"
|
||
|
|
|
||
|
|
with pytest.raises(ValueError, match=msg):
|
||
|
|
to_offset(str(args[0]) + args[1])
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"freqstr,expected",
|
||
|
|
[
|
||
|
|
("1H", "2021-01-01T09:00:00"),
|
||
|
|
("1D", "2021-01-02T08:00:00"),
|
||
|
|
("1W", "2021-01-03T08:00:00"),
|
||
|
|
("1M", "2021-01-31T08:00:00"),
|
||
|
|
("1Y", "2021-12-31T08:00:00"),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_compatibility(freqstr, expected):
|
||
|
|
ts_np = np.datetime64("2021-01-01T08:00:00.00")
|
||
|
|
do = to_offset(freqstr)
|
||
|
|
assert ts_np + do == np.datetime64(expected)
|