add low python version compability.

myint · Nov 4, 2020 · e5a0a29 · e5a0a29
1 parent e166163
commit e5a0a29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/test_unify.py b/test_unify.py
@@ -6,6 +6,7 @@
 
 import contextlib
 import io
+import sys
 import tempfile
 from textwrap import dedent
 
@@ -131,6 +132,7 @@ def test_backslash_train(self):
         self.assertEqual(result, '''"\\\\'a"''')
 
 
+@unittest.skipIf(sys.version_info < (3, 6), 'Low python version')
 class TestUnitsSimpleQuotedFormatString(unittest.TestCase):
 
     def test_no_quote_in_expression_area(self):
@@ -274,6 +276,7 @@ def test_no_change(self):
         self.assertEqual(result, '''b"""foo"""''')
 
 
+@unittest.skipIf(sys.version_info < (3, 6), 'Low python version')
 class TestFstringParser(unittest.TestCase):
 
     def test_find_expression_areas(self):

diff --git a/unify.py b/unify.py
@@ -452,11 +452,10 @@ def get_editable_string(token_type, token_string):
             text_has_single = parser.text_has_single_quote()
             text_has_double = parser.text_has_double_quote()
             if not text_has_single and not text_has_double:
-                return SimpleEscapeFormatString(
-                    **parsed_string,
-                    parsed_body=parser.parsed_body,
-                    expr_ids=parser.expression_ids
-                )
+                params = parsed_string.copy()
+                params['parsed_body'] = parser.parsed_body
+                params['expr_ids'] = parser.expression_ids
+                return SimpleEscapeFormatString(**params)
         return ImmutableString(token_string)
     if 'f' in parsed_string['prefix'].lower():
         parser = FstringParser(parsed_string['body'])