def test_manage(self): manager = TaskManager() self._create_employees(10) self._create_tasks(100*10) manager.manage() self._test_load_diff()
def test_create_task(self): data = {'name': 'anytank', 'complexity': 2} manager = TaskManager() manager.create_task(**data) self.assertEqual(len(manager.tasks), 1) self.assertEqual(manager.tasks[0].name, data['name']) self.assertEqual(manager.tasks[0].complexity, data['complexity'])
def test_create_employee(self): data = {'name': 'anyemployee', 'cp_per_time': 2} manager = TaskManager() manager.create_empoyee(**data) self.assertEqual(len(manager.employees), 1) self.assertEqual(manager.employees[0].name, data['name']) self.assertEqual(manager.employees[0].cp_per_time, data['cp_per_time'])
def _create_employees(self, count): manager = TaskManager() for i in range(count): manager.create_empoyee('employee_%d' % i, randint(1, self.max_complexity))
def _create_tasks(self, count): manager = TaskManager() for i in range(count): manager.create_task('task_%d' % i, randint(1, self.max_complexity))