@@ -1693,8 +1693,8 @@ def new_function2(value):
1693
1693
print("Hello world")
1694
1694
"""
1695
1695
expected_code = """import numpy as np
1696
- print("Hello world")
1697
1696
1697
+ print("Hello world")
1698
1698
a=2
1699
1699
print("Hello world")
1700
1700
def some_fn():
@@ -1712,8 +1712,7 @@ def __init__(self, name):
1712
1712
def __call__(self, value):
1713
1713
return "I am still old"
1714
1714
def new_function2(value):
1715
- return cst.ensure_type(value, str)
1716
- """
1715
+ return cst.ensure_type(value, str)"""
1717
1716
code_path = (Path (__file__ ).parent .resolve () / "../code_to_optimize/global_var_original.py" ).resolve ()
1718
1717
code_path .write_text (original_code , encoding = "utf-8" )
1719
1718
tests_root = Path ("/Users/codeflash/Downloads/codeflash-dev/codeflash/code_to_optimize/tests/pytest/" )
@@ -1769,8 +1768,8 @@ def new_function2(value):
1769
1768
print("Hello world")
1770
1769
"""
1771
1770
expected_code = """import numpy as np
1772
- print("Hello world")
1773
1771
1772
+ print("Hello world")
1774
1773
print("Hello world")
1775
1774
def some_fn():
1776
1775
a=np.zeros(10)
@@ -1846,8 +1845,8 @@ def new_function2(value):
1846
1845
print("Hello world")
1847
1846
"""
1848
1847
expected_code = """import numpy as np
1849
- print("Hello world")
1850
1848
1849
+ print("Hello world")
1851
1850
a=3
1852
1851
print("Hello world")
1853
1852
def some_fn():
@@ -1922,8 +1921,8 @@ def new_function2(value):
1922
1921
print("Hello world")
1923
1922
"""
1924
1923
expected_code = """import numpy as np
1925
- print("Hello world")
1926
1924
1925
+ print("Hello world")
1927
1926
a=2
1928
1927
print("Hello world")
1929
1928
def some_fn():
@@ -1999,8 +1998,8 @@ def new_function2(value):
1999
1998
print("Hello world")
2000
1999
"""
2001
2000
expected_code = """import numpy as np
2002
- print("Hello world")
2003
2001
2002
+ print("Hello world")
2004
2003
a=3
2005
2004
print("Hello world")
2006
2005
def some_fn():
@@ -2082,8 +2081,8 @@ def new_function2(value):
2082
2081
print("Hello world")
2083
2082
"""
2084
2083
expected_code = """import numpy as np
2085
- print("Hello world")
2086
2084
2085
+ print("Hello world")
2087
2086
if 2<3:
2088
2087
a=4
2089
2088
else:
0 commit comments