1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
(import
asyncio
unittest.mock [Mock]
pytest
tests.resources.pydemo [AsyncWithTest async-exits])
(defn test-context []
(with [fd (open "tests/resources/text.txt" "r")] (assert fd))
(with [(open "tests/resources/text.txt" "r")] (do)))
(defn test-with-return []
(defn read-file [filename]
(with [fd (open filename "r" :encoding "UTF-8")] (.read fd)))
(assert (= (read-file "tests/resources/text.txt") "TAARGÜS TAARGÜS\n")))
(setv exits [])
(defclass WithTest [object]
(defn __init__ [self val]
(setv self.val val))
(defn __enter__ [self]
self.val)
(defn __exit__ [self type value traceback]
(.append exits self.val)))
(defn test-single-with []
(setv (cut exits) [])
(with [t (WithTest 1)]
(setv out t))
(assert (= out 1))
(assert (= exits [1])))
(defn test-quince-with []
(setv (cut exits) [])
(with [t1 (WithTest 1) t2 (WithTest 2) t3 (WithTest 3) _ (WithTest 4)]
(setv out [t1 t2 t3]))
(assert (= out [1 2 3]))
(assert (= exits [4 3 2 1])))
(defn test-single-with-async []
(.clear async-exits)
(setv out [])
(asyncio.run
((fn :async []
(with [:async t (AsyncWithTest 1)]
(.append out t)))))
(assert (= out [1]))
(assert (= async-exits [1])))
(defn test-quince-with-async []
(.clear async-exits)
(setv out [])
(asyncio.run
((fn :async []
(with [
:async t1 (AsyncWithTest 1)
:async t2 (AsyncWithTest 2)
:async t3 (AsyncWithTest 3)
:async _ (AsyncWithTest 4)]
(.extend out [t1 t2 t3])))))
(assert (= out [1 2 3]))
(assert (= async-exits [4 3 2 1])))
(defn test-with-mixed-async []
(setv (cut exits) [])
(setv out [])
(asyncio.run
((fn :async []
(with [:async t1 (AsyncWithTest 1)
t2 (WithTest 2)
:async t3 (AsyncWithTest 3)
_ (WithTest 4)]
(.extend out [t1 t2 t3])))))
(assert (= out [1 2 3])))
(defn test-unnamed-context-with []
"`_` get compiled to unnamed context"
(with [_ (WithTest 1)
[b d] (WithTest (range 2 5 2))
_ (WithTest 3)]
(assert (= [b d] [2 4]))
(with [(pytest.raises NameError)]
_)))
(defclass SuppressZDE [object]
(defn __enter__ [self])
(defn __exit__ [self exc-type exc-value traceback]
(and (is-not exc-type None) (issubclass exc-type ZeroDivisionError))))
(defn test-exception-suppressing-with []
; https://github.com/hylang/hy/issues/1320
(setv x (with [(SuppressZDE)] 5))
(assert (= x 5))
(setv y (with [(SuppressZDE)] (/ 1 0)))
(assert (is y None))
(setv z (with [(SuppressZDE)] (/ 1 0) 5))
(assert (is z None))
(defn f [] (with [(SuppressZDE)] (/ 1 0)))
(assert (is (f) None))
(setv w 7 l [])
(setv w (with [(SuppressZDE)] (.append l w) (/ 1 0) 5))
(assert (is w None))
(assert (= l [7])))
(defn test-statements []
(setv m (Mock))
(with [t (do (m) (WithTest 2))]
(setv out t))
(assert (= m.call-count 1))
(assert (= out 2))
; https://github.com/hylang/hy/issues/2605
(with [t1 (WithTest 1) t2 (do (setv foo t1) (WithTest 2))]
(setv out [t1 t2]))
(assert (= out [1 2]))
(assert (= foo 1)))
|
