``is_generator`` correctly considers `Yield` nodes in `AugAssign` nodes

This fixes a false positive with the `assignment-from-no-return` pylint check.

Close PyCQA/pylint#3904

3 files changed, 22 insertions, 0 deletions

diff --git a/ChangeLog b/ChangeLog
index a28029c2..de28a9c0 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -98,6 +98,12 @@ Release Date: TBA
 
   Close PyCQA/pylint#3686
 
+* ``is_generator`` correctly considers `Yield` nodes in `AugAssign` nodes
+
+  This fixes a false positive with the `assignment-from-no-return` pylint check.
+
+  Close PyCQA/pylint#3904
+
 
 What's New in astroid 2.4.2?
 ============================
diff --git a/astroid/node_classes.py b/astroid/node_classes.py
index 86529e5a..62438e62 100644
--- a/astroid/node_classes.py
+++ b/astroid/node_classes.py
@@ -2125,6 +2125,11 @@ class AugAssign(mixins.AssignTypeMixin, Statement):
         yield self.target
         yield self.value
 
+    def _get_yield_nodes_skip_lambdas(self):
+        """An AugAssign node can contain a Yield node in the value"""
+        yield from self.value._get_yield_nodes_skip_lambdas()
+        yield from super()._get_yield_nodes_skip_lambdas()
+
 
 class Repr(NodeNG):
     """Class representing an :class:`ast.Repr` node.
diff --git a/tests/unittest_nodes.py b/tests/unittest_nodes.py
index 5b6a39e3..d138ee17 100644
--- a/tests/unittest_nodes.py
+++ b/tests/unittest_nodes.py
@@ -1360,5 +1360,16 @@ def test_is_generator_for_yield_in_if():
     assert bool(node.is_generator())
 
 
+def test_is_generator_for_yield_in_aug_assign():
+    code = """
+    def test():
+        buf = ''
+        while True:
+            buf += yield
+    """
+    node = astroid.extract_node(code)
+    assert bool(node.is_generator())
+
+
 if __name__ == "__main__":
     unittest.main()