hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-20 18:56:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

Python: Add tests for iterable unpacking

Browse Source 
in for-iterations and comprehensions.
This commit is contained in:
Rasmus Lerchedahl Petersen
2021-01-18 09:36:13 +01:00
parent 175e43d6f2
commit 66426bf0cc
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
python/ql/test/experimental/dataflow/coverage/test.py
View File

@@ -200,6 +200,12 @@ def test_nested_comprehension_paren():
SINK(x[0])
# Iterable unpacking in comprehensions
def test_unpacking_comprehension():
x = [a for (a, b) in [(SOURCE, NONSOURCE)]]
SINK(x[0]) # Flow missing
# 6.2.8. Generator expressions
def test_generator():
x = (SOURCE for y in [NONSOURCE])
@@ -620,6 +626,23 @@ def test_iterated_unpacking_assignment_conversion():
SINK_F(b[0])
@expects(4)
def test_iterable_unpacking_in_for():
tl = [(SOURCE, NONSOURCE), (SOURCE, NONSOURCE)]
for x,y in tl:
SINK(x) # Flow missing
SINK_F(y)
@expects(6)
def test_iterable_star_unpacking_in_for():
tl = [(SOURCE, NONSOURCE), (SOURCE, NONSOURCE)]
for *x,y in tl:
SINK_F(x)
SINK(x[0]) # Flow missing
SINK_F(y)
def test_deep_callgraph():
# port of python/ql/test/library-tests/taint/general/deep.py