def test_invalid_remove_task(self): subtask = Subtask() subtask.title = "First Subtask" task = Task() task.add_subtask(subtask) task.remove_subtask(subtask) with self.assertRaises(KeyError): task.remove_subtask(subtask)
def test_valid_remove_subtask(self): subtask = Subtask() task = Task() task.add_subtask(subtask) task.remove_subtask(subtask) self.assertEqual([], task.subtasks)
def test_invalid_add_subtask(self): subtask = "" task = Task() with self.assertRaises(ValueError): task.add_subtask(subtask)
def test_valid_add_subtask(self): subtask = Subtask() task = Task() task.add_subtask(subtask) self.assertTrue(subtask in task.subtasks)