Merge lp:~pyflakes-dev/pyflakes/1108904-augmented-assignment into lp:pyflakes

[Florent]

Fix for augmented assignments

[lp:~pyflakes-dev/pyflakes/1108904-augmented-assignment]

68. By Florent on 2013-01-29

More tests with augmented assignments.

=== modified file 'pyflakes/checker.py'
--- pyflakes/checker.py	2013-01-29 08:10:46 +0000
+++ pyflakes/checker.py	2013-01-29 15:37:21 +0000
@@ -688,7 +688,7 @@
     def AUGASSIGN(self, node):
         self.handleNodeLoad(node.target)
         self.handleNode(node.value, node)
-        self.handleNodeStore(node.target)
+        self.handleNode(node.target, node)
 
     def IMPORT(self, node):
         for alias in node.names:

=== modified file 'pyflakes/test/test_undefined_names.py'
--- pyflakes/test/test_undefined_names.py	2013-01-28 23:35:39 +0000
+++ pyflakes/test/test_undefined_names.py	2013-01-29 15:37:21 +0000
@@ -220,6 +220,24 @@
                 return a
         ''', m.UndefinedLocal)
 
+    def test_undefinedAugmentedAssignment(self):
+        self.flakes(
+            '''
+            def f(seq):
+                a = 0
+                seq[a] += 1
+                seq[b] /= 2
+                c[0] *= 2
+                a -= 3
+                d += 4
+                e[any] = 5
+            ''',
+            m.UndefinedName,    # b
+            m.UndefinedName,    # c
+            m.UndefinedName, m.UnusedVariable,  # d
+            m.UndefinedName,    # e
+        )
+
     def test_nestedClass(self):
         '''nested classes can access enclosing scope'''
         self.flakes('''