fix a test

This commit is contained in:
Saurabh Misra 2025-01-03 18:09:25 -08:00
parent a9a6ad73e9
commit 04612142b4

View file

@ -45,7 +45,7 @@ def test_InjectPerfAndLogging_with() -> None:
codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
hdbscan = HDBSCAN()
with pytest.raises(AttributeError):
codeflash_return_value = codeflash_wrap(hdbscan.relative_validity_, 'code_to_optimize_path', None, 'test_relative_validity_no_tree', 'relative_validity_', '1_0', codeflash_loop_index, )"""
codeflash_return_value = codeflash_wrap(hdbscan.relative_validity_, 'code_to_optimize_path', None, 'test_relative_validity_no_tree', 'relative_validity_', '1_0', codeflash_loop_index)"""
assert ast.unparse(new_module_node).strip("\n") == expected
@ -78,7 +78,7 @@ def test_InjectPerfAndLogging() -> None:
expected = """def test_relative_validity_no_tree():
codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
hdbscan = HDBSCAN()
codeflash_return_value = codeflash_wrap(hdbscan.relative_validity_, 'code_to_optimize_path', None, 'test_relative_validity_no_tree', 'relative_validity_', '1', codeflash_loop_index, )"""
codeflash_return_value = codeflash_wrap(hdbscan.relative_validity_, 'code_to_optimize_path', None, 'test_relative_validity_no_tree', 'relative_validity_', '1', codeflash_loop_index)"""
assert ast.unparse(new_module_node).strip("\n") == expected