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

test_run_with_file_utf_unicode_error

Browse source
This commit is contained in:
Paul Gauthier 2023-07-06 11:57:44 -07:00
parent 7cadfd9b4c
commit 4a15b6c010
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_coder.py
View file

@ -265,17 +265,22 @@ class TestCoder(unittest.TestCase):
self.assertNotEqual(coder.fence[0], "```") self.assertNotEqual(coder.fence[0], "```")
def test_run_with_file_utf_unicode_error(self): def test_run_with_file_utf_unicode_error(self):
"make sure that we honor InputOutput(encoding) and don't just assume utf-8"
# Create a few temporary files # Create a few temporary files
_, file1 = tempfile.mkstemp() _, file1 = tempfile.mkstemp()
_, file2 = tempfile.mkstemp() _, file2 = tempfile.mkstemp()
files = [file1, file2] files = [file1, file2]
encoding = 'latin-1' encoding = "utf-16"
# Initialize the Coder object with the mocked IO and mocked repo # Initialize the Coder object with the mocked IO and mocked repo
coder = Coder.create( coder = Coder.create(
models.GPT4, None, io=InputOutput(encoding=encoding), openai_api_key="fake_key", fnames=files models.GPT4,
None,
io=InputOutput(encoding=encoding),
openai_api_key="fake_key",
fnames=files,
) )
def mock_send(*args, **kwargs): def mock_send(*args, **kwargs):
@ -288,7 +293,7 @@ class TestCoder(unittest.TestCase):
coder.run(with_message="hi") coder.run(with_message="hi")
self.assertEqual(len(coder.abs_fnames), 2) self.assertEqual(len(coder.abs_fnames), 2)
some_content_which_will_error_if_read_with_encoding_utf8 = 'ÅÍÎÏ'.encode('utf-16') some_content_which_will_error_if_read_with_encoding_utf8 = "ÅÍÎÏ".encode(encoding)
with open(file1, "wb") as f: with open(file1, "wb") as f:
f.write(some_content_which_will_error_if_read_with_encoding_utf8) f.write(some_content_which_will_error_if_read_with_encoding_utf8)