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

added relay test

Browse source
This commit is contained in:
Saga4 2024-10-20 03:55:51 +05:30
parent 56a8909ce5
commit ed03f7a7a6
1 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_instrumentation.py
View file

@ -2405,7 +2405,7 @@ def test_sleepfunc_sequence_long(n, expected_total_sleep_time):
def test_time_correction_only_replay_test() -> None: def test_time_correction_only_replay_test() -> None:
code = """import dill as pickle code = """import dill as pickle
import unittest import pytest
from codeflash.tracing.replay_test import get_next_arg_and_return from codeflash.tracing.replay_test import get_next_arg_and_return
from codeflash.validation.equivalence import compare_results from codeflash.validation.equivalence import compare_results
from code_to_optimize.sleeptime import sleepfunc_sequence from code_to_optimize.sleeptime import sleepfunc_sequence
@ -2429,10 +2429,8 @@ import time
import dill as pickle import dill as pickle
import pytest import pytest
from packagename.ml.yolo.image_reshaping_utils import \\
prepare_image_for_yolo as \\
packagename_ml_yolo_image_reshaping_utils_prepare_image_for_yolo
from code_to_optimize.sleeptime import sleepfunc_sequence
from codeflash.tracing.replay_test import get_next_arg_and_return from codeflash.tracing.replay_test import get_next_arg_and_return
from codeflash.validation.equivalence import compare_results from codeflash.validation.equivalence import compare_results
@ -2466,7 +2464,8 @@ def codeflash_wrap(wrapped, test_module_name, test_class_name, test_name, functi
codeflash_con.commit() codeflash_con.commit()
return return_value return return_value
def test_prepare_image_for_yolo(): @pytest.mark.parametrize('n, expected_total_sleep_time', [(1, 0.01), (2, 0.02), (3, 0.03), (4, 0.04)])
def test_sleepfunc_sequence_short(n, expected_total_sleep_time):
codeflash_iteration = os.environ['CODEFLASH_TEST_ITERATION'] codeflash_iteration = os.environ['CODEFLASH_TEST_ITERATION']
codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX']) codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
codeflash_con = sqlite3.connect(f'{tmp_dir_path}_{{codeflash_iteration}}.sqlite') codeflash_con = sqlite3.connect(f'{tmp_dir_path}_{{codeflash_iteration}}.sqlite')
@ -2481,7 +2480,7 @@ def test_prepare_image_for_yolo():
""" """
expected += """ args = pickle.loads(arg_val_pkl) expected += """ args = pickle.loads(arg_val_pkl)
return_val_1 = pickle.loads(return_val_pkl) return_val_1 = pickle.loads(return_val_pkl)
ret = codeflash_wrap(sleepfunc_sequence, '{module_path}', None, 'test_prepare_image_for_yolo', 'packagename_ml_yolo_image_reshaping_utils_prepare_image_for_yolo', '0_2', codeflash_loop_index, codeflash_cur, codeflash_con, **args) ret = codeflash_wrap(sleepfunc_sequence, '{module_path}', None, 'test_sleepfunc_sequence_short', 'sleepfunc_sequence', '0_2', codeflash_loop_index, codeflash_cur, codeflash_con, **args)
assert compare_results(return_val_1, ret) assert compare_results(return_val_1, ret)
codeflash_con.close() codeflash_con.close()
""" """
@ -2498,7 +2497,7 @@ def test_prepare_image_for_yolo():
os.chdir(run_cwd) os.chdir(run_cwd)
success, new_test = inject_profiling_into_existing_test( success, new_test = inject_profiling_into_existing_test(
Path(f.name), Path(f.name),
[CodePosition(10, 13), CodePosition(20, 13)], [CodePosition(16, 15)],
func, func,
Path(f.name).parent, Path(f.name).parent,
"pytest", "pytest",