Add a test for function with comprehension or generator

Or a function returning a generator.

Author: Pierre-Sassoulas

tests/functional/l/len_checks.py

@@ -158,5 +158,18 @@ def function_returning_int(r):
             return 1
         return 2
 
-    assert len(function_returning_list(z)) # [len-as-condition]
+    def function_returning_generator(r):
+        for i in [r, 1, 2, 3]:
+            yield i
+
+    def function_returning_comprehension(r):
+        return [x+1 for x in [r, 1, 2, 3]]
+
+    def function_returning_function(r):
+        return function_returning_generator(r)
+
+    assert len(function_returning_list(z))  # [len-as-condition]
     assert len(function_returning_int(z))
+    assert len(function_returning_generator(z))  # [len-as-condition]
+    assert len(function_returning_comprehension(z))  # [len-as-condition]
+    assert len(function_returning_function(z))  # [len-as-condition]

tests/functional/l/len_checks.txt

@@ -22,3 +22,5 @@ len-as-condition:129:github_issue_1879:Do not use `len(SEQUENCE)` without compar
 len-as-condition:130:github_issue_1879:Do not use `len(SEQUENCE)` without comparison to determine if a sequence is empty
 len-as-condition:131:github_issue_1879:Do not use `len(SEQUENCE)` without comparison to determine if a sequence is empty
 len-as-condition:161:github_issue_1879:Do not use `len(SEQUENCE)` without comparison to determine if a sequence is empty
+len-as-condition:170:github_issue_1879:Do not use `len(SEQUENCE)` without comparison to determine if a sequence is empty
+len-as-condition:171:github_issue_1879:Do not use `len(SEQUENCE)` without comparison to determine if a sequence is empty