From a0eadb62c5f9b89317f6f44c63d78bbf67c0a5a1 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sun, 28 Jul 2024 16:40:39 -0300 Subject: [PATCH] Add test case for `check_for_urls` to handle repeated URLs --- tests/basic/test_coder.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py index 730eb88da..8e1213e4a 100644 --- a/tests/basic/test_coder.py +++ b/tests/basic/test_coder.py @@ -689,6 +689,12 @@ two result = coder.check_for_urls(no_url_input) self.assertEqual(result, no_url_input) + # Test case with the same URL appearing multiple times + repeated_url_input = "Check https://example.com, then https://example.com again, and https://example.com one more time" + result = coder.check_for_urls(repeated_url_input) + self.assertEqual(result.count("https://example.com"), 1) + self.assertIn("https://example.com", result) + if __name__ == "__main__": unittest.main()