def test_schedule_project(self):
# Assignees
person_id1 = 101
person_id2 = 102
person_id3 = 103
# Projects
p1 = Project(1)
p1_w1 = util.construct_work(1, "p1 w1", 1.0, [], person_id1, None)
p1_w2 = util.construct_work(2, "p1 w2", 1.0, [], person_id1, None)
p1_w3 = util.construct_work(5, "p1 w3", 0.1, [], person_id1, util.nov15)
p1.work = [p1_w1, p1_w2, p1_w3]
p2 = Project(2)
p2_w1 = util.construct_work(3, "p2 w1", 1.0, [], person_id2, None)
p2_w2 = util.construct_work(4, "p2 w2", 1.0, [2], person_id3, None)
p2.work = [p2_w1, p2_w2]
# Scheduler
scheduler = Scheduler(util.nov1)
projects = scheduler.schedule_projects([p1, p2])
# Check work end dates
self.assertEqual(util.nov2, p1_w1.est_end_date())
self.assertEqual(util.nov5, p1_w2.est_end_date())
self.assertEqual(util.nov2, p2_w1.est_end_date())
self.assertEqual(util.nov6, p2_w2.est_end_date())
# Check project end dates
self.assertEqual(util.nov15, p1.est_end_date)
self.assertEqual(util.nov6, p2.est_end_date)
return
def test_schedule_project(self):
# Assignees
person_id1 = 101
person_id2 = 102
person_id3 = 103
# Projects
p1 = Project(1)
p1_w1 = util.construct_work(1, "p1 w1", 1.0, [], person_id1, None)
p1_w2 = util.construct_work(2, "p1 w2", 1.0, [], person_id1, None)
p1_w3 = util.construct_work(5, "p1 w3", 0.1, [], person_id1,
util.nov15)
p1.work = [p1_w1, p1_w2, p1_w3]
p2 = Project(2)
p2_w1 = util.construct_work(3, "p2 w1", 1.0, [], person_id2, None)
p2_w2 = util.construct_work(4, "p2 w2", 1.0, [2], person_id3, None)
p2.work = [p2_w1, p2_w2]
# Scheduler
scheduler = Scheduler(util.nov1)
projects = scheduler.schedule_projects([p1, p2])
# Check work end dates
self.assertEqual(util.nov2, p1_w1.est_end_date())
self.assertEqual(util.nov5, p1_w2.est_end_date())
self.assertEqual(util.nov2, p2_w1.est_end_date())
self.assertEqual(util.nov6, p2_w2.est_end_date())
# Check project end dates
self.assertEqual(util.nov15, p1.est_end_date)
self.assertEqual(util.nov6, p2.est_end_date)
return
def test_schedule_with_key_date_in_past(self):
# Assignees
person_id1 = 101
# Projects
p1 = Project(1)
p1_w3 = util.construct_work(5, "p1 w3", 0.1, [], person_id1, util.nov1)
p1.work = [p1_w3]
# Scheduler
scheduler = Scheduler(util.nov15)
projects = scheduler.schedule_projects([p1])
self.assertEqual(util.nov15, p1.est_end_date)
return
def test_schedule_with_key_date_in_past(self):
# Assignees
person_id1 = 101
# Projects
p1 = Project(1)
p1_w3 = util.construct_work(5, "p1 w3", 0.1, [], person_id1, util.nov1)
p1.work = [p1_w3]
# Scheduler
scheduler = Scheduler(util.nov15)
projects = scheduler.schedule_projects([p1])
self.assertEqual(util.nov15, p1.est_end_date)
return
def test_construct_assignee_timelines(self):
scheduler = Scheduler(util.nov1)
t101 = scheduler.get_assignee_timeline(101)
self.assertEqual(0, len(t101.dates))
self.assertEqual(util.nov1, t101.cur_date)
return
def test_construct_assignee_timelines(self):
scheduler = Scheduler(util.nov1)
t101 = scheduler.get_assignee_timeline(101)
self.assertEqual(0, len(t101.dates))
self.assertEqual(util.nov1, t101.cur_date)
return