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))