Added assertion to check for more than 5 <doc> entries in the result.

This commit is contained in:
Paul Gauthier (aider) 2024-07-05 17:36:32 -03:00
parent d21d8da374
commit 287b6e8158

View file

@ -25,5 +25,8 @@ class TestHelp(unittest.TestCase):
self.assertIn("ai", result.lower())
self.assertIn("chat", result.lower())
# Assert that there are more than 5 <doc> entries
self.assertGreater(result.count("<doc"), 5)
if __name__ == '__main__':
unittest.main()