def testAddHoursFriday(self):
friday_at_16 = datetime(2011,8,19,16,0,0)
monday_at_12 = datetime(2011,8,22,12,0,0)
cal = BusinessCalendar(start_date=friday_at_16)
self.assertEquals(monday_at_12,cal.add_business_hours(4))
def testAddHoursOutsideOfficeHours(self):
monday_at_4 = datetime(2011,8,15,4,0,0)
monday_at_12 = datetime(2011,8,15,12,0,0)
cal = BusinessCalendar(start_date=monday_at_4)
self.assertEquals(monday_at_12, cal.add_business_hours(3))
def testAddHoursInWorkingDay(self):
monday_at_9 = datetime(2011,8,15,9,0,0)
monday_at_12 = datetime(2011,8,15,12,0,0)
cal = BusinessCalendar(start_date=monday_at_9)
self.assertEquals(monday_at_12, cal.add_business_hours(3))
def testAdd10HoursinWorkingDay(self):
monday_at_9 = datetime(2011,8,15,9,0,0)
tuesday_at_11 = datetime(2011,8,16,11,0,0)
cal = BusinessCalendar(start_date=monday_at_9)
self.assertEquals(tuesday_at_11, cal.add_business_hours(10))
def testNoOperation(self):
monday_at_9 = datetime(2011,8,15,9,0,0)
cal = BusinessCalendar(start_date=monday_at_9)
self.assertEquals(monday_at_9, cal.add_business_days(0))
self.assertEquals(monday_at_9, cal.add_business_hours(0))
