Merge pull request #197 from azdkj532/fix-slash-equal

Fix slash equal in fix_classic_division

Author: takluyver

libmodernize/fixes/fix_classic_division.py

@@ -8,17 +8,23 @@
 
 
 class FixClassicDivision(fixer_base.BaseFix):
-    _accept_type = token.SLASH
+    PATTERN = '''
+    '/=' | '/'
+    '''
 
     def start_tree(self, tree, name):
         super(FixClassicDivision, self).start_tree(tree, name)
         self.skip = "division" in tree.future_features
 
     def match(self, node):
-        return node.value == "/"
+        return node.value in ('/', '/=')
 
     def transform(self, node, results):
         if self.skip:
             return
         libmodernize.add_future(node, u'division')
-        return pytree.Leaf(token.SLASH, "//", prefix=node.prefix)
+
+        if node.value == '/':
+            return pytree.Leaf(token.DOUBLESLASH, '//', prefix=node.prefix)
+        else:
+            return pytree.Leaf(token.DOUBLESLASHEQUAL, '//=', prefix=node.prefix)

tests/test_fix_classic_division.py

@@ -4,20 +4,24 @@
 
 
 CLASSIC_DIVISION = ("""\
+a /= 3
 1 / 2
 """,
 """\
 from __future__ import division
+a //= 3
 1 // 2
 """)
 
 NEW_DIVISION = ("""\
 from __future__ import division
 1 / 2
+a /= 3
 """,
 """\
 from __future__ import division
 1 / 2
+a /= 3
 """)