codeflash-ai/codeflash
1
0
Fork 0
Code Issues 18 Pull requests 33 Projects Releases Packages Wiki Activity Actions 1.5k

Fix instrumentation unit tests

Browse source
This commit is contained in:
ihitamandal 2024-06-05 14:04:10 -07:00
parent 5e59754f99
commit 2503871b0e
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_instrumentation.py
View file

@ -133,6 +133,7 @@ import time
import dill as pickle
import pytest
import timeout_decorator
from packagename.ml.yolo.image_reshaping_utils import \\
prepare_image_for_yolo as \\
packagename_ml_yolo_image_reshaping_utils_prepare_image_for_yolo
@ -867,6 +868,7 @@ import unittest
import dill as pickle
import timeout_decorator
from code_to_optimize.bubble_sort import sorter
@ -1528,8 +1530,7 @@ class TestPigLatin(unittest.TestCase):
def test_update_line_node() -> None:
injectperf = InjectPerfOnly(
"sorter",
"code_to_optimize.tests.pytest.test_bubble_sort",
"sorter", "code_to_optimize.tests.pytest.test_bubble_sort", "pytest"
)
node = ast.Assign(
targets=[ast.Name(id="output", ctx=ast.Store())],