mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-22 21:34:59 +00:00
159 lines
4.4 KiB
Python
159 lines
4.4 KiB
Python
# flake8: noqa: E501
|
|
|
|
import unittest
|
|
|
|
from aider import utils
|
|
|
|
|
|
class TestUtils(unittest.TestCase):
|
|
def test_replace_most_similar_chunk(self):
|
|
whole = "This is a sample text.\nAnother line of text.\nYet another line.\n"
|
|
part = "This is a sample text"
|
|
replace = "This is a replaced text."
|
|
expected_output = "This is a replaced text..\nAnother line of text.\nYet another line.\n"
|
|
|
|
result = utils.replace_most_similar_chunk(whole, part, replace)
|
|
self.assertEqual(result, expected_output)
|
|
|
|
def test_replace_most_similar_chunk_not_perfect_match(self):
|
|
whole = "This is a sample text.\nAnother line of text.\nYet another line."
|
|
part = "This was a sample text.\nAnother line of txt"
|
|
replace = "This is a replaced text.\nModified line of text."
|
|
expected_output = "This is a replaced text.\nModified line of text.\nYet another line."
|
|
|
|
result = utils.replace_most_similar_chunk(whole, part, replace)
|
|
self.assertEqual(result, expected_output)
|
|
|
|
def test_strip_quoted_wrapping(self):
|
|
input_text = (
|
|
"filename.ext\n```\nWe just want this content\nNot the filename and triple quotes\n```"
|
|
)
|
|
expected_output = "We just want this content\nNot the filename and triple quotes\n"
|
|
result = utils.strip_quoted_wrapping(input_text, "filename.ext")
|
|
self.assertEqual(result, expected_output)
|
|
|
|
def test_strip_quoted_wrapping_no_filename(self):
|
|
input_text = "```\nWe just want this content\nNot the triple quotes\n```"
|
|
expected_output = "We just want this content\nNot the triple quotes\n"
|
|
result = utils.strip_quoted_wrapping(input_text)
|
|
self.assertEqual(result, expected_output)
|
|
|
|
def test_strip_quoted_wrapping_no_wrapping(self):
|
|
input_text = "We just want this content\nNot the triple quotes\n"
|
|
expected_output = "We just want this content\nNot the triple quotes\n"
|
|
result = utils.strip_quoted_wrapping(input_text)
|
|
self.assertEqual(result, expected_output)
|
|
|
|
def test_find_original_update_blocks(self):
|
|
edit = """
|
|
Here's the change:
|
|
|
|
```text
|
|
foo.txt
|
|
<<<<<<< ORIGINAL
|
|
Two
|
|
=======
|
|
Tooooo
|
|
>>>>>>> UPDATED
|
|
```
|
|
|
|
Hope you like it!
|
|
"""
|
|
|
|
edits = list(utils.find_original_update_blocks(edit))
|
|
self.assertEqual(edits, [("foo.txt", "Two\n", "Tooooo\n")])
|
|
|
|
def test_find_original_update_blocks_quote_below_filename(self):
|
|
edit = """
|
|
Here's the change:
|
|
|
|
foo.txt
|
|
```text
|
|
<<<<<<< ORIGINAL
|
|
Two
|
|
=======
|
|
Tooooo
|
|
>>>>>>> UPDATED
|
|
```
|
|
|
|
Hope you like it!
|
|
"""
|
|
|
|
edits = list(utils.find_original_update_blocks(edit))
|
|
self.assertEqual(edits, [("foo.txt", "Two\n", "Tooooo\n")])
|
|
|
|
def test_find_original_update_blocks_unclosed(self):
|
|
edit = """
|
|
Here's the change:
|
|
|
|
```text
|
|
foo.txt
|
|
<<<<<<< ORIGINAL
|
|
Two
|
|
=======
|
|
Tooooo
|
|
|
|
|
|
oops!
|
|
"""
|
|
|
|
with self.assertRaises(ValueError) as cm:
|
|
list(utils.find_original_update_blocks(edit))
|
|
self.assertIn("Incomplete", str(cm.exception))
|
|
|
|
def test_find_original_update_blocks_missing_filename(self):
|
|
edit = """
|
|
Here's the change:
|
|
|
|
```text
|
|
<<<<<<< ORIGINAL
|
|
Two
|
|
=======
|
|
Tooooo
|
|
|
|
|
|
oops!
|
|
"""
|
|
|
|
with self.assertRaises(ValueError) as cm:
|
|
list(utils.find_original_update_blocks(edit))
|
|
self.assertIn("filename", str(cm.exception))
|
|
|
|
def test_find_original_update_blocks_no_final_newline(self):
|
|
edit = """
|
|
aider/coder.py
|
|
<<<<<<< ORIGINAL
|
|
self.console.print("[red]^C again to quit")
|
|
=======
|
|
self.io.tool_error("^C again to quit")
|
|
>>>>>>> UPDATED
|
|
|
|
aider/coder.py
|
|
<<<<<<< ORIGINAL
|
|
self.io.tool_error("Malformed ORIGINAL/UPDATE blocks, retrying...")
|
|
self.io.tool_error(err)
|
|
=======
|
|
self.io.tool_error("Malformed ORIGINAL/UPDATE blocks, retrying...")
|
|
self.io.tool_error(str(err))
|
|
>>>>>>> UPDATED
|
|
|
|
aider/coder.py
|
|
<<<<<<< ORIGINAL
|
|
self.console.print("[red]Unable to get commit message from gpt-3.5-turbo. Use /commit to try again.\n")
|
|
=======
|
|
self.io.tool_error("Unable to get commit message from gpt-3.5-turbo. Use /commit to try again.")
|
|
>>>>>>> UPDATED
|
|
|
|
aider/coder.py
|
|
<<<<<<< ORIGINAL
|
|
self.console.print("[red]Skipped commmit.")
|
|
=======
|
|
self.io.tool_error("Skipped commmit.")
|
|
>>>>>>> UPDATED"""
|
|
|
|
# Should not raise a ValueError
|
|
list(utils.find_original_update_blocks(edit))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|