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 _create_tasks(self, count): manager = TaskManager() for i in range(count): manager.create_task('task_%d' % i, randint(1, self.max_complexity))