def test_adding_task_and_editing_there_itself(self, mock_input): formatter = Format() todo = ToDoList(formatter) todo.add_task() with patch('sys.stdout', new=StringIO()) as fake_out: todo.view_tasks() self.assertEqual("Incomplete Tasks:\n\n1. Meet Ema at 7:00\n", fake_out.getvalue()) todo.edit_task() with patch('sys.stdout', new=StringIO()) as fake_out: todo.view_tasks() self.assertEqual("Incomplete Tasks:\n\n1. Meet Ema at 8:00\n", fake_out.getvalue())