mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 20:35:00 +00:00
style: Format Python code with black
This commit is contained in:
parent
a480b90217
commit
b29a6e8fa4
1 changed files with 16 additions and 14 deletions
|
@ -1,13 +1,14 @@
|
|||
import unittest
|
||||
from unittest.mock import patch, MagicMock
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
from aider.coders import Coder
|
||||
from aider.models import Model
|
||||
from aider.io import InputOutput
|
||||
from aider.models import Model
|
||||
|
||||
|
||||
class TestScriptingAPI(unittest.TestCase):
|
||||
|
||||
@patch('aider.coders.Coder.create')
|
||||
@patch('aider.models.Model')
|
||||
@patch("aider.coders.Coder.create")
|
||||
@patch("aider.models.Model")
|
||||
def test_basic_scripting(self, mock_model, mock_coder_create):
|
||||
# Setup
|
||||
mock_coder = MagicMock()
|
||||
|
@ -28,8 +29,8 @@ class TestScriptingAPI(unittest.TestCase):
|
|||
mock_coder.run.assert_any_call("make a script that prints hello world")
|
||||
mock_coder.run.assert_any_call("make it say goodbye")
|
||||
|
||||
@patch('aider.coders.Coder.create')
|
||||
@patch('aider.models.Model')
|
||||
@patch("aider.coders.Coder.create")
|
||||
@patch("aider.models.Model")
|
||||
def test_scripting_with_io(self, mock_model, mock_coder_create):
|
||||
# Setup
|
||||
mock_coder = MagicMock()
|
||||
|
@ -49,5 +50,6 @@ class TestScriptingAPI(unittest.TestCase):
|
|||
mock_coder.run.assert_called_once_with("add a new function")
|
||||
self.assertTrue(io.yes) # Check that 'yes' is set to True
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue