Skip to content

Commit 12f5f2a

Browse files
committed
ENH: Split test_offsets to test_bussiness_offsets and test_date_offsets (pandas-dev#27085)
1 parent 65e123c commit 12f5f2a

File tree

2 files changed

+45
-38
lines changed

2 files changed

+45
-38
lines changed

pandas/tests/tseries/offsets/test_offsets.py renamed to pandas/tests/tseries/offsets/test_business_offsets.py

-38
Original file line numberDiff line numberDiff line change
@@ -623,44 +623,6 @@ def test_pickle_v0_15_2(self, datapath):
623623
tm.assert_dict_equal(offsets, read_pickle(pickle_path))
624624

625625

626-
class TestDateOffset(Base):
627-
def setup_method(self, method):
628-
self.d = Timestamp(datetime(2008, 1, 2))
629-
_offset_map.clear()
630-
631-
def test_repr(self):
632-
repr(DateOffset())
633-
repr(DateOffset(2))
634-
repr(2 * DateOffset())
635-
repr(2 * DateOffset(months=2))
636-
637-
def test_mul(self):
638-
assert DateOffset(2) == 2 * DateOffset(1)
639-
assert DateOffset(2) == DateOffset(1) * 2
640-
641-
def test_constructor(self):
642-
643-
assert (self.d + DateOffset(months=2)) == datetime(2008, 3, 2)
644-
assert (self.d - DateOffset(months=2)) == datetime(2007, 11, 2)
645-
646-
assert (self.d + DateOffset(2)) == datetime(2008, 1, 4)
647-
648-
assert not DateOffset(2).isAnchored()
649-
assert DateOffset(1).isAnchored()
650-
651-
d = datetime(2008, 1, 31)
652-
assert (d + DateOffset(months=1)) == datetime(2008, 2, 29)
653-
654-
def test_copy(self):
655-
assert DateOffset(months=2).copy() == DateOffset(months=2)
656-
657-
def test_eq(self):
658-
offset1 = DateOffset(days=1)
659-
offset2 = DateOffset(days=365)
660-
661-
assert offset1 != offset2
662-
663-
664626
class TestBusinessDay(Base):
665627
_offset = BDay
666628

Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
from pandas._libs.tslibs import Timestamp
2+
from pandas.tseries.frequencies import _offset_map
3+
4+
from datetime import datetime
5+
6+
from pandas.tseries.offsets import DateOffset
7+
from .test_business_offsets import Base
8+
9+
10+
class TestDateOffset(Base):
11+
def setup_method(self, method):
12+
self.d = Timestamp(datetime(2008, 1, 2))
13+
_offset_map.clear()
14+
15+
def test_repr(self):
16+
repr(DateOffset())
17+
repr(DateOffset(2))
18+
repr(2 * DateOffset())
19+
repr(2 * DateOffset(months=2))
20+
21+
def test_mul(self):
22+
assert DateOffset(2) == 2 * DateOffset(1)
23+
assert DateOffset(2) == DateOffset(1) * 2
24+
25+
def test_constructor(self):
26+
27+
assert (self.d + DateOffset(months=2)) == datetime(2008, 3, 2)
28+
assert (self.d - DateOffset(months=2)) == datetime(2007, 11, 2)
29+
30+
assert (self.d + DateOffset(2)) == datetime(2008, 1, 4)
31+
32+
assert not DateOffset(2).isAnchored()
33+
assert DateOffset(1).isAnchored()
34+
35+
d = datetime(2008, 1, 31)
36+
assert (d + DateOffset(months=1)) == datetime(2008, 2, 29)
37+
38+
def test_copy(self):
39+
assert DateOffset(months=2).copy() == DateOffset(months=2)
40+
41+
def test_eq(self):
42+
offset1 = DateOffset(days=1)
43+
offset2 = DateOffset(days=365)
44+
45+
assert offset1 != offset2

0 commit comments

Comments
 (0)