def test_workload(self):
p1 = Project(30, 1, 3)
p2 = Project(30, 1, 3)
w = Workflow(20, 0, [p1, p2])
self.assertEqual(w.average_workload(), 3)
w.reserved_resources = 4
self.assertAlmostEqual(w.average_workload(), 3.75)
p3 = Project(30, 1, 3)
p3.hours_left = 5
w2 = w.add_project(p3)
self.assertAlmostEqual(w2.average_workload(), 4.0625)
w3 = Workflow(20, 20, [])
self.assertEqual(w3.average_workload(), 0)
w4 = Workflow(20, 20, [p1, p2])
self.assertEqual(w4.average_workload(), 0)
