class TodoTestCase(object): def __init__(self): self._todo = Todo() def set_up(self): self._todo.clear() def test_add_todo_adds_pending_item(self): self._todo.add("Sandwich") items = self._todo.items() assert_equal(("Sandwich", Todo.PENDING), items) def test_add_return_value(self): add_return_index = self._todo.add("Sandwich") assert_equal(0, add_return_index)
class TodoTestCase(object): def __init__(self): self._todo = Todo() def set_up(self): self._todo.clear() def test_add_todo_adds_pending_item(self): self._todo.add("Sandwich") items = self._todo.items() assert_equal(("Sandwich", Todo.PENDING), items) def test_add_return_value(self): add_return_index = self._todo.add("Sandwich") assert_equal(0, add_return_index)
def todo(): todos = Todo() if len(sys.argv) == 1: print('Incorrect usage!') else: cmd = sys.argv[1] if cmd == 'add': name = sys.argv[2] todos.add(name) elif cmd == 'remove': id = sys.argv[2] todos.remove(id) elif cmd == 'show': todos.show() elif cmd == 'done': id = sys.argv[2] todos.done(id) elif cmd == 'undone': id = sys.argv[2] todos.undone(id) elif cmd == 'clear': todos.clear() todos.save()