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})
