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_sort_work(self):
# Assignees
person_id1 = 101
person_id2 = 102
person_id3 = 103
# Projects
p1 = Project(1)
p1_w1 = construct_work(1, "p1 w1", 1.0, [6], person_id1, None)
p1_w2 = construct_work(2, "p1 w2", 1.0, [], person_id1, None)
p1_w3 = construct_work(5, "p1 w3", 0.1, [2, 1], person_id1, None)
p1_w4 = construct_work(6, "p1 w4", 0.1, [], person_id1, None)
p1.work = [p1_w1, p1_w2, p1_w3, p1_w4]
p1.topo_sort_work()
self.assertTrue(p1.work.index(p1_w1) < p1.work.index(p1_w3))
self.assertTrue(p1.work.index(p1_w4) < p1.work.index(p1_w1))
return
def test_sort_work(self):
# Assignees
person_id1 = 101
person_id2 = 102
person_id3 = 103
# Projects
p1 = Project(1)
p1_w1 = construct_work(1, "p1 w1", 1.0, [6], person_id1, None)
p1_w2 = construct_work(2, "p1 w2", 1.0, [], person_id1, None)
p1_w3 = construct_work(5, "p1 w3", 0.1, [2, 1], person_id1, None)
p1_w4 = construct_work(6, "p1 w4", 0.1, [], person_id1, None)
p1.work = [p1_w1, p1_w2, p1_w3, p1_w4]
p1.topo_sort_work()
self.assertTrue(p1.work.index(p1_w1) < p1.work.index(p1_w3))
self.assertTrue(p1.work.index(p1_w4) < p1.work.index(p1_w1))
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
