@@ -1693,8 +1693,8 @@ def new_function2(value):
16931693print("Hello world")
16941694 """
16951695 expected_code = """import numpy as np
1696- print("Hello world")
16971696
1697+ print("Hello world")
16981698a=2
16991699print("Hello world")
17001700def some_fn():
@@ -1712,8 +1712,7 @@ def __init__(self, name):
17121712 def __call__(self, value):
17131713 return "I am still old"
17141714 def new_function2(value):
1715- return cst.ensure_type(value, str)
1716- """
1715+ return cst.ensure_type(value, str)"""
17171716 code_path = (Path (__file__ ).parent .resolve () / "../code_to_optimize/global_var_original.py" ).resolve ()
17181717 code_path .write_text (original_code , encoding = "utf-8" )
17191718 tests_root = Path ("/Users/codeflash/Downloads/codeflash-dev/codeflash/code_to_optimize/tests/pytest/" )
@@ -1769,8 +1768,8 @@ def new_function2(value):
17691768print("Hello world")
17701769 """
17711770 expected_code = """import numpy as np
1772- print("Hello world")
17731771
1772+ print("Hello world")
17741773print("Hello world")
17751774def some_fn():
17761775 a=np.zeros(10)
@@ -1846,8 +1845,8 @@ def new_function2(value):
18461845print("Hello world")
18471846 """
18481847 expected_code = """import numpy as np
1849- print("Hello world")
18501848
1849+ print("Hello world")
18511850a=3
18521851print("Hello world")
18531852def some_fn():
@@ -1922,8 +1921,8 @@ def new_function2(value):
19221921print("Hello world")
19231922 """
19241923 expected_code = """import numpy as np
1925- print("Hello world")
19261924
1925+ print("Hello world")
19271926a=2
19281927print("Hello world")
19291928def some_fn():
@@ -1999,8 +1998,8 @@ def new_function2(value):
19991998print("Hello world")
20001999 """
20012000 expected_code = """import numpy as np
2002- print("Hello world")
20032001
2002+ print("Hello world")
20042003a=3
20052004print("Hello world")
20062005def some_fn():
@@ -2082,8 +2081,8 @@ def new_function2(value):
20822081print("Hello world")
20832082"""
20842083 expected_code = """import numpy as np
2085- print("Hello world")
20862084
2085+ print("Hello world")
20872086if 2<3:
20882087 a=4
20892088else:
0 commit comments