style: format code with black and isort

This commit is contained in:
Paul Gauthier (aider) 2024-09-04 13:55:53 -07:00
parent cd551dbf63
commit 14fc3e3ad1

View file

@ -1,8 +1,10 @@
import unittest import unittest
from unittest.mock import patch, MagicMock
from pathlib import Path from pathlib import Path
from unittest.mock import MagicMock, patch
from aider.linter import Linter from aider.linter import Linter
class TestLinter(unittest.TestCase): class TestLinter(unittest.TestCase):
def setUp(self): def setUp(self):
self.linter = Linter(encoding="utf-8", root="/test/root") self.linter = Linter(encoding="utf-8", root="/test/root")
@ -12,7 +14,7 @@ class TestLinter(unittest.TestCase):
self.assertEqual(self.linter.root, "/test/root") self.assertEqual(self.linter.root, "/test/root")
self.assertIn("python", self.linter.languages) self.assertIn("python", self.linter.languages)
@patch('pathlib.Path.is_file') @patch("pathlib.Path.is_file")
def test_check_eslint_unix(self, mock_is_file): def test_check_eslint_unix(self, mock_is_file):
mock_is_file.return_value = True mock_is_file.return_value = True
self.linter._check_eslint() self.linter._check_eslint()
@ -20,10 +22,11 @@ class TestLinter(unittest.TestCase):
self.assertTrue(self.linter.languages["typescript"].startswith('"')) self.assertTrue(self.linter.languages["typescript"].startswith('"'))
self.assertTrue(self.linter.languages["typescript"].endswith('" --format unix')) self.assertTrue(self.linter.languages["typescript"].endswith('" --format unix'))
@patch('pathlib.Path.is_file') @patch("pathlib.Path.is_file")
def test_check_eslint_windows(self, mock_is_file): def test_check_eslint_windows(self, mock_is_file):
def side_effect(path): def side_effect(path):
return str(path).endswith('eslint.cmd') return str(path).endswith("eslint.cmd")
mock_is_file.side_effect = side_effect mock_is_file.side_effect = side_effect
self.linter._check_eslint() self.linter._check_eslint()
self.assertIn("typescript", self.linter.languages) self.assertIn("typescript", self.linter.languages)
@ -37,7 +40,7 @@ class TestLinter(unittest.TestCase):
self.assertEqual(self.linter.get_rel_fname("/test/root/file.py"), "file.py") self.assertEqual(self.linter.get_rel_fname("/test/root/file.py"), "file.py")
self.assertEqual(self.linter.get_rel_fname("/other/path/file.py"), "/other/path/file.py") self.assertEqual(self.linter.get_rel_fname("/other/path/file.py"), "/other/path/file.py")
@patch('subprocess.Popen') @patch("subprocess.Popen")
def test_run_cmd(self, mock_popen): def test_run_cmd(self, mock_popen):
mock_process = MagicMock() mock_process = MagicMock()
mock_process.returncode = 0 mock_process.returncode = 0
@ -47,7 +50,7 @@ class TestLinter(unittest.TestCase):
result = self.linter.run_cmd("test_cmd", "test_file.py", "code") result = self.linter.run_cmd("test_cmd", "test_file.py", "code")
self.assertIsNone(result) self.assertIsNone(result)
@patch('subprocess.Popen') @patch("subprocess.Popen")
def test_run_cmd_with_errors(self, mock_popen): def test_run_cmd_with_errors(self, mock_popen):
mock_process = MagicMock() mock_process = MagicMock()
mock_process.returncode = 1 mock_process.returncode = 1
@ -58,5 +61,6 @@ class TestLinter(unittest.TestCase):
self.assertIsNotNone(result) self.assertIsNotNone(result)
self.assertIn("Error message", result.text) self.assertIn("Error message", result.text)
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main() unittest.main()