From 94a77b6cf7697027d6795534029bbc3a66b670f4 Mon Sep 17 00:00:00 2001 From: rlbr Date: Wed, 27 Nov 2024 15:38:56 -0600 Subject: [PATCH] Added test with fstring --- ropetest/refactor/extracttest.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/ropetest/refactor/extracttest.py b/ropetest/refactor/extracttest.py index 4b34d58d..54fc83bd 100644 --- a/ropetest/refactor/extracttest.py +++ b/ropetest/refactor/extracttest.py @@ -108,6 +108,28 @@ def extracted(): """) self.assertEqual(expected, refactored) + def test_extract_function_with_fstring(self): + code = dedent("""\ + def main(): + h = 1 + g = f"#{h}" + print(g) + """) + start, end = self._convert_line_range_to_offset(code, 3, 3) + refactored = self.do_extract_method(code, start, end, "extracted") + + expected = dedent("""\ + def main(): + h = 1 + g = extracted(h) + print(g) + + def extracted(h): + g = f"#{h}" + return g + """) + self.assertEqual(expected, refactored) + def test_extract_function_containing_dict_generalized_unpacking(self): code = dedent("""\ def a_func(dict1):