Merge branch 'master' of github.com:micropython/micropython

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/basics/comprehension1.py b/tests/basics/comprehension1.py
new file mode 100644
index 000000000..7f541ee53
--- /dev/null
+++ b/tests/basics/comprehension1.py
@@ -0,0 +1,20 @@
+def f():
+    # list comprehension
+
+    print([a + 1 for a in range(5)])
+    print([(a, b) for a in range(3) for b in range(2)])
+    print([a * 2 for a in range(7) if a > 3])
+
+    print([a for a in [1, 3, 5]])
+    print([a for a in [a for a in range(4)]])
+
+    # dict comprehension
+
+    d = {a : 2 * a for a in range(5)}
+    print(d[0], d[1], d[2], d[3], d[4])
+
+    # set comprehension
+
+    print({a for a in range(5)})
+
+f()