diff --git a/benchmark/tests/test_benchmark.py b/benchmark/tests/test_benchmark.py
index e69de29bb..52b48cdd6 100644
--- a/benchmark/tests/test_benchmark.py
+++ b/benchmark/tests/test_benchmark.py
@@ -0,0 +1,14 @@
+import unittest
+from benchmark.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)