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

Close PyCQA/pylint#3583

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/unittest_nodes.py b/tests/unittest_nodes.py
index 07733e5f..5b6a39e3 100644
--- a/tests/unittest_nodes.py
+++ b/tests/unittest_nodes.py
@@ -1347,5 +1347,18 @@ def test_is_generator_for_yield_in_while():
     assert bool(node.is_generator())
 
 
+def test_is_generator_for_yield_in_if():
+    code = """
+    import asyncio
+
+    def paused_iter(iterable):
+        if (yield from asyncio.sleep(0.01)):
+            pass
+            return
+    """
+    node = astroid.extract_node(code)
+    assert bool(node.is_generator())
+
+
 if __name__ == "__main__":
     unittest.main()