Merge pull request #28 from AWhetter/fix_long_argname

Fixed long getting converted to int when used as a keyword argument
amyreese · Nov 12, 2020 · a67281e · a67281e
2 parents c52d218 + c158164
commit a67281e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/fissix/fixer_util.py b/fissix/fixer_util.py
@@ -302,7 +302,7 @@ def is_probably_builtin(node):
         # Assignment.
         return False
     if parent.type == syms.parameters or (
-        parent.type == syms.typedargslist
+        parent.type in (syms.typedargslist, syms.argument)
         and (
             (prev is not None and prev.type == token.COMMA)
             or parent.children[0] is node

diff --git a/fissix/tests/test_fixers.py b/fissix/tests/test_fixers.py
@@ -1185,6 +1185,11 @@ def test_3(self):
         a = """z = type(x) in (int, int)"""
         self.check(b, a)
 
+    def test_4(self):
+        b = """f(arg=long)"""
+        a = """f(arg=int)"""
+        self.check(b, a)
+
     def test_unchanged(self):
         s = """long = True"""
         self.unchanged(s)
@@ -1207,6 +1212,9 @@ def test_unchanged(self):
         s = """def f(x, long=True): pass"""
         self.unchanged(s)
 
+        s = """f(long=True)"""
+        self.unchanged(s)
+
     def test_prefix_preservation(self):
         b = """x =   long(  x  )"""
         a = """x =   int(  x  )"""