def testRemoveDaysAfter(self): day = Employee.myDateTime(6,6) day2 = Employee.myDateTime(3,3) middleDay = Employee.myDateTime(3,3) self.e.addReqDay(day) self.e.addReqDay(day2) self.assertTrue(len(self.e.getReqDays()) == 2) self.e.removeReqDaysBefore(middleDay) self.assertTrue(len(self.e.getReqDays()) == 1) self.assertEqual(self.e.getReqDays(),{day})
def testRequestedDays(self): day = Employee.myDateTime(1,1) self.e.addReqDay(day) self.assertEqual(self.e.getReqDays(),{day})