def test_todo_error_fix_only_accepts_one_line():
todo_error = TodoError()
wrong_input = ['', '']
with pytest.raises(Exception):
todo_error.fix(wrong_input)
def test_todo_error_fix(test_input, expected):
todo_error = TodoError()
assert todo_error.fix([test_input])[0] == expected