def testReadCalendar(self):
expected = [
dt.date(2015, 1, 2),
dt.date(2015, 1, 5),
dt.date(2015, 1, 6),
dt.date(2015, 1, 7),
dt.date(2015, 1, 8),
dt.date(2015, 1, 9),
dt.date(2015, 1, 12),
dt.date(2015, 1, 13),
dt.date(2015, 1, 14),
dt.date(2015, 1, 15),
dt.date(2015, 1, 16),
dt.date(2015, 1, 20),
dt.date(2015, 1, 21),
dt.date(2015, 1, 22),
dt.date(2015, 1, 23),
dt.date(2015, 1, 26),
dt.date(2015, 1, 27),
dt.date(2015, 1, 28),
dt.date(2015, 1, 29),
dt.date(2015, 1, 30),
]
result = target.read_calendar(self.test_mktcal)
self.assertEqual(expected, result)
def testMakeDateSets4(self):
cal = target.read_calendar(self.test_mktcal)
refs = [dt.date(2015, 1, 3), dt.date(2015, 1, 6)]
tests = [dt.date(2015, 1, 9), dt.date(2015, 1, 11)]
days = 8
inc = 1
future = 10
fn_dst = None
try:
with tempfile.NamedTemporaryFile("w", prefix="dates_", delete=False) as f_dst:
fn_dst = f_dst.name
target.make_date_sets(cal, refs, tests, days, inc, future, f_dst)
self.assertTrue(filecmp.cmp(self.expected_datesets4, fn_dst))
finally:
if fn_dst is not None:
os.remove(fn_dst)
def testMakeDateSets4(self):
cal = target.read_calendar(self.test_mktcal)
refs = [dt.date(2015, 1, 3), dt.date(2015, 1, 6)]
tests = [dt.date(2015, 1, 9), dt.date(2015, 1,11)]
days = 8
inc = 1
future = 10
fn_dst = None
try:
with tempfile.NamedTemporaryFile('w', prefix='dates_', delete=False) as f_dst:
fn_dst = f_dst.name
target.make_date_sets(cal, refs, tests, days, inc, future, f_dst)
self.assertTrue(filecmp.cmp(self.expected_datesets4, fn_dst))
finally:
if fn_dst is not None:
os.remove(fn_dst)
def testReadCalendar(self):
expected = [ dt.date(2015, 1, 2)
,dt.date(2015, 1, 5)
,dt.date(2015, 1, 6)
,dt.date(2015, 1, 7)
,dt.date(2015, 1, 8)
,dt.date(2015, 1, 9)
,dt.date(2015, 1,12)
,dt.date(2015, 1,13)
,dt.date(2015, 1,14)
,dt.date(2015, 1,15)
,dt.date(2015, 1,16)
,dt.date(2015, 1,20)
,dt.date(2015, 1,21)
,dt.date(2015, 1,22)
,dt.date(2015, 1,23)
,dt.date(2015, 1,26)
,dt.date(2015, 1,27)
,dt.date(2015, 1,28)
,dt.date(2015, 1,29)
,dt.date(2015, 1,30)
]
result = target.read_calendar(self.test_mktcal)
self.assertEqual(expected, result)
