`FunctionDef.is_generator` properly handles `yield` nodes in `If` tests (#799)

Close PyCQA/pylint#3583

1 files changed, 5 insertions, 0 deletions

diff --git a/astroid/node_classes.py b/astroid/node_classes.py
index 621dc5f2..8cc8585d 100644
--- a/astroid/node_classes.py
+++ b/astroid/node_classes.py
@@ -3508,6 +3508,11 @@ class If(mixins.MultiLineBlockMixin, mixins.BlockRangeMixIn, Statement):
     def has_elif_block(self):
         return len(self.orelse) == 1 and isinstance(self.orelse[0], If)
 
+    def _get_yield_nodes_skip_lambdas(self):
+        """An If node can contain a Yield node in the test"""
+        yield from self.test._get_yield_nodes_skip_lambdas()
+        yield from super()._get_yield_nodes_skip_lambdas()
+
 
 class IfExp(NodeNG):
     """Class representing an :class:`ast.IfExp` node.