def testAdvanceDate(self): referenceDate = dt.date.today() expected = advanceDate(referenceDate, '3M') calculated = Date.fromDateTime(referenceDate) + '3M' self.assertEqual(expected, calculated.toDateTime()) expected = advanceDate(referenceDate, '-3M') calculated = Date.fromDateTime(referenceDate) - '3M' self.assertEqual(expected, calculated.toDateTime())
def testHolDatesList(self): holDtList = holDatesList('China.SSE', self.fromDate, self.toDate) fromDate = Date.fromDateTime(self.fromDate) currentDate = fromDate.toDateTime() cal = Calendar('China.SSE') while currentDate <= self.toDate: finpyDate = Date.fromDateTime(currentDate) if not cal.isBizDay(finpyDate): self.assertTrue(finpyDate.toDateTime() in holDtList, "{0} is expected as a holiday in {1}".format(finpyDate, cal)) currentDate = (finpyDate + 1).toDateTime()
def testDatesList(self): dtList = datesList(self.fromDate, self.toDate) fromDate = Date.fromDateTime(self.fromDate) for i, date in enumerate(dtList): expected = date calculated = (fromDate + i).toDateTime() self.assertEqual(date, (fromDate + i).toDateTime(), "at index {0:d}" "expected: {1}" "calculated: {2}".format(i, expected, calculated))