qa: division_support

my2to3/fixes/fix_division_support.py

@@ -41,7 +41,8 @@ class FixDivisionSupport(BaseFix):
             """
 
   def transform(self, node, results):
-    filename, lineno, id_ = [l.value for l in node.children[1].children[1].children[:-4:2]]
+    args = results['args']
+    filename, lineno, id_ = [l.value for l in args.children[:-4:2]]
     should_change, status = analyze_data(
       get_data(
         "division_trace",
@@ -55,14 +56,14 @@ class FixDivisionSupport(BaseFix):
       operator = Leaf(lib2to3.pgen2.token.DOUBLESLASH, "//")
     else:
       operator = Leaf(lib2to3.pgen2.token.SLASH, "/")
-    operator.prefix = node.children[1].children[1].children[3].prefix
+    operator.prefix = args.children[3].prefix
 
     node.replace(
       Node(
         syms.term,
         [
-          node.children[1].children[1].children[2].clone(),
+          args.children[2].clone(),
           operator,
-          node.children[1].children[1].children[4].clone()
+          args.children[4].clone()
         ],
         prefix=node.prefix))