no timeout_decorator windows

Author: KRRT7

codeflash/code_utils/instrument_existing_tests.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import ast
+import platform
 from pathlib import Path
 from typing import TYPE_CHECKING
 
@@ -135,7 +136,10 @@ def visit_ClassDef(self, node: ast.ClassDef) -> ast.ClassDef:
     def visit_FunctionDef(self, node: ast.FunctionDef, test_class_name: str | None = None) -> ast.FunctionDef:
         if node.name.startswith("test_"):
             did_update = False
-            if self.test_framework == "unittest":
+            if self.test_framework == "unittest" and platform.system() != "Windows":
+                # Only add timeout decorator on non-Windows platforms
+                # Windows doesn't support SIGALRM signal required by timeout_decorator
+
                 node.decorator_list.append(
                     ast.Call(
                         func=ast.Name(id="timeout_decorator.timeout", ctx=ast.Load()),
@@ -354,7 +358,7 @@ def inject_profiling_into_existing_test(
         new_imports.extend(
             [ast.Import(names=[ast.alias(name="sqlite3")]), ast.Import(names=[ast.alias(name="dill", asname="pickle")])]
         )
-    if test_framework == "unittest":
+    if test_framework == "unittest" and platform.system() != "Windows":
         new_imports.append(ast.Import(names=[ast.alias(name="timeout_decorator")]))
     tree.body = [*new_imports, create_wrapper_function(mode), *tree.body]
     return True, isort.code(ast.unparse(tree), float_to_top=True)

tests/test_instrument_tests.py

@@ -24,6 +24,8 @@
     TestsInFile,
     TestType,
 )
+import platform
+
 from codeflash.optimization.function_optimizer import FunctionOptimizer
 from codeflash.verification.verification_utils import TestConfig
 
@@ -1451,6 +1453,7 @@ def test_sort():
 
 
 def test_perfinjector_bubble_sort_unittest_results() -> None:
+    
     code = """import unittest
 
 from code_to_optimize.bubble_sort import sorter
@@ -1471,8 +1474,74 @@ def test_sort(self):
         self.assertEqual(output, list(range(50)))
 """
 
-    expected = (
-        """import gc
+    is_windows = platform.system() == "Windows"
+    
+    if is_windows:
+        expected = (
+            """import gc
+import os
+import sqlite3
+import time
+import unittest
+
+import dill as pickle
+
+from code_to_optimize.bubble_sort import sorter
+
+
+"""
+            + codeflash_wrap_string
+            + """
+class TestPigLatin(unittest.TestCase):
+
+    def test_sort(self):
+        codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
+        codeflash_iteration = os.environ['CODEFLASH_TEST_ITERATION']
+        codeflash_con = sqlite3.connect(f'{tmp_dir_path}_{{codeflash_iteration}}.sqlite')
+        codeflash_cur = codeflash_con.cursor()
+        codeflash_cur.execute('CREATE TABLE IF NOT EXISTS test_results (test_module_path TEXT, test_class_name TEXT, test_function_name TEXT, function_getting_tested TEXT, loop_index INTEGER, iteration_id TEXT, runtime INTEGER, return_value BLOB, verification_type TEXT)')
+        input = [5, 4, 3, 2, 1, 0]
+        output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '1', codeflash_loop_index, codeflash_cur, codeflash_con, input)
+        self.assertEqual(output, [0, 1, 2, 3, 4, 5])
+        input = [5.0, 4.0, 3.0, 2.0, 1.0, 0.0]
+        output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '4', codeflash_loop_index, codeflash_cur, codeflash_con, input)
+        self.assertEqual(output, [0.0, 1.0, 2.0, 3.0, 4.0, 5.0])
+        input = list(reversed(range(50)))
+        output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '7', codeflash_loop_index, codeflash_cur, codeflash_con, input)
+        self.assertEqual(output, list(range(50)))
+        codeflash_con.close()
+"""
+        )
+        expected_perf = (
+            """import gc
+import os
+import time
+import unittest
+
+from code_to_optimize.bubble_sort import sorter
+
+
+"""
+            + codeflash_wrap_perfonly_string
+            + """
+class TestPigLatin(unittest.TestCase):
+
+    def test_sort(self):
+        codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
+        input = [5, 4, 3, 2, 1, 0]
+        output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '1', codeflash_loop_index, input)
+        self.assertEqual(output, [0, 1, 2, 3, 4, 5])
+        input = [5.0, 4.0, 3.0, 2.0, 1.0, 0.0]
+        output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '4', codeflash_loop_index, input)
+        self.assertEqual(output, [0.0, 1.0, 2.0, 3.0, 4.0, 5.0])
+        input = list(reversed(range(50)))
+        output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '7', codeflash_loop_index, input)
+        self.assertEqual(output, list(range(50)))
+"""
+        )
+    else:
+        expected = (
+            """import gc
 import os
 import sqlite3
 import time
@@ -1485,8 +1554,8 @@ def test_sort(self):
 
 
 """
-        + codeflash_wrap_string
-        + """
+            + codeflash_wrap_string
+            + """
 class TestPigLatin(unittest.TestCase):
 
     @timeout_decorator.timeout(15)
@@ -1507,9 +1576,9 @@ def test_sort(self):
         self.assertEqual(output, list(range(50)))
         codeflash_con.close()
 """
-    )
-    expected_perf = (
-        """import gc
+        )
+        expected_perf = (
+            """import gc
 import os
 import time
 import unittest
@@ -1520,8 +1589,8 @@ def test_sort(self):
 
 
 """
-        + codeflash_wrap_perfonly_string
-        + """
+            + codeflash_wrap_perfonly_string
+            + """
 class TestPigLatin(unittest.TestCase):
 
     @timeout_decorator.timeout(15)
@@ -1537,7 +1606,7 @@ def test_sort(self):
         output = codeflash_wrap(sorter, '{module_path}', 'TestPigLatin', 'test_sort', 'sorter', '7', codeflash_loop_index, input)
         self.assertEqual(output, list(range(50)))
 """
-    )
+        )
     code_path = (Path(__file__).parent.resolve() / "../code_to_optimize/bubble_sort.py").resolve()
     test_path = (
         Path(__file__).parent.resolve()