style: Remove trailing whitespace in test_io.py

This commit is contained in:
Paul Gauthier (aider) 2025-02-06 13:56:22 -08:00
parent 9b80b693c1
commit 6d0078d39b

View file

@ -357,18 +357,18 @@ class TestInputOutputMultilineMode(unittest.TestCase):
"""Test that multiline mode is restored after KeyboardInterrupt"""
io = InputOutput(fancy_input=True)
io.prompt_session = MagicMock()
# Start in multiline mode
io.multiline_mode = True
# Mock prompt() to raise KeyboardInterrupt
io.prompt_session.prompt.side_effect = KeyboardInterrupt
# Test confirm_ask()
with self.assertRaises(KeyboardInterrupt):
io.confirm_ask("Test question?")
self.assertTrue(io.multiline_mode) # Should be restored
# Test prompt_ask()
with self.assertRaises(KeyboardInterrupt):
io.prompt_ask("Test prompt?")
@ -378,17 +378,17 @@ class TestInputOutputMultilineMode(unittest.TestCase):
"""Test that multiline mode is restored after normal exit"""
io = InputOutput(fancy_input=True)
io.prompt_session = MagicMock()
# Start in multiline mode
io.multiline_mode = True
# Mock prompt() to return normally
io.prompt_session.prompt.return_value = "y"
# Test confirm_ask()
io.confirm_ask("Test question?")
self.assertTrue(io.multiline_mode) # Should be restored
# Test prompt_ask()
io.prompt_ask("Test prompt?")
self.assertTrue(io.multiline_mode) # Should be restored