1 # tea functions
2
3 #### setvar f()[2] = 42 (setitem)
4 shopt -s oil:all
5 shopt -s parse_tea
6
7 var mylist = [1,2,3]
8 func f() {
9 return mylist
10 }
11 setvar f()[2] = 42
12 write @mylist
13 ## STDOUT:
14 1
15 2
16 42
17 ## END
18
19 #### Untyped func
20 shopt -s oil:all
21 shopt -s parse_tea
22
23 func add(x, y) Int {
24 echo 'hi'
25 return x + y
26 }
27 var result = add(42, 1)
28 echo $result
29 ## STDOUT:
30 hi
31 43
32 ## END
33
34 #### Typed func
35 shopt -s oil:all
36 shopt -s parse_tea
37
38 func add(x Int, y Int) Int {
39 echo 'hi'
40 return x+y
41 }
42 var result = add(42, 1)
43 echo $result
44 ## STDOUT:
45 hi
46 43
47 ## END
48
49 #### func: default values for positional params
50 shopt -s oil:all
51 shopt -s parse_tea
52
53 func add(x Int, y=1, z=1+2*3) {
54 return x + y + z
55 }
56 echo $[add(3)]
57 echo $[add(3,4)]
58 ## STDOUT:
59 11
60 14
61 ## END
62
63 #### pass too many positional params to func (without spread)
64 shopt -s oil:all
65 shopt -s parse_tea
66
67 func add(x, y) {
68 return x + y
69 }
70 var f = add(1,2)
71 echo f=$f
72 var f = add(1,2,3)
73 echo $f
74 ## status: 1
75 ## STDOUT:
76 f=3
77 ## END
78
79 #### Positional Spread
80 shopt -s oil:all
81 shopt -s parse_tea
82
83 func add(x, y, ...args) {
84 pp cell ':args' # This works in Tea too for debugging
85 return x + y
86 }
87 var args = %[5 6 7 8]
88 var y = add(...args)
89 echo y=$y
90 ## STDOUT:
91 7
92 8
93 y=11
94 ## END
95
96 #### pass named arg to func
97 shopt -s oil:all
98 shopt -s parse_tea
99
100 func f(; x=42) {
101 echo $x
102 }
103 _ f()
104 _ f(x=99)
105 ## STDOUT:
106 42
107 99
108 ## END
109
110 #### Func with missing named param with no default
111 shopt -s oil:all
112 shopt -s parse_tea
113
114 func add(x Int, y Int ; verbose Bool) {
115 if (verbose) {
116 echo 'verbose'
117 }
118 return x + y
119 }
120 var a = add(3, 2, verbose=true)
121 echo $a
122
123 # crashes
124 var a = add(3, 2)
125 echo "shouldn't get here"
126 ## status: 1
127 ## STDOUT:
128 verbose
129 5
130 ## END
131
132 #### Func passed wrong named param
133 shopt -s oil:all
134 shopt -s parse_tea
135
136 func add(x, y) {
137 return x + y
138 }
139 var x = add(2, 3)
140 echo x=$x
141 var y = add(2, 3, verbose=true)
142
143 ## status: 1
144 ## STDOUT:
145 x=5
146 ## END
147
148
149 #### Named Spread
150 shopt -s oil:all
151 shopt -s parse_tea
152
153 func add(x, y; verbose=true, ...named) {
154 if (verbose) { echo 'verbose' }
155
156 # Can list splatting work in tea? Maybe it should be
157 # x = splice(named)
158 # pp cell 'x'
159 # Or just
160 # = named
161 # = splice(named)
162 write @named | sort
163 return x + y
164 }
165 var args = {verbose: false, a: 1, b: 2}
166 var args2 = {f: 3}
167 var ret = add(2, 3; ...args, ...args2)
168 echo ret=$ret
169 ## STDOUT:
170 a
171 b
172 f
173 ret=5
174 ## END
175
176 #### Func with varargs
177 shopt -s oil:all
178 shopt -s parse_tea
179
180 func printf(fmt, ...args) {
181 = fmt
182 # Should be a LIST
183 = args
184 }
185 _ printf('foo', 'a', 42, null)
186
187 ## STDOUT:
188 (Str) 'foo'
189 (Tuple) ('a', 42, None)
190 ## END
191
192 #### return expression then return builtin
193 shopt -s parse_tea
194
195 func f(x) {
196 return x + 2*3
197 }
198 # this goes in proc
199 f() {
200 local x=42
201 return $x
202 }
203 var x = f(36)
204 echo x=$x
205 f
206 echo status=$?
207 ## STDOUT:
208 x=42
209 status=42
210 ## END
211
212 #### inline function calls with spread, named args, etc.
213 shopt -s oil:all
214
215 func f(a, b=0, ...args; c, d=0, ...named) {
216 write __ args: @args
217 write __ named:
218 write @named | sort
219 if (named) {
220 return [a, b, c, d]
221 } else {
222 return a + b + c + d
223 }
224 }
225 var a = [42, 43]
226 var n = %{x: 99, y: 100}
227
228 echo ____
229 write string $[f(0, 1, ...a, c=2, d=3)]
230
231 # Now get a list back
232 echo ____
233 write array @[f(5, 6, ...a, c=7, d=8; ...n)]
234
235 ## STDOUT:
236 ____
237 __
238 args:
239 42
240 43
241 __
242 named:
243
244 string
245 6
246 ____
247 __
248 args:
249 42
250 43
251 __
252 named:
253 x
254 y
255 array
256 5
257 6
258 7
259 8
260 ## END
261
262 #### basic lambda
263 var f = |x| x+1
264 var y = f(0)
265 echo $y
266 echo $[f(42)]
267 ## STDOUT:
268 1
269 43
270 ## END
271
272 #### Nested functions ??? Bug from Till.
273 shopt -s oil:all
274 shopt -s parse_tea
275
276 # I think we should disallow these.
277 # The bug is related to return as a keyword!!! Which we will get rid of in the
278 # newer Tea parser.
279
280 func returnsString() {
281 func unrelated() {return "hm?"}
282 return "test" # This fails
283 }
284
285 = returnsString()
286
287 ## STDOUT:
288 ## END