Make sure that assign nodes can find yield statements in their va…

…lues

Close pylint-dev/pylint#2400

ChangeLog

@@ -2,6 +2,14 @@
 astroid's ChangeLog
 ===================
 
+What's New in astroid 2.0.4?
+============================
+Release Date: TBA
+
+   * Make sure that assign nodes can find ``yield`` statements in their values
+
+     Close PyCQA/pylint#2400
+
 What's New in astroid 2.0.3?
 ============================
 

astroid/node_classes.py

@@ -1763,6 +1763,9 @@ def _get_assign_nodes(self):
 
         yield from self.value._get_assign_nodes()
 
+    def _get_yield_nodes_skip_lambdas(self):
+        yield from self.value._get_yield_nodes_skip_lambdas()
+
 
 class AnnAssign(mixins.AssignTypeMixin, Statement):
     """Class representing an :class:`ast.AnnAssign` node.

astroid/tests/unittest_nodes.py

@@ -963,5 +963,21 @@ def func2():
         assert node.type_comment_returns.as_string() == expected_returns_string
 
 
+def test_is_generator_for_yield_assignments():
+    node = astroid.extract_node('''
+    class A:
+        def test(self):
+            a = yield
+            while True:
+                print(a)
+                yield a
+    a = A()
+    a.test
+    ''')
+    inferred = next(node.infer())
+    assert isinstance(inferred, astroid.BoundMethod)
+    assert bool(inferred.is_generator())
+
+
 if __name__ == '__main__':
     unittest.main()