Skip to content

Commit 190d593

Browse files
fix: tests expected code new lines
1 parent 2e394f6 commit 190d593

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

tests/test_code_replacement.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1693,8 +1693,8 @@ def new_function2(value):
16931693
print("Hello world")
16941694
"""
16951695
expected_code = """import numpy as np
1696-
print("Hello world")
16971696
1697+
print("Hello world")
16981698
a=2
16991699
print("Hello world")
17001700
def 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):
17691768
print("Hello world")
17701769
"""
17711770
expected_code = """import numpy as np
1772-
print("Hello world")
17731771
1772+
print("Hello world")
17741773
print("Hello world")
17751774
def some_fn():
17761775
a=np.zeros(10)
@@ -1846,8 +1845,8 @@ def new_function2(value):
18461845
print("Hello world")
18471846
"""
18481847
expected_code = """import numpy as np
1849-
print("Hello world")
18501848
1849+
print("Hello world")
18511850
a=3
18521851
print("Hello world")
18531852
def some_fn():
@@ -1922,8 +1921,8 @@ def new_function2(value):
19221921
print("Hello world")
19231922
"""
19241923
expected_code = """import numpy as np
1925-
print("Hello world")
19261924
1925+
print("Hello world")
19271926
a=2
19281927
print("Hello world")
19291928
def some_fn():
@@ -1999,8 +1998,8 @@ def new_function2(value):
19991998
print("Hello world")
20001999
"""
20012000
expected_code = """import numpy as np
2002-
print("Hello world")
20032001
2002+
print("Hello world")
20042003
a=3
20052004
print("Hello world")
20062005
def some_fn():
@@ -2082,8 +2081,8 @@ def new_function2(value):
20822081
print("Hello world")
20832082
"""
20842083
expected_code = """import numpy as np
2085-
print("Hello world")
20862084
2085+
print("Hello world")
20872086
if 2<3:
20882087
a=4
20892088
else:

0 commit comments

Comments
 (0)