absolute value of percentage

Author: aseembits93

codeflash/code_utils/edit_generated_tests.py

@@ -133,11 +133,16 @@ def leave_SimpleStatementLine(
                     optimized_time = min(matching_optimized_times)
                     if original_time != 0 and optimized_time != 0:
                         perf_gain = format_perf(
-                            performance_gain(original_runtime_ns=original_time, optimized_runtime_ns=optimized_time)
-                            * 100
+                            abs(
+                                performance_gain(original_runtime_ns=original_time, optimized_runtime_ns=optimized_time)
+                                * 100
+                            )
                         )
+                        status = "slower" if optimized_time > original_time else "faster"
                         # Create the runtime comment
-                        comment_text = f"# {format_time(original_time)} -> {format_time(optimized_time)} ({perf_gain}%)"
+                        comment_text = (
+                            f"# {format_time(original_time)} -> {format_time(optimized_time)} ({perf_gain}% {status})"
+                        )
 
                         # Add comment to the trailing whitespace
                         new_trailing_whitespace = cst.TrailingWhitespace(

tests/test_add_runtime_comments.py

@@ -530,7 +530,7 @@ def test_add_runtime_comments_simple_function(self, test_config):
         )
 
         expected_source = '''def test_function():
-    codeflash_output = some_function() # 1.00s -> 500ms (100%)
+    codeflash_output = some_function() # 1.00s -> 500ms (100% faster)
     assert codeflash_output == expected
 '''
 
@@ -573,7 +573,7 @@ def test_function(self):
 
         expected_source = '''class TestClass:
     def test_function(self):
-        codeflash_output = some_function() # 2.00s -> 1.00s (100%)
+        codeflash_output = some_function() # 2.00s -> 1.00s (100% faster)
         assert codeflash_output == expected
 '''
 
@@ -617,9 +617,9 @@ def test_add_runtime_comments_multiple_assignments(self, test_config):
 
         expected_source = '''def test_function():
     setup_data = prepare_test()
-    codeflash_output = some_function() # 1.50s -> 750ms (100%)
+    codeflash_output = some_function() # 1.50s -> 750ms (100% faster)
     assert codeflash_output == expected
-    codeflash_output = another_function() # 1.50s -> 750ms (100%)
+    codeflash_output = another_function() # 1.50s -> 750ms (100% faster)
     assert codeflash_output == expected2
 '''
 
@@ -759,12 +759,12 @@ def test_add_runtime_comments_multiple_tests(self, test_config):
         )
 
         expected_source1 = '''def test_function1():
-    codeflash_output = some_function() # 1.00s -> 500ms (100%)
+    codeflash_output = some_function() # 1.00s -> 500ms (100% faster)
     assert codeflash_output == expected
 '''
 
         expected_source2 = '''def test_function2():
-    codeflash_output = another_function() # 2.00s -> 800ms (150%)
+    codeflash_output = another_function() # 2.00s -> 800ms (150% faster)
     assert codeflash_output == expected
 '''
 
@@ -805,7 +805,7 @@ def test_add_runtime_comments_performance_regression(self, test_config):
         )
 
         expected_source = '''def test_function():
-    codeflash_output = some_function() # 1.00s -> 1.50s (-33.3%)
+    codeflash_output = some_function() # 1.00s -> 1.50s (33.3% slower)
     assert codeflash_output == expected
 '''