659 lines
29 KiB
Python
659 lines
29 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Tests for Fiscal Year and Fiscal Quarter offset classes
|
||
|
|
"""
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from dateutil.relativedelta import relativedelta
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import pandas.util.testing as tm
|
||
|
|
|
||
|
|
from pandas import Timestamp
|
||
|
|
from pandas.tseries.frequencies import get_offset
|
||
|
|
from pandas._libs.tslibs.frequencies import _INVALID_FREQ_ERROR
|
||
|
|
from pandas.tseries.offsets import FY5253Quarter, FY5253
|
||
|
|
from pandas._libs.tslibs.offsets import WeekDay
|
||
|
|
|
||
|
|
from .common import assert_offset_equal, assert_onOffset
|
||
|
|
from .test_offsets import Base
|
||
|
|
|
||
|
|
|
||
|
|
def makeFY5253LastOfMonthQuarter(*args, **kwds):
|
||
|
|
return FY5253Quarter(*args, variation="last", **kwds)
|
||
|
|
|
||
|
|
|
||
|
|
def makeFY5253NearestEndMonthQuarter(*args, **kwds):
|
||
|
|
return FY5253Quarter(*args, variation="nearest", **kwds)
|
||
|
|
|
||
|
|
|
||
|
|
def makeFY5253NearestEndMonth(*args, **kwds):
|
||
|
|
return FY5253(*args, variation="nearest", **kwds)
|
||
|
|
|
||
|
|
|
||
|
|
def makeFY5253LastOfMonth(*args, **kwds):
|
||
|
|
return FY5253(*args, variation="last", **kwds)
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_offset_name():
|
||
|
|
assert (makeFY5253LastOfMonthQuarter(
|
||
|
|
weekday=1, startingMonth=3,
|
||
|
|
qtr_with_extra_week=4).freqstr == "REQ-L-MAR-TUE-4")
|
||
|
|
assert (makeFY5253NearestEndMonthQuarter(
|
||
|
|
weekday=1, startingMonth=3,
|
||
|
|
qtr_with_extra_week=3).freqstr == "REQ-N-MAR-TUE-3")
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_offset():
|
||
|
|
with tm.assert_raises_regex(ValueError, _INVALID_FREQ_ERROR):
|
||
|
|
get_offset('gibberish')
|
||
|
|
with tm.assert_raises_regex(ValueError, _INVALID_FREQ_ERROR):
|
||
|
|
get_offset('QS-JAN-B')
|
||
|
|
|
||
|
|
pairs = [
|
||
|
|
("RE-N-DEC-MON",
|
||
|
|
makeFY5253NearestEndMonth(weekday=0, startingMonth=12)),
|
||
|
|
("RE-L-DEC-TUE",
|
||
|
|
makeFY5253LastOfMonth(weekday=1, startingMonth=12)),
|
||
|
|
("REQ-L-MAR-TUE-4",
|
||
|
|
makeFY5253LastOfMonthQuarter(weekday=1,
|
||
|
|
startingMonth=3,
|
||
|
|
qtr_with_extra_week=4)),
|
||
|
|
("REQ-L-DEC-MON-3",
|
||
|
|
makeFY5253LastOfMonthQuarter(weekday=0,
|
||
|
|
startingMonth=12,
|
||
|
|
qtr_with_extra_week=3)),
|
||
|
|
("REQ-N-DEC-MON-3",
|
||
|
|
makeFY5253NearestEndMonthQuarter(weekday=0,
|
||
|
|
startingMonth=12,
|
||
|
|
qtr_with_extra_week=3))]
|
||
|
|
|
||
|
|
for name, expected in pairs:
|
||
|
|
offset = get_offset(name)
|
||
|
|
assert offset == expected, ("Expected %r to yield %r (actual: %r)" %
|
||
|
|
(name, expected, offset))
|
||
|
|
|
||
|
|
|
||
|
|
class TestFY5253LastOfMonth(Base):
|
||
|
|
offset_lom_sat_aug = makeFY5253LastOfMonth(1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT)
|
||
|
|
offset_lom_sat_sep = makeFY5253LastOfMonth(1, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT)
|
||
|
|
|
||
|
|
on_offset_cases = [
|
||
|
|
# From Wikipedia (see:
|
||
|
|
# http://en.wikipedia.org/wiki/4%E2%80%934%E2%80%935_calendar#Last_Saturday_of_the_month_at_fiscal_year_end)
|
||
|
|
(offset_lom_sat_aug, datetime(2006, 8, 26), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2007, 8, 25), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2008, 8, 30), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2009, 8, 29), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2010, 8, 28), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2011, 8, 27), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2012, 8, 25), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2013, 8, 31), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2014, 8, 30), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2015, 8, 29), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2016, 8, 27), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2017, 8, 26), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2018, 8, 25), True),
|
||
|
|
(offset_lom_sat_aug, datetime(2019, 8, 31), True),
|
||
|
|
|
||
|
|
(offset_lom_sat_aug, datetime(2006, 8, 27), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2007, 8, 28), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2008, 8, 31), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2009, 8, 30), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2010, 8, 29), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2011, 8, 28), False),
|
||
|
|
|
||
|
|
(offset_lom_sat_aug, datetime(2006, 8, 25), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2007, 8, 24), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2008, 8, 29), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2009, 8, 28), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2010, 8, 27), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2011, 8, 26), False),
|
||
|
|
(offset_lom_sat_aug, datetime(2019, 8, 30), False),
|
||
|
|
|
||
|
|
# From GMCR (see for example:
|
||
|
|
# http://yahoo.brand.edgar-online.com/Default.aspx?
|
||
|
|
# companyid=3184&formtypeID=7)
|
||
|
|
(offset_lom_sat_sep, datetime(2010, 9, 25), True),
|
||
|
|
(offset_lom_sat_sep, datetime(2011, 9, 24), True),
|
||
|
|
(offset_lom_sat_sep, datetime(2012, 9, 29), True)]
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('case', on_offset_cases)
|
||
|
|
def test_onOffset(self, case):
|
||
|
|
offset, dt, expected = case
|
||
|
|
assert_onOffset(offset, dt, expected)
|
||
|
|
|
||
|
|
def test_apply(self):
|
||
|
|
offset_lom_aug_sat = makeFY5253LastOfMonth(startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT)
|
||
|
|
offset_lom_aug_sat_1 = makeFY5253LastOfMonth(n=1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT)
|
||
|
|
|
||
|
|
date_seq_lom_aug_sat = [datetime(2006, 8, 26), datetime(2007, 8, 25),
|
||
|
|
datetime(2008, 8, 30), datetime(2009, 8, 29),
|
||
|
|
datetime(2010, 8, 28), datetime(2011, 8, 27),
|
||
|
|
datetime(2012, 8, 25), datetime(2013, 8, 31),
|
||
|
|
datetime(2014, 8, 30), datetime(2015, 8, 29),
|
||
|
|
datetime(2016, 8, 27)]
|
||
|
|
|
||
|
|
tests = [
|
||
|
|
(offset_lom_aug_sat, date_seq_lom_aug_sat),
|
||
|
|
(offset_lom_aug_sat_1, date_seq_lom_aug_sat),
|
||
|
|
(offset_lom_aug_sat, [
|
||
|
|
datetime(2006, 8, 25)] + date_seq_lom_aug_sat),
|
||
|
|
(offset_lom_aug_sat_1, [
|
||
|
|
datetime(2006, 8, 27)] + date_seq_lom_aug_sat[1:]),
|
||
|
|
(makeFY5253LastOfMonth(n=-1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT),
|
||
|
|
list(reversed(date_seq_lom_aug_sat))),
|
||
|
|
]
|
||
|
|
for test in tests:
|
||
|
|
offset, data = test
|
||
|
|
current = data[0]
|
||
|
|
for datum in data[1:]:
|
||
|
|
current = current + offset
|
||
|
|
assert current == datum
|
||
|
|
|
||
|
|
|
||
|
|
class TestFY5253NearestEndMonth(Base):
|
||
|
|
|
||
|
|
def test_get_year_end(self):
|
||
|
|
assert (makeFY5253NearestEndMonth(
|
||
|
|
startingMonth=8, weekday=WeekDay.SAT).get_year_end(
|
||
|
|
datetime(2013, 1, 1)) == datetime(2013, 8, 31))
|
||
|
|
assert (makeFY5253NearestEndMonth(
|
||
|
|
startingMonth=8, weekday=WeekDay.SUN).get_year_end(
|
||
|
|
datetime(2013, 1, 1)) == datetime(2013, 9, 1))
|
||
|
|
assert (makeFY5253NearestEndMonth(
|
||
|
|
startingMonth=8, weekday=WeekDay.FRI).get_year_end(
|
||
|
|
datetime(2013, 1, 1)) == datetime(2013, 8, 30))
|
||
|
|
|
||
|
|
offset_n = FY5253(weekday=WeekDay.TUE, startingMonth=12,
|
||
|
|
variation="nearest")
|
||
|
|
assert (offset_n.get_year_end(datetime(2012, 1, 1)) ==
|
||
|
|
datetime(2013, 1, 1))
|
||
|
|
assert (offset_n.get_year_end(datetime(2012, 1, 10)) ==
|
||
|
|
datetime(2013, 1, 1))
|
||
|
|
|
||
|
|
assert (offset_n.get_year_end(datetime(2013, 1, 1)) ==
|
||
|
|
datetime(2013, 12, 31))
|
||
|
|
assert (offset_n.get_year_end(datetime(2013, 1, 2)) ==
|
||
|
|
datetime(2013, 12, 31))
|
||
|
|
assert (offset_n.get_year_end(datetime(2013, 1, 3)) ==
|
||
|
|
datetime(2013, 12, 31))
|
||
|
|
assert (offset_n.get_year_end(datetime(2013, 1, 10)) ==
|
||
|
|
datetime(2013, 12, 31))
|
||
|
|
|
||
|
|
JNJ = FY5253(n=1, startingMonth=12, weekday=6, variation="nearest")
|
||
|
|
assert (JNJ.get_year_end(datetime(2006, 1, 1)) ==
|
||
|
|
datetime(2006, 12, 31))
|
||
|
|
|
||
|
|
offset_lom_aug_sat = makeFY5253NearestEndMonth(1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT)
|
||
|
|
offset_lom_aug_thu = makeFY5253NearestEndMonth(1, startingMonth=8,
|
||
|
|
weekday=WeekDay.THU)
|
||
|
|
offset_n = FY5253(weekday=WeekDay.TUE, startingMonth=12,
|
||
|
|
variation="nearest")
|
||
|
|
|
||
|
|
on_offset_cases = [
|
||
|
|
# From Wikipedia (see:
|
||
|
|
# http://en.wikipedia.org/wiki/4%E2%80%934%E2%80%935_calendar
|
||
|
|
# #Saturday_nearest_the_end_of_month)
|
||
|
|
# 2006-09-02 2006 September 2
|
||
|
|
# 2007-09-01 2007 September 1
|
||
|
|
# 2008-08-30 2008 August 30 (leap year)
|
||
|
|
# 2009-08-29 2009 August 29
|
||
|
|
# 2010-08-28 2010 August 28
|
||
|
|
# 2011-09-03 2011 September 3
|
||
|
|
# 2012-09-01 2012 September 1 (leap year)
|
||
|
|
# 2013-08-31 2013 August 31
|
||
|
|
# 2014-08-30 2014 August 30
|
||
|
|
# 2015-08-29 2015 August 29
|
||
|
|
# 2016-09-03 2016 September 3 (leap year)
|
||
|
|
# 2017-09-02 2017 September 2
|
||
|
|
# 2018-09-01 2018 September 1
|
||
|
|
# 2019-08-31 2019 August 31
|
||
|
|
(offset_lom_aug_sat, datetime(2006, 9, 2), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2007, 9, 1), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2008, 8, 30), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2009, 8, 29), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2010, 8, 28), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2011, 9, 3), True),
|
||
|
|
|
||
|
|
(offset_lom_aug_sat, datetime(2016, 9, 3), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2017, 9, 2), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2018, 9, 1), True),
|
||
|
|
(offset_lom_aug_sat, datetime(2019, 8, 31), True),
|
||
|
|
|
||
|
|
(offset_lom_aug_sat, datetime(2006, 8, 27), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2007, 8, 28), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2008, 8, 31), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2009, 8, 30), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2010, 8, 29), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2011, 8, 28), False),
|
||
|
|
|
||
|
|
(offset_lom_aug_sat, datetime(2006, 8, 25), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2007, 8, 24), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2008, 8, 29), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2009, 8, 28), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2010, 8, 27), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2011, 8, 26), False),
|
||
|
|
(offset_lom_aug_sat, datetime(2019, 8, 30), False),
|
||
|
|
|
||
|
|
# From Micron, see:
|
||
|
|
# http://google.brand.edgar-online.com/?sym=MU&formtypeID=7
|
||
|
|
(offset_lom_aug_thu, datetime(2012, 8, 30), True),
|
||
|
|
(offset_lom_aug_thu, datetime(2011, 9, 1), True),
|
||
|
|
|
||
|
|
(offset_n, datetime(2012, 12, 31), False),
|
||
|
|
(offset_n, datetime(2013, 1, 1), True),
|
||
|
|
(offset_n, datetime(2013, 1, 2), False)]
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('case', on_offset_cases)
|
||
|
|
def test_onOffset(self, case):
|
||
|
|
offset, dt, expected = case
|
||
|
|
assert_onOffset(offset, dt, expected)
|
||
|
|
|
||
|
|
def test_apply(self):
|
||
|
|
date_seq_nem_8_sat = [datetime(2006, 9, 2), datetime(2007, 9, 1),
|
||
|
|
datetime(2008, 8, 30), datetime(2009, 8, 29),
|
||
|
|
datetime(2010, 8, 28), datetime(2011, 9, 3)]
|
||
|
|
|
||
|
|
JNJ = [datetime(2005, 1, 2), datetime(2006, 1, 1),
|
||
|
|
datetime(2006, 12, 31), datetime(2007, 12, 30),
|
||
|
|
datetime(2008, 12, 28), datetime(2010, 1, 3),
|
||
|
|
datetime(2011, 1, 2), datetime(2012, 1, 1),
|
||
|
|
datetime(2012, 12, 30)]
|
||
|
|
|
||
|
|
DEC_SAT = FY5253(n=-1, startingMonth=12, weekday=5,
|
||
|
|
variation="nearest")
|
||
|
|
|
||
|
|
tests = [
|
||
|
|
(makeFY5253NearestEndMonth(startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT),
|
||
|
|
date_seq_nem_8_sat),
|
||
|
|
(makeFY5253NearestEndMonth(n=1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT),
|
||
|
|
date_seq_nem_8_sat),
|
||
|
|
(makeFY5253NearestEndMonth(startingMonth=8, weekday=WeekDay.SAT),
|
||
|
|
[datetime(2006, 9, 1)] + date_seq_nem_8_sat),
|
||
|
|
(makeFY5253NearestEndMonth(n=1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT),
|
||
|
|
[datetime(2006, 9, 3)] + date_seq_nem_8_sat[1:]),
|
||
|
|
(makeFY5253NearestEndMonth(n=-1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT),
|
||
|
|
list(reversed(date_seq_nem_8_sat))),
|
||
|
|
(makeFY5253NearestEndMonth(n=1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SUN), JNJ),
|
||
|
|
(makeFY5253NearestEndMonth(n=-1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SUN),
|
||
|
|
list(reversed(JNJ))),
|
||
|
|
(makeFY5253NearestEndMonth(n=1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SUN),
|
||
|
|
[datetime(2005, 1, 2), datetime(2006, 1, 1)]),
|
||
|
|
(makeFY5253NearestEndMonth(n=1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SUN),
|
||
|
|
[datetime(2006, 1, 2), datetime(2006, 12, 31)]),
|
||
|
|
(DEC_SAT, [datetime(2013, 1, 15), datetime(2012, 12, 29)])
|
||
|
|
]
|
||
|
|
for test in tests:
|
||
|
|
offset, data = test
|
||
|
|
current = data[0]
|
||
|
|
for datum in data[1:]:
|
||
|
|
current = current + offset
|
||
|
|
assert current == datum
|
||
|
|
|
||
|
|
|
||
|
|
class TestFY5253LastOfMonthQuarter(Base):
|
||
|
|
|
||
|
|
def test_isAnchored(self):
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=1, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4).isAnchored()
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
weekday=WeekDay.SAT, startingMonth=3,
|
||
|
|
qtr_with_extra_week=4).isAnchored()
|
||
|
|
assert not makeFY5253LastOfMonthQuarter(
|
||
|
|
2, startingMonth=1, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4).isAnchored()
|
||
|
|
|
||
|
|
def test_equality(self):
|
||
|
|
assert (makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=1, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4) == makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=1, weekday=WeekDay.SAT, qtr_with_extra_week=4))
|
||
|
|
assert (makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=1, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4) != makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=1, weekday=WeekDay.SUN, qtr_with_extra_week=4))
|
||
|
|
assert (makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=1, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4) != makeFY5253LastOfMonthQuarter(
|
||
|
|
startingMonth=2, weekday=WeekDay.SAT, qtr_with_extra_week=4))
|
||
|
|
|
||
|
|
def test_offset(self):
|
||
|
|
offset = makeFY5253LastOfMonthQuarter(1, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
offset2 = makeFY5253LastOfMonthQuarter(2, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
offset4 = makeFY5253LastOfMonthQuarter(4, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
|
||
|
|
offset_neg1 = makeFY5253LastOfMonthQuarter(-1, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
offset_neg2 = makeFY5253LastOfMonthQuarter(-2, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
|
||
|
|
GMCR = [datetime(2010, 3, 27), datetime(2010, 6, 26),
|
||
|
|
datetime(2010, 9, 25), datetime(2010, 12, 25),
|
||
|
|
datetime(2011, 3, 26), datetime(2011, 6, 25),
|
||
|
|
datetime(2011, 9, 24), datetime(2011, 12, 24),
|
||
|
|
datetime(2012, 3, 24), datetime(2012, 6, 23),
|
||
|
|
datetime(2012, 9, 29), datetime(2012, 12, 29),
|
||
|
|
datetime(2013, 3, 30), datetime(2013, 6, 29)]
|
||
|
|
|
||
|
|
assert_offset_equal(offset, base=GMCR[0], expected=GMCR[1])
|
||
|
|
assert_offset_equal(offset, base=GMCR[0] + relativedelta(days=-1),
|
||
|
|
expected=GMCR[0])
|
||
|
|
assert_offset_equal(offset, base=GMCR[1], expected=GMCR[2])
|
||
|
|
|
||
|
|
assert_offset_equal(offset2, base=GMCR[0], expected=GMCR[2])
|
||
|
|
assert_offset_equal(offset4, base=GMCR[0], expected=GMCR[4])
|
||
|
|
|
||
|
|
assert_offset_equal(offset_neg1, base=GMCR[-1], expected=GMCR[-2])
|
||
|
|
assert_offset_equal(offset_neg1,
|
||
|
|
base=GMCR[-1] + relativedelta(days=+1),
|
||
|
|
expected=GMCR[-1])
|
||
|
|
assert_offset_equal(offset_neg2, base=GMCR[-1], expected=GMCR[-3])
|
||
|
|
|
||
|
|
date = GMCR[0] + relativedelta(days=-1)
|
||
|
|
for expected in GMCR:
|
||
|
|
assert_offset_equal(offset, date, expected)
|
||
|
|
date = date + offset
|
||
|
|
|
||
|
|
date = GMCR[-1] + relativedelta(days=+1)
|
||
|
|
for expected in reversed(GMCR):
|
||
|
|
assert_offset_equal(offset_neg1, date, expected)
|
||
|
|
date = date + offset_neg1
|
||
|
|
|
||
|
|
lomq_aug_sat_4 = makeFY5253LastOfMonthQuarter(1, startingMonth=8,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
lomq_sep_sat_4 = makeFY5253LastOfMonthQuarter(1, startingMonth=9,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
|
||
|
|
on_offset_cases = [
|
||
|
|
# From Wikipedia
|
||
|
|
(lomq_aug_sat_4, datetime(2006, 8, 26), True),
|
||
|
|
(lomq_aug_sat_4, datetime(2007, 8, 25), True),
|
||
|
|
(lomq_aug_sat_4, datetime(2008, 8, 30), True),
|
||
|
|
(lomq_aug_sat_4, datetime(2009, 8, 29), True),
|
||
|
|
(lomq_aug_sat_4, datetime(2010, 8, 28), True),
|
||
|
|
(lomq_aug_sat_4, datetime(2011, 8, 27), True),
|
||
|
|
(lomq_aug_sat_4, datetime(2019, 8, 31), True),
|
||
|
|
|
||
|
|
(lomq_aug_sat_4, datetime(2006, 8, 27), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2007, 8, 28), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2008, 8, 31), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2009, 8, 30), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2010, 8, 29), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2011, 8, 28), False),
|
||
|
|
|
||
|
|
(lomq_aug_sat_4, datetime(2006, 8, 25), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2007, 8, 24), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2008, 8, 29), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2009, 8, 28), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2010, 8, 27), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2011, 8, 26), False),
|
||
|
|
(lomq_aug_sat_4, datetime(2019, 8, 30), False),
|
||
|
|
|
||
|
|
# From GMCR
|
||
|
|
(lomq_sep_sat_4, datetime(2010, 9, 25), True),
|
||
|
|
(lomq_sep_sat_4, datetime(2011, 9, 24), True),
|
||
|
|
(lomq_sep_sat_4, datetime(2012, 9, 29), True),
|
||
|
|
|
||
|
|
(lomq_sep_sat_4, datetime(2013, 6, 29), True),
|
||
|
|
(lomq_sep_sat_4, datetime(2012, 6, 23), True),
|
||
|
|
(lomq_sep_sat_4, datetime(2012, 6, 30), False),
|
||
|
|
|
||
|
|
(lomq_sep_sat_4, datetime(2013, 3, 30), True),
|
||
|
|
(lomq_sep_sat_4, datetime(2012, 3, 24), True),
|
||
|
|
|
||
|
|
(lomq_sep_sat_4, datetime(2012, 12, 29), True),
|
||
|
|
(lomq_sep_sat_4, datetime(2011, 12, 24), True),
|
||
|
|
|
||
|
|
# INTC (extra week in Q1)
|
||
|
|
# See: http://www.intc.com/releasedetail.cfm?ReleaseID=542844
|
||
|
|
(makeFY5253LastOfMonthQuarter(1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1),
|
||
|
|
datetime(2011, 4, 2), True),
|
||
|
|
|
||
|
|
# see: http://google.brand.edgar-online.com/?sym=INTC&formtypeID=7
|
||
|
|
(makeFY5253LastOfMonthQuarter(1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1),
|
||
|
|
datetime(2012, 12, 29), True),
|
||
|
|
(makeFY5253LastOfMonthQuarter(1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1),
|
||
|
|
datetime(2011, 12, 31), True),
|
||
|
|
(makeFY5253LastOfMonthQuarter(1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1),
|
||
|
|
datetime(2010, 12, 25), True)]
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('case', on_offset_cases)
|
||
|
|
def test_onOffset(self, case):
|
||
|
|
offset, dt, expected = case
|
||
|
|
assert_onOffset(offset, dt, expected)
|
||
|
|
|
||
|
|
def test_year_has_extra_week(self):
|
||
|
|
# End of long Q1
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(datetime(2011, 4, 2))
|
||
|
|
|
||
|
|
# Start of long Q1
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(datetime(2010, 12, 26))
|
||
|
|
|
||
|
|
# End of year before year with long Q1
|
||
|
|
assert not makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(datetime(2010, 12, 25))
|
||
|
|
|
||
|
|
for year in [x
|
||
|
|
for x in range(1994, 2011 + 1)
|
||
|
|
if x not in [2011, 2005, 2000, 1994]]:
|
||
|
|
assert not makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(
|
||
|
|
datetime(year, 4, 2))
|
||
|
|
|
||
|
|
# Other long years
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(datetime(2005, 4, 2))
|
||
|
|
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(datetime(2000, 4, 2))
|
||
|
|
|
||
|
|
assert makeFY5253LastOfMonthQuarter(
|
||
|
|
1, startingMonth=12, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1).year_has_extra_week(datetime(1994, 4, 2))
|
||
|
|
|
||
|
|
def test_get_weeks(self):
|
||
|
|
sat_dec_1 = makeFY5253LastOfMonthQuarter(1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=1)
|
||
|
|
sat_dec_4 = makeFY5253LastOfMonthQuarter(1, startingMonth=12,
|
||
|
|
weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
|
||
|
|
assert sat_dec_1.get_weeks(datetime(2011, 4, 2)) == [14, 13, 13, 13]
|
||
|
|
assert sat_dec_4.get_weeks(datetime(2011, 4, 2)) == [13, 13, 13, 14]
|
||
|
|
assert sat_dec_1.get_weeks(datetime(2010, 12, 25)) == [13, 13, 13, 13]
|
||
|
|
|
||
|
|
|
||
|
|
class TestFY5253NearestEndMonthQuarter(Base):
|
||
|
|
|
||
|
|
offset_nem_sat_aug_4 = makeFY5253NearestEndMonthQuarter(
|
||
|
|
1, startingMonth=8, weekday=WeekDay.SAT,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
offset_nem_thu_aug_4 = makeFY5253NearestEndMonthQuarter(
|
||
|
|
1, startingMonth=8, weekday=WeekDay.THU,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
offset_n = FY5253(weekday=WeekDay.TUE, startingMonth=12,
|
||
|
|
variation="nearest")
|
||
|
|
|
||
|
|
on_offset_cases = [
|
||
|
|
# From Wikipedia
|
||
|
|
(offset_nem_sat_aug_4, datetime(2006, 9, 2), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2007, 9, 1), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2008, 8, 30), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2009, 8, 29), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2010, 8, 28), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2011, 9, 3), True),
|
||
|
|
|
||
|
|
(offset_nem_sat_aug_4, datetime(2016, 9, 3), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2017, 9, 2), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2018, 9, 1), True),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2019, 8, 31), True),
|
||
|
|
|
||
|
|
(offset_nem_sat_aug_4, datetime(2006, 8, 27), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2007, 8, 28), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2008, 8, 31), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2009, 8, 30), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2010, 8, 29), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2011, 8, 28), False),
|
||
|
|
|
||
|
|
(offset_nem_sat_aug_4, datetime(2006, 8, 25), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2007, 8, 24), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2008, 8, 29), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2009, 8, 28), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2010, 8, 27), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2011, 8, 26), False),
|
||
|
|
(offset_nem_sat_aug_4, datetime(2019, 8, 30), False),
|
||
|
|
|
||
|
|
# From Micron, see:
|
||
|
|
# http://google.brand.edgar-online.com/?sym=MU&formtypeID=7
|
||
|
|
(offset_nem_thu_aug_4, datetime(2012, 8, 30), True),
|
||
|
|
(offset_nem_thu_aug_4, datetime(2011, 9, 1), True),
|
||
|
|
|
||
|
|
# See: http://google.brand.edgar-online.com/?sym=MU&formtypeID=13
|
||
|
|
(offset_nem_thu_aug_4, datetime(2013, 5, 30), True),
|
||
|
|
(offset_nem_thu_aug_4, datetime(2013, 2, 28), True),
|
||
|
|
(offset_nem_thu_aug_4, datetime(2012, 11, 29), True),
|
||
|
|
(offset_nem_thu_aug_4, datetime(2012, 5, 31), True),
|
||
|
|
(offset_nem_thu_aug_4, datetime(2007, 3, 1), True),
|
||
|
|
(offset_nem_thu_aug_4, datetime(1994, 3, 3), True),
|
||
|
|
|
||
|
|
(offset_n, datetime(2012, 12, 31), False),
|
||
|
|
(offset_n, datetime(2013, 1, 1), True),
|
||
|
|
(offset_n, datetime(2013, 1, 2), False)]
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('case', on_offset_cases)
|
||
|
|
def test_onOffset(self, case):
|
||
|
|
offset, dt, expected = case
|
||
|
|
assert_onOffset(offset, dt, expected)
|
||
|
|
|
||
|
|
def test_offset(self):
|
||
|
|
offset = makeFY5253NearestEndMonthQuarter(1, startingMonth=8,
|
||
|
|
weekday=WeekDay.THU,
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
|
||
|
|
MU = [datetime(2012, 5, 31),
|
||
|
|
datetime(2012, 8, 30), datetime(2012, 11, 29),
|
||
|
|
datetime(2013, 2, 28), datetime(2013, 5, 30)]
|
||
|
|
|
||
|
|
date = MU[0] + relativedelta(days=-1)
|
||
|
|
for expected in MU:
|
||
|
|
assert_offset_equal(offset, date, expected)
|
||
|
|
date = date + offset
|
||
|
|
|
||
|
|
assert_offset_equal(offset,
|
||
|
|
datetime(2012, 5, 31),
|
||
|
|
datetime(2012, 8, 30))
|
||
|
|
assert_offset_equal(offset,
|
||
|
|
datetime(2012, 5, 30),
|
||
|
|
datetime(2012, 5, 31))
|
||
|
|
|
||
|
|
offset2 = FY5253Quarter(weekday=5, startingMonth=12, variation="last",
|
||
|
|
qtr_with_extra_week=4)
|
||
|
|
|
||
|
|
assert_offset_equal(offset2,
|
||
|
|
datetime(2013, 1, 15),
|
||
|
|
datetime(2013, 3, 30))
|
||
|
|
|
||
|
|
|
||
|
|
def test_bunched_yearends():
|
||
|
|
# GH#14774 cases with two fiscal year-ends in the same calendar-year
|
||
|
|
fy = FY5253(n=1, weekday=5, startingMonth=12, variation='nearest')
|
||
|
|
dt = Timestamp('2004-01-01')
|
||
|
|
assert fy.rollback(dt) == Timestamp('2002-12-28')
|
||
|
|
assert (-fy).apply(dt) == Timestamp('2002-12-28')
|
||
|
|
assert dt - fy == Timestamp('2002-12-28')
|
||
|
|
|
||
|
|
assert fy.rollforward(dt) == Timestamp('2004-01-03')
|
||
|
|
assert fy.apply(dt) == Timestamp('2004-01-03')
|
||
|
|
assert fy + dt == Timestamp('2004-01-03')
|
||
|
|
assert dt + fy == Timestamp('2004-01-03')
|
||
|
|
|
||
|
|
# Same thing, but starting from a Timestamp in the previous year.
|
||
|
|
dt = Timestamp('2003-12-31')
|
||
|
|
assert fy.rollback(dt) == Timestamp('2002-12-28')
|
||
|
|
assert (-fy).apply(dt) == Timestamp('2002-12-28')
|
||
|
|
assert dt - fy == Timestamp('2002-12-28')
|
||
|
|
|
||
|
|
|
||
|
|
def test_fy5253_last_onoffset():
|
||
|
|
# GH#18877 dates on the year-end but not normalized to midnight
|
||
|
|
offset = FY5253(n=-5, startingMonth=5, variation="last", weekday=0)
|
||
|
|
ts = Timestamp('1984-05-28 06:29:43.955911354+0200',
|
||
|
|
tz='Europe/San_Marino')
|
||
|
|
fast = offset.onOffset(ts)
|
||
|
|
slow = (ts + offset) - offset == ts
|
||
|
|
assert fast == slow
|
||
|
|
|
||
|
|
|
||
|
|
def test_fy5253_nearest_onoffset():
|
||
|
|
# GH#18877 dates on the year-end but not normalized to midnight
|
||
|
|
offset = FY5253(n=3, startingMonth=7, variation="nearest", weekday=2)
|
||
|
|
ts = Timestamp('2032-07-28 00:12:59.035729419+0000', tz='Africa/Dakar')
|
||
|
|
fast = offset.onOffset(ts)
|
||
|
|
slow = (ts + offset) - offset == ts
|
||
|
|
assert fast == slow
|
||
|
|
|
||
|
|
|
||
|
|
def test_fy5253qtr_onoffset_nearest():
|
||
|
|
# GH#19036
|
||
|
|
ts = Timestamp('1985-09-02 23:57:46.232550356-0300',
|
||
|
|
tz='Atlantic/Bermuda')
|
||
|
|
offset = FY5253Quarter(n=3, qtr_with_extra_week=1, startingMonth=2,
|
||
|
|
variation="nearest", weekday=0)
|
||
|
|
fast = offset.onOffset(ts)
|
||
|
|
slow = (ts + offset) - offset == ts
|
||
|
|
assert fast == slow
|
||
|
|
|
||
|
|
|
||
|
|
def test_fy5253qtr_onoffset_last():
|
||
|
|
# GH#19036
|
||
|
|
offset = FY5253Quarter(n=-2, qtr_with_extra_week=1,
|
||
|
|
startingMonth=7, variation="last", weekday=2)
|
||
|
|
ts = Timestamp('2011-01-26 19:03:40.331096129+0200',
|
||
|
|
tz='Africa/Windhoek')
|
||
|
|
slow = (ts + offset) - offset == ts
|
||
|
|
fast = offset.onOffset(ts)
|
||
|
|
assert fast == slow
|