Skip to content

Commit b78c976

Browse files
committed
Added test with fstring
1 parent a32584f commit b78c976

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

ropetest/refactor/extracttest.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,28 @@ def extracted():
108108
""")
109109
self.assertEqual(expected, refactored)
110110

111+
def test_extract_function_with_fstring(self):
112+
code = dedent("""\
113+
def main():
114+
h = 1
115+
g = f"#{h}"
116+
print(g)
117+
""")
118+
start, end = self._convert_line_range_to_offset(code, 3, 3)
119+
refactored = self.do_extract_method(code, start, end, "extracted")
120+
121+
expected = dedent("""\
122+
def main():
123+
h = 1
124+
g = extracted(h)
125+
print(g)
126+
127+
def extracted(h):
128+
g = f"#{h}"
129+
return g
130+
""")
131+
self.assertEqual(expected, refactored)
132+
111133
def test_extract_function_containing_dict_generalized_unpacking(self):
112134
code = dedent("""\
113135
def a_func(dict1):

0 commit comments

Comments
 (0)