mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 08:44:59 +00:00
style: Remove trailing whitespace in test_io.py
This commit is contained in:
parent
9b80b693c1
commit
6d0078d39b
1 changed files with 8 additions and 8 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue