def test_load_with_tasks(self):
employee = Employee(**self.employee_data)
employee.tasks = [Task(name='task_%d' % i, complexity=randint(1, 5))
for i in range(1, 11)]
self.assertEqual(employee.load,
employee.total_cp / float(employee.cp_per_time))
def test_total_cp_with_tasks(self):
tasks = [Task(name='task_%d' % i, complexity=randint(1, 5)) for i in
range(1, 11)]
employee = Employee(**self.employee_data)
employee.tasks = tasks
self.assertEqual(employee.total_cp,
sum(task.complexity for task in tasks))