Yaltik/hy
2
0
Fork 0
Code Pull Requests Releases Activity

Test comprehension scoping

Browse Source
This commit is contained in:
Kodi Arfer 2018-06-07 11:08:17 -07:00
parent 4754b152a9
commit df4e49ec94
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/native_tests/comprehensions.hy
View File

@ -1,6 +1,7 @@
(import
types
pytest)
pytest
[hy._compat [PY3]])
(defn test-comprehension-types []
@ -124,6 +125,30 @@
(assert (= l [0 1 2 3 4 5])))
(defn test-scoping []
(setv x 0)
(for [x [1 2 3]])
(assert (= x 3))
; An `lfor` that gets compiled to a real comprehension
(setv x 0)
(assert (= (lfor x [1 2 3] (inc x)) [2 3 4]))
(assert (= x (if PY3 0 3)))
; Python 2 list comprehensions leak their variables.
; An `lfor` that gets compiled to a loop
(setv x 0 l [])
(assert (= (lfor x [4 5 6] :do (.append l 1) (inc x)) [5 6 7]))
(assert (= l [1 1 1]))
(assert (= x 0))
; An `sfor` that gets compiled to a real comprehension
(setv x 0)
(assert (= (sfor x [1 2 3] (inc x)) #{2 3 4}))
(assert (= x 0)))
(defn test-for-loop []
"NATIVE: test for loops"
(setv count1 0 count2 0)