309 lines
12 KiB
Python
309 lines
12 KiB
Python
|
"""
|
||
|
|
Tests for the following offsets:
|
||
|
|
- BQuarterBegin
|
||
|
|
- BQuarterEnd
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas.tests.tseries.offsets.common import (
|
||
|
|
assert_is_on_offset,
|
||
|
|
assert_offset_equal,
|
||
|
|
)
|
||
|
|
|
||
|
|
from pandas.tseries.offsets import (
|
||
|
|
BQuarterBegin,
|
||
|
|
BQuarterEnd,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_quarterly_dont_normalize():
|
||
|
|
date = datetime(2012, 3, 31, 5, 30)
|
||
|
|
|
||
|
|
offsets = (BQuarterEnd, BQuarterBegin)
|
||
|
|
|
||
|
|
for klass in offsets:
|
||
|
|
result = date + klass()
|
||
|
|
assert result.time() == date.time()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("offset", [BQuarterBegin(), BQuarterEnd()])
|
||
|
|
def test_on_offset(offset):
|
||
|
|
dates = [
|
||
|
|
datetime(2016, m, d)
|
||
|
|
for m in [10, 11, 12]
|
||
|
|
for d in [1, 2, 3, 28, 29, 30, 31]
|
||
|
|
if not (m == 11 and d == 31)
|
||
|
|
]
|
||
|
|
for date in dates:
|
||
|
|
res = offset.is_on_offset(date)
|
||
|
|
slow_version = date == (date + offset) - offset
|
||
|
|
assert res == slow_version
|
||
|
|
|
||
|
|
|
||
|
|
class TestBQuarterBegin:
|
||
|
|
def test_repr(self):
|
||
|
|
expected = "<BusinessQuarterBegin: startingMonth=3>"
|
||
|
|
assert repr(BQuarterBegin()) == expected
|
||
|
|
expected = "<BusinessQuarterBegin: startingMonth=3>"
|
||
|
|
assert repr(BQuarterBegin(startingMonth=3)) == expected
|
||
|
|
expected = "<BusinessQuarterBegin: startingMonth=1>"
|
||
|
|
assert repr(BQuarterBegin(startingMonth=1)) == expected
|
||
|
|
|
||
|
|
def test_is_anchored(self):
|
||
|
|
assert BQuarterBegin(startingMonth=1).is_anchored()
|
||
|
|
assert BQuarterBegin().is_anchored()
|
||
|
|
assert not BQuarterBegin(2, startingMonth=1).is_anchored()
|
||
|
|
|
||
|
|
def test_offset_corner_case(self):
|
||
|
|
# corner
|
||
|
|
offset = BQuarterBegin(n=-1, startingMonth=1)
|
||
|
|
assert datetime(2007, 4, 3) + offset == datetime(2007, 4, 2)
|
||
|
|
|
||
|
|
offset_cases = []
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterBegin(startingMonth=1),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 7, 1),
|
||
|
|
datetime(2007, 3, 15): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 2, 28): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 1, 1): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 4, 15): datetime(2007, 7, 2),
|
||
|
|
datetime(2007, 7, 1): datetime(2007, 7, 2),
|
||
|
|
datetime(2007, 4, 1): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 4, 2): datetime(2007, 7, 2),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 7, 1),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterBegin(startingMonth=2),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 2, 1),
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 2, 1),
|
||
|
|
datetime(2008, 1, 15): datetime(2008, 2, 1),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 5, 1),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 5, 1),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 5, 1),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 5, 1),
|
||
|
|
datetime(2008, 8, 15): datetime(2008, 11, 3),
|
||
|
|
datetime(2008, 9, 15): datetime(2008, 11, 3),
|
||
|
|
datetime(2008, 11, 1): datetime(2008, 11, 3),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 5, 1),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterBegin(startingMonth=1, n=0),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 1, 1),
|
||
|
|
datetime(2007, 12, 31): datetime(2008, 1, 1),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 1, 15): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 2, 27): datetime(2008, 4, 1),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 4, 1),
|
||
|
|
datetime(2007, 4, 1): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 4, 2): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 7, 1): datetime(2007, 7, 2),
|
||
|
|
datetime(2007, 4, 15): datetime(2007, 7, 2),
|
||
|
|
datetime(2007, 7, 2): datetime(2007, 7, 2),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterBegin(startingMonth=1, n=-1),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2007, 10, 1),
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 1, 1),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 1, 1),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 1, 1),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 1, 1),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 1, 1),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 4, 1),
|
||
|
|
datetime(2007, 7, 3): datetime(2007, 7, 2),
|
||
|
|
datetime(2007, 4, 3): datetime(2007, 4, 2),
|
||
|
|
datetime(2007, 7, 2): datetime(2007, 4, 2),
|
||
|
|
datetime(2008, 4, 1): datetime(2008, 1, 1),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterBegin(startingMonth=1, n=2),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 7, 1),
|
||
|
|
datetime(2008, 1, 15): datetime(2008, 7, 1),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 7, 1),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 7, 1),
|
||
|
|
datetime(2007, 3, 31): datetime(2007, 7, 2),
|
||
|
|
datetime(2007, 4, 15): datetime(2007, 10, 1),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 10, 1),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("case", offset_cases)
|
||
|
|
def test_offset(self, case):
|
||
|
|
offset, cases = case
|
||
|
|
for base, expected in cases.items():
|
||
|
|
assert_offset_equal(offset, base, expected)
|
||
|
|
|
||
|
|
|
||
|
|
class TestBQuarterEnd:
|
||
|
|
def test_repr(self):
|
||
|
|
expected = "<BusinessQuarterEnd: startingMonth=3>"
|
||
|
|
assert repr(BQuarterEnd()) == expected
|
||
|
|
expected = "<BusinessQuarterEnd: startingMonth=3>"
|
||
|
|
assert repr(BQuarterEnd(startingMonth=3)) == expected
|
||
|
|
expected = "<BusinessQuarterEnd: startingMonth=1>"
|
||
|
|
assert repr(BQuarterEnd(startingMonth=1)) == expected
|
||
|
|
|
||
|
|
def test_is_anchored(self):
|
||
|
|
assert BQuarterEnd(startingMonth=1).is_anchored()
|
||
|
|
assert BQuarterEnd().is_anchored()
|
||
|
|
assert not BQuarterEnd(2, startingMonth=1).is_anchored()
|
||
|
|
|
||
|
|
def test_offset_corner_case(self):
|
||
|
|
# corner
|
||
|
|
offset = BQuarterEnd(n=-1, startingMonth=1)
|
||
|
|
assert datetime(2010, 1, 31) + offset == datetime(2010, 1, 29)
|
||
|
|
|
||
|
|
offset_cases = []
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterEnd(startingMonth=1),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 7, 31),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterEnd(startingMonth=2),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 2, 29),
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 2, 29),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 2, 29),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 5, 30),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 5, 30),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 5, 30),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 5, 30),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 5, 30),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterEnd(startingMonth=1, n=0),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 4, 30),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 4, 30),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterEnd(startingMonth=1, n=-1),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 1): datetime(2007, 10, 31),
|
||
|
|
datetime(2008, 1, 31): datetime(2007, 10, 31),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 1, 31),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 1, 31),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
offset_cases.append(
|
||
|
|
(
|
||
|
|
BQuarterEnd(startingMonth=1, n=2),
|
||
|
|
{
|
||
|
|
datetime(2008, 1, 31): datetime(2008, 7, 31),
|
||
|
|
datetime(2008, 2, 15): datetime(2008, 7, 31),
|
||
|
|
datetime(2008, 2, 29): datetime(2008, 7, 31),
|
||
|
|
datetime(2008, 3, 15): datetime(2008, 7, 31),
|
||
|
|
datetime(2008, 3, 31): datetime(2008, 7, 31),
|
||
|
|
datetime(2008, 4, 15): datetime(2008, 7, 31),
|
||
|
|
datetime(2008, 4, 30): datetime(2008, 10, 31),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("case", offset_cases)
|
||
|
|
def test_offset(self, case):
|
||
|
|
offset, cases = case
|
||
|
|
for base, expected in cases.items():
|
||
|
|
assert_offset_equal(offset, base, expected)
|
||
|
|
|
||
|
|
on_offset_cases = [
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2008, 1, 31), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2007, 12, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2008, 2, 29), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2007, 3, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2007, 3, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2008, 4, 30), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2008, 5, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2007, 6, 29), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=1), datetime(2007, 6, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2008, 1, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2007, 12, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2008, 2, 29), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2007, 3, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2007, 3, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2008, 4, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2008, 5, 30), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2007, 6, 29), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=2), datetime(2007, 6, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2008, 1, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2007, 12, 31), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2008, 2, 29), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2007, 3, 30), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2007, 3, 31), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2008, 4, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2008, 5, 30), False),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2007, 6, 29), True),
|
||
|
|
(BQuarterEnd(1, startingMonth=3), datetime(2007, 6, 30), False),
|
||
|
|
]
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("case", on_offset_cases)
|
||
|
|
def test_is_on_offset(self, case):
|
||
|
|
offset, dt, expected = case
|
||
|
|
assert_is_on_offset(offset, dt, expected)
|