dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-27 15:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

added missing import

Browse source
This commit is contained in:
Paul Gauthier 2023-05-27 08:29:05 -07:00
parent bcf258f1c6
commit 7e75cfb1fa
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_coder.py
View file

@ -1,7 +1,9 @@
import os
import unittest import unittest
from unittest.mock import MagicMock from unittest.mock import MagicMock
from aider.coder import Coder from aider.coder import Coder
class TestCoder(unittest.TestCase): class TestCoder(unittest.TestCase):
def test_check_for_file_mentions(self): def test_check_for_file_mentions(self):
# Mock the IO object # Mock the IO object
@ -18,11 +20,12 @@ class TestCoder(unittest.TestCase):
coder.repo = mock_repo coder.repo = mock_repo
# Call the check_for_file_mentions method # Call the check_for_file_mentions method
result = coder.check_for_file_mentions("Please check file1.txt and file2.py") coder.check_for_file_mentions("Please check file1.txt and file2.py")
# Check if coder.abs_fnames contains both files # Check if coder.abs_fnames contains both files
expected_files = {os.path.abspath("file1.txt"), os.path.abspath("file2.py")} expected_files = {os.path.abspath("file1.txt"), os.path.abspath("file2.py")}
self.assertEqual(coder.abs_fnames, expected_files) self.assertEqual(coder.abs_fnames, expected_files)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()