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)