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

remove long ==--- lines to save tokens

Browse source
This commit is contained in:
Paul Gauthier 2023-06-28 15:44:23 -07:00
parent 51638d0989
commit ea83df1ea3
3 changed files with 62 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

47
benchmark/test_benchmark.py Normal file
View file

@ -0,0 +1,47 @@
# flake8: noqa: E501
import unittest
from benchmark import cleanup_test_output
class TestCleanupTestOutput(unittest.TestCase):
def test_cleanup_test_output(self):
# Test case with timing info
output = "Ran 5 tests in 0.003s\nOK"
expected = "\nOK"
self.assertEqual(cleanup_test_output(output), expected)
# Test case without timing info
output = "OK"
expected = "OK"
self.assertEqual(cleanup_test_output(output), expected)
def test_cleanup_test_output_lines(self):
# Test case with timing info
output = """F
======================================================================
FAIL: test_cleanup_test_output (test_benchmark.TestCleanupTestOutput.test_cleanup_test_output)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/Users/gauthier/Projects/aider/benchmark/test_benchmark.py", line 14, in test_cleanup_test_output
self.assertEqual(cleanup_test_output(output), expected)
AssertionError: 'OK' != 'OKx'
- OK
+ OKx
? +
"""
expected = """F
====
FAIL: test_cleanup_test_output (test_benchmark.TestCleanupTestOutput.test_cleanup_test_output)
----
Traceback (most recent call last):
File "/Users/gauthier/Projects/aider/benchmark/test_benchmark.py", line 14, in test_cleanup_test_output
self.assertEqual(cleanup_test_output(output), expected)
AssertionError: 'OK' != 'OKx'
- OK
+ OKx
? +
"""
self.assertEqual(cleanup_test_output(output), expected)