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