blob: f3bdcd4bd1ab81a70bdf20959cd7732e7ecd818c [file] [log] [blame]
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001" Tests for the List and Dict types
rbtnn0ccb5842021-12-18 18:33:46 +00002scriptencoding utf-8
Bram Moolenaarfb094e12017-11-05 20:59:28 +01003
Bram Moolenaar62aec932022-01-29 21:45:34 +00004import './vim9.vim' as v9
Bram Moolenaar63cb6562021-07-20 22:21:59 +02005
Bram Moolenaarfb094e12017-11-05 20:59:28 +01006func TearDown()
7 " Run garbage collection after every test
8 call test_garbagecollect_now()
9endfunc
10
11" Tests for List type
12
13" List creation
14func Test_list_create()
15 " Creating List directly with different types
16 let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
17 call assert_equal("[1, 'as''d', [1, 2, function('strlen')], {'a': 1}]", string(l))
18 call assert_equal({'a' : 1}, l[-1])
19 call assert_equal(1, l[-4])
20 let x = 10
21 try
22 let x = l[-5]
23 catch
24 call assert_match('E684:', v:exception)
25 endtry
26 call assert_equal(10, x)
27endfunc
28
Bram Moolenaar4d4d1cd2020-07-30 22:14:33 +020029" This was allowed in legacy Vim script
30let s:list_with_spaces = [1 , 2 , 3]
31
Bram Moolenaarfb094e12017-11-05 20:59:28 +010032" List slices
33func Test_list_slice()
34 let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
35 call assert_equal([1, 'as''d', [1, 2, function('strlen')], {'a': 1}], l[:])
36 call assert_equal(['as''d', [1, 2, function('strlen')], {'a': 1}], l[1:])
37 call assert_equal([1, 'as''d', [1, 2, function('strlen')]], l[:-2])
38 call assert_equal([1, 'as''d', [1, 2, function('strlen')], {'a': 1}], l[0:8])
39 call assert_equal([], l[8:-1])
Bram Moolenaar8b633132020-03-20 18:20:51 +010040 call assert_equal([], l[0:-10])
Bram Moolenaarea04a6e2020-04-23 13:38:02 +020041 " perform an operation on a list slice
42 let l = [1, 2, 3]
43 let l[:1] += [1, 2]
44 let l[2:] -= [1]
45 call assert_equal([2, 4, 2], l)
Bram Moolenaar92f05f22021-08-12 21:12:56 +020046
47 let lines =<< trim END
48 VAR l = [1, 2]
49 call assert_equal([1, 2], l[:])
50 call assert_equal([2], l[-1 : -1])
51 call assert_equal([1, 2], l[-2 : -1])
52 END
Bram Moolenaar62aec932022-01-29 21:45:34 +000053 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar92f05f22021-08-12 21:12:56 +020054
55 let l = [1, 2]
56 call assert_equal([], l[-3 : -1])
57
58 let lines =<< trim END
59 var l = [1, 2]
60 assert_equal([1, 2], l[-3 : -1])
61 END
Bram Moolenaar62aec932022-01-29 21:45:34 +000062 call v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananfe424d12024-05-17 18:20:43 +020063
64 call assert_fails('let l[[]] = 1', 'E730: Using a List as a String')
65 call assert_fails('let l[1 : []] = [1]', 'E730: Using a List as a String')
Bram Moolenaarfb094e12017-11-05 20:59:28 +010066endfunc
67
68" List identity
69func Test_list_identity()
Bram Moolenaar63cb6562021-07-20 22:21:59 +020070 let lines =<< trim END
71 VAR l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
72 VAR ll = l
73 VAR lx = copy(l)
74 call assert_true(l == ll)
75 call assert_false(l isnot ll)
76 call assert_true(l is ll)
77 call assert_true(l == lx)
78 call assert_false(l is lx)
79 call assert_true(l isnot lx)
80 END
Bram Moolenaar62aec932022-01-29 21:45:34 +000081 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010082endfunc
83
84" removing items with :unlet
85func Test_list_unlet()
Bram Moolenaar63cb6562021-07-20 22:21:59 +020086 let lines =<< trim END
87 VAR l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
88 unlet l[2]
89 call assert_equal([1, 'as''d', {'a': 1}], l)
90 LET l = range(8)
91 unlet l[: 3]
92 unlet l[1 :]
93 call assert_equal([4], l)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010094
Bram Moolenaar63cb6562021-07-20 22:21:59 +020095 #" removing items out of range: silently skip items that don't exist
96 LET l = [0, 1, 2, 3]
97 unlet l[2 : 2]
98 call assert_equal([0, 1, 3], l)
99 LET l = [0, 1, 2, 3]
100 unlet l[2 : 3]
101 call assert_equal([0, 1], l)
102 LET l = [0, 1, 2, 3]
103 unlet l[2 : 4]
104 call assert_equal([0, 1], l)
105 LET l = [0, 1, 2, 3]
106 unlet l[2 : 5]
107 call assert_equal([0, 1], l)
108 LET l = [0, 1, 2, 3]
109 unlet l[-2 : 2]
110 call assert_equal([0, 1, 3], l)
111 LET l = [0, 1, 2, 3]
112 unlet l[-3 : 2]
113 call assert_equal([0, 3], l)
114 LET l = [0, 1, 2, 3]
115 unlet l[-4 : 2]
116 call assert_equal([3], l)
117 LET l = [0, 1, 2, 3]
118 unlet l[-5 : 2]
119 call assert_equal([3], l)
120 LET l = [0, 1, 2, 3]
121 unlet l[-6 : 2]
122 call assert_equal([3], l)
123 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000124 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200125
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100126 let l = [0, 1, 2, 3]
127 unlet l[2:2]
128 call assert_equal([0, 1, 3], l)
129 let l = [0, 1, 2, 3]
130 unlet l[2:3]
131 call assert_equal([0, 1], l)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200132
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200133 let lines =<< trim END
134 VAR l = [0, 1, 2, 3]
135 unlet l[2 : 1]
136 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000137 call v9.CheckLegacyAndVim9Failure(lines, 'E684:')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200138
139 let lines =<< trim END
140 VAR l = [0, 1, 2, 3]
141 unlet l[-1 : 2]
142 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000143 call v9.CheckLegacyAndVim9Failure(lines, 'E684:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100144endfunc
145
146" assignment to a list
147func Test_list_assign()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200148 let lines =<< trim END
149 VAR l = [0, 1, 2, 3]
150 VAR va = 0
151 VAR vb = 0
152 LET [va, vb] = l[2 : 3]
153 call assert_equal([2, 3], [va, vb])
154 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000155 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200156
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200157 let lines =<< trim END
158 let l = [0, 1, 2, 3]
159 let [va, vb] = l
160 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000161 call v9.CheckScriptFailure(lines, 'E687:')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200162 let lines =<< trim END
163 var l = [0, 1, 2, 3]
164 var va = 0
165 var vb = 0
166 [va, vb] = l
167 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000168 call v9.CheckScriptFailure(['vim9script'] + lines, 'E687:')
169 call v9.CheckDefExecFailure(lines, 'E1093: Expected 2 items but got 4')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200170
171 let lines =<< trim END
172 let l = [0, 1, 2, 3]
173 let [va, vb] = l[1:1]
174 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000175 call v9.CheckScriptFailure(lines, 'E688:')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200176 let lines =<< trim END
177 var l = [0, 1, 2, 3]
178 var va = 0
179 var vb = 0
180 [va, vb] = l[1 : 1]
181 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000182 call v9.CheckScriptFailure(['vim9script'] + lines, 'E688:')
183 call v9.CheckDefExecFailure(lines, 'E1093: Expected 2 items but got 1')
Yegappan Lakshmananfe424d12024-05-17 18:20:43 +0200184
185 let lines =<< trim END
186 VAR l = [2]
187 LET l += test_null_list()
188 call assert_equal([2], l)
189 LET l = test_null_list()
190 LET l += [1]
191 call assert_equal([1], l)
192 END
193 call v9.CheckLegacyAndVim9Success(lines)
194
195 let d = {'abc': [1, 2, 3]}
196 call assert_fails('let d.abc[0:0z10] = [10, 20]', 'E976: Using a Blob as a String')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100197endfunc
198
199" test for range assign
200func Test_list_range_assign()
Bram Moolenaar4f0884d2021-08-11 21:49:23 +0200201 let lines =<< trim END
202 VAR l = [0]
203 LET l[:] = [1, 2]
204 call assert_equal([1, 2], l)
205 LET l[-4 : -1] = [5, 6]
206 call assert_equal([5, 6], l)
207 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000208 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar89071cb2021-08-13 18:20:09 +0200209
210 let lines =<< trim END
211 var l = [7]
212 l[:] = ['text']
213 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000214 call v9.CheckDefAndScriptFailure(lines, 'E1012:', 2)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100215endfunc
216
Bram Moolenaar976f8592022-08-30 14:34:52 +0100217func Test_list_items()
218 let r = []
219 let l = ['a', 'b', 'c']
220 for [idx, val] in items(l)
221 call extend(r, [[idx, val]])
222 endfor
223 call assert_equal([[0, 'a'], [1, 'b'], [2, 'c']], r)
224
Bram Moolenaar3e518a82022-08-30 17:45:33 +0100225 call assert_fails('call items(3)', 'E1225:')
226endfunc
227
228func Test_string_items()
229 let r = []
230 let s = 'ábツ'
231 for [idx, val] in items(s)
232 call extend(r, [[idx, val]])
233 endfor
234 call assert_equal([[0, 'á'], [1, 'b'], [2, 'ツ']], r)
Bram Moolenaar976f8592022-08-30 14:34:52 +0100235endfunc
236
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200237" Test removing items in list
238func Test_list_func_remove()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200239 let lines =<< trim END
240 #" Test removing 1 element
241 VAR l = [1, 2, 3, 4]
242 call assert_equal(1, remove(l, 0))
243 call assert_equal([2, 3, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200244
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200245 LET l = [1, 2, 3, 4]
246 call assert_equal(2, remove(l, 1))
247 call assert_equal([1, 3, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200248
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200249 LET l = [1, 2, 3, 4]
250 call assert_equal(4, remove(l, -1))
251 call assert_equal([1, 2, 3], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200252
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200253 #" Test removing range of element(s)
254 LET l = [1, 2, 3, 4]
255 call assert_equal([3], remove(l, 2, 2))
256 call assert_equal([1, 2, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200257
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200258 LET l = [1, 2, 3, 4]
259 call assert_equal([2, 3], remove(l, 1, 2))
260 call assert_equal([1, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200261
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200262 LET l = [1, 2, 3, 4]
263 call assert_equal([2, 3], remove(l, -3, -2))
264 call assert_equal([1, 4], l)
265 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000266 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200267
268 " Test invalid cases
269 let l = [1, 2, 3, 4]
270 call assert_fails("call remove(l, 5)", 'E684:')
271 call assert_fails("call remove(l, 1, 5)", 'E684:')
272 call assert_fails("call remove(l, 3, 2)", 'E16:')
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +0100273 call assert_fails("call remove(1, 0)", 'E896:')
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200274 call assert_fails("call remove(l, l)", 'E745:')
275endfunc
276
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200277" List add() function
278func Test_list_add()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200279 let lines =<< trim END
280 VAR l = []
281 call add(l, 1)
282 call add(l, [2, 3])
283 call add(l, [])
284 call add(l, test_null_list())
285 call add(l, {'k': 3})
286 call add(l, {})
287 call add(l, test_null_dict())
288 call assert_equal([1, [2, 3], [], [], {'k': 3}, {}, {}], l)
289 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000290 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200291
292 " weird legacy behavior
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200293 call assert_equal(1, add(test_null_list(), 4))
294endfunc
295
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100296" Tests for Dictionary type
297
298func Test_dict()
299 " Creating Dictionary directly with different types
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200300 let lines =<< trim END
301 VAR d = {'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}, }
302 call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
303 call assert_equal('asd', d.1)
304 call assert_equal(['-1', '1', 'b'], sort(keys(d)))
305 call assert_equal(['asd', [1, 2, function('strlen')], {'a': 1}], values(d))
306 call extend(d, {3: 33, 1: 99})
307 call extend(d, {'b': 'bbb', 'c': 'ccc'}, "keep")
308 call assert_equal({'c': 'ccc', '1': 99, 'b': [1, 2, function('strlen')], '3': 33, '-1': {'a': 1}}, d)
309 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000310 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200311
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100312 let d = {001: 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1},}
313 call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200314
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100315 let v = []
316 for [key, val] in items(d)
317 call extend(v, [key, val])
318 unlet key val
319 endfor
320 call assert_equal(['1','asd','b',[1, 2, function('strlen')],'-1',{'a': 1}], v)
321
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200322 call extend(d, {3: 33, 1: 99})
Bram Moolenaare2e40752020-09-04 21:18:46 +0200323 call assert_fails("call extend(d, {3:333,4:444}, 'error')", 'E737:')
Bram Moolenaar08f41572020-04-20 16:50:00 +0200324
325 " duplicate key
326 call assert_fails("let d = {'k' : 10, 'k' : 20}", 'E721:')
327 " missing comma
328 call assert_fails("let d = {'k' : 10 'k' : 20}", 'E722:')
329 " missing curly brace
330 call assert_fails("let d = {'k' : 10,", 'E723:')
331 " invalid key
332 call assert_fails('let d = #{++ : 10}', 'E15:')
333 " wrong type for key
334 call assert_fails('let d={[] : 10}', 'E730:')
335 " undefined variable as value
336 call assert_fails("let d={'k' : i}", 'E121:')
Bram Moolenaar98cb90e2021-11-30 11:56:22 +0000337
338 " allow key starting with number at the start, not a curly expression
339 call assert_equal({'1foo': 77}, #{1foo: 77})
ii147c7e1e92022-09-07 19:40:17 +0100340
341 " #{expr} is not a curly expression
342 let x = 'x'
343 call assert_equal(#{g: x}, #{g:x})
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100344endfunc
345
Bram Moolenaar4d4d1cd2020-07-30 22:14:33 +0200346" This was allowed in legacy Vim script
347let s:dict_with_spaces = {'one' : 1 , 'two' : 2 , 'three' : 3}
348let s:dict_with_spaces_lit = #{one : 1 , two : 2 , three : 3}
349
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100350" Dictionary identity
351func Test_dict_identity()
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200352 let lines =<< trim END
353 VAR d = {'1': 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1}, }
354 VAR dd = d
355 VAR dx = copy(d)
356 call assert_true(d == dd)
357 call assert_false(d isnot dd)
358 call assert_true(d is dd)
359 call assert_true(d == dx)
360 call assert_false(d is dx)
361 call assert_true(d isnot dx)
362 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000363 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100364endfunc
365
366" removing items with :unlet
367func Test_dict_unlet()
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200368 let lines =<< trim END
369 VAR d = {'b': 'bbb', '1': 99, '3': 33, '-1': {'a': 1}}
370 unlet d.b
371 unlet d[-1]
372 call assert_equal({'1': 99, '3': 33}, d)
373 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000374 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100375endfunc
376
377" manipulating a big Dictionary (hashtable.c has a border of 1000 entries)
378func Test_dict_big()
379 let d = {}
380 for i in range(1500)
381 let d[i] = 3000 - i
382 endfor
383 call assert_equal([3000, 2900, 2001, 1600, 1501], [d[0], d[100], d[999], d[1400], d[1499]])
384 let str = ''
385 try
386 let n = d[1500]
387 catch
Bram Moolenaar6d967122020-10-30 19:06:18 +0100388 let str = substitute(v:exception, '\v(.{14}).*( "\d{4}").*', '\1\2', '')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100389 endtry
Bram Moolenaar6d967122020-10-30 19:06:18 +0100390 call assert_equal('Vim(let):E716: "1500"', str)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100391
Christian Brabandtee17b6f2023-09-09 11:23:50 +0200392 " lookup each item
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100393 for i in range(1500)
394 call assert_equal(3000 - i, d[i])
395 endfor
396 let i += 1
397
398 " delete even items
399 while i >= 2
400 let i -= 2
401 unlet d[i]
402 endwhile
403 call assert_equal('NONE', get(d, 1500 - 100, 'NONE'))
404 call assert_equal(2999, d[1])
405
406 " delete odd items, checking value, one intentionally wrong
407 let d[33] = 999
408 let i = 1
409 while i < 1500
410 if i != 33
411 call assert_equal(3000 - i, d[i])
412 else
413 call assert_equal(999, d[i])
414 endif
415 unlet d[i]
416 let i += 2
417 endwhile
418 call assert_equal({}, d)
419 unlet d
420endfunc
421
422" Dictionary function
423func Test_dict_func()
424 let d = {}
425 func d.func(a) dict
426 return a:a . len(self.data)
427 endfunc
428 let d.data = [1,2,3]
429 call assert_equal('len: 3', d.func('len: '))
430 let x = d.func('again: ')
431 call assert_equal('again: 3', x)
432 let Fn = d.func
433 call assert_equal('xxx3', Fn('xxx'))
434endfunc
435
Bram Moolenaarb13ab992020-07-27 21:43:28 +0200436func Test_dict_assign()
437 let d = {}
438 let d.1 = 1
439 let d._ = 2
440 call assert_equal({'1': 1, '_': 2}, d)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +0200441
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200442 let lines =<< trim END
443 VAR d = {}
444 LET d.a = 1
445 LET d._ = 2
446 call assert_equal({'a': 1, '_': 2}, d)
447 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000448 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200449
450 let lines =<< trim END
451 let n = 0
452 let n.key = 3
453 END
Ernie Raelf8da3242023-10-11 21:22:12 +0200454 call v9.CheckScriptFailure(lines, 'E1203: Dot not allowed after a number: n.key = 3')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200455 let lines =<< trim END
456 vim9script
457 var n = 0
458 n.key = 3
459 END
Ernie Raelf8da3242023-10-11 21:22:12 +0200460 call v9.CheckScriptFailure(lines, 'E1203: Dot not allowed after a number: n.key = 3')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200461 let lines =<< trim END
462 var n = 0
463 n.key = 3
464 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000465 call v9.CheckDefFailure(lines, 'E1141:')
Yegappan Lakshmananfe424d12024-05-17 18:20:43 +0200466
467 let d = {'abc': {}}
468 call assert_fails("let d.abc[0z10] = 10", 'E976: Using a Blob as a String')
Bram Moolenaarb13ab992020-07-27 21:43:28 +0200469endfunc
470
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100471" Function in script-local List or Dict
472func Test_script_local_dict_func()
473 let g:dict = {}
474 function g:dict.func() dict
475 return 'g:dict.func' . self.foo[1] . self.foo[0]('asdf')
476 endfunc
477 let g:dict.foo = ['-', 2, 3]
478 call insert(g:dict.foo, function('strlen'))
479 call assert_equal('g:dict.func-4', g:dict.func())
480 unlet g:dict
481endfunc
482
Bram Moolenaar08f41572020-04-20 16:50:00 +0200483" Test removing items in a dictionary
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200484func Test_dict_func_remove()
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200485 let lines =<< trim END
486 VAR d = {1: 'a', 2: 'b', 3: 'c'}
487 call assert_equal('b', remove(d, 2))
488 call assert_equal({1: 'a', 3: 'c'}, d)
489 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000490 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200491
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200492 let lines =<< trim END
493 VAR d = {1: 'a', 3: 'c'}
494 call remove(d, 1, 2)
495 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000496 call v9.CheckLegacyAndVim9Failure(lines, 'E118:')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200497
498 let lines =<< trim END
499 VAR d = {1: 'a', 3: 'c'}
500 call remove(d, 'a')
501 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000502 call v9.CheckLegacyAndVim9Failure(lines, 'E716:')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200503
504 let lines =<< trim END
Bram Moolenaar5c1ec432021-11-29 13:44:55 +0000505 let d = {'a-b': 55}
506 echo d.a-b
507 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000508 call v9.CheckScriptFailure(lines, 'E716: Key not present in Dictionary: "a"')
Bram Moolenaar5c1ec432021-11-29 13:44:55 +0000509
510 let lines =<< trim END
511 vim9script
512 var d = {'a-b': 55}
513 echo d.a-b
514 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000515 call v9.CheckScriptFailure(lines, 'E716: Key not present in Dictionary: "a"')
Bram Moolenaar5c1ec432021-11-29 13:44:55 +0000516
517 let lines =<< trim END
518 var d = {'a-b': 55}
519 echo d.a-b
520 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000521 call v9.CheckDefFailure(lines, 'E1004: White space required before and after ''-''')
Bram Moolenaar5c1ec432021-11-29 13:44:55 +0000522
523 let lines =<< trim END
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200524 let d = {1: 'a', 3: 'c'}
525 call remove(d, [])
526 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000527 call v9.CheckScriptFailure(lines, 'E730:')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200528 let lines =<< trim END
529 vim9script
530 var d = {1: 'a', 3: 'c'}
531 call remove(d, [])
532 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000533 call v9.CheckScriptFailure(lines, 'E1220: String or Number required for argument 2')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200534 let lines =<< trim END
535 var d = {1: 'a', 3: 'c'}
536 call remove(d, [])
537 END
Yegappan Lakshmanan66897192023-12-05 15:51:50 +0100538 call v9.CheckDefExecFailure(lines, 'E1013: Argument 2: type mismatch, expected string but got list<any>')
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200539endfunc
540
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100541" Nasty: remove func from Dict that's being called (works)
542func Test_dict_func_remove_in_use()
543 let d = {1:1}
544 func d.func(a)
545 return "a:" . a:a
546 endfunc
547 let expected = 'a:' . string(get(d, 'func'))
548 call assert_equal(expected, d.func(string(remove(d, 'func'))))
Bram Moolenaar3e9c0b92021-08-12 10:39:10 +0200549
550 " similar, in a way it also works in Vim9
551 let lines =<< trim END
552 VAR d = {1: 1, 2: 'x'}
553 func GetArg(a)
554 return "a:" .. a:a
555 endfunc
556 LET d.func = function('GetArg')
557 VAR expected = 'a:' .. string(get(d, 'func'))
558 call assert_equal(expected, d.func(string(remove(d, 'func'))))
559 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000560 call v9.CheckTransLegacySuccess(lines)
561 call v9.CheckTransVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100562endfunc
563
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200564func Test_dict_literal_keys()
Bram Moolenaar4c6d9042019-07-16 22:04:02 +0200565 call assert_equal({'one': 1, 'two2': 2, '3three': 3, '44': 4}, #{one: 1, two2: 2, 3three: 3, 44: 4},)
Bram Moolenaarb8be54d2019-07-14 18:22:59 +0200566
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200567 " why *{} cannot be used for a literal dictionary
Bram Moolenaarb8be54d2019-07-14 18:22:59 +0200568 let blue = 'blue'
569 call assert_equal('6', trim(execute('echo 2 *{blue: 3}.blue')))
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200570endfunc
571
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100572" Nasty: deepcopy() dict that refers to itself (fails when noref used)
573func Test_dict_deepcopy()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200574 let lines =<< trim END
575 VAR d = {1: 1, 2: '2'}
576 VAR l = [4, d, 6]
577 LET d[3] = l
578 VAR dc = deepcopy(d)
579 call deepcopy(d, 1)
580 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000581 call v9.CheckLegacyAndVim9Failure(lines, 'E698:')
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200582
583 let lines =<< trim END
584 VAR d = {1: 1, 2: '2'}
585 VAR l = [4, d, 6]
586 LET d[3] = l
587 VAR l2 = [0, l, l, 3]
588 LET l[1] = l2
589 VAR l3 = deepcopy(l2)
590 call assert_true(l3[1] is l3[2])
591 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000592 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200593
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100594 call assert_fails("call deepcopy([1, 2], 2)", 'E1212:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100595endfunc
596
597" Locked variables
598func Test_list_locked_var()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200599 " Not tested with :def function, local vars cannot be locked.
600 let lines =<< trim END
601 VAR expected = [
602 \ [['1000-000', 'ppppppF'],
603 \ ['0000-000', 'ppppppp'],
604 \ ['0000-000', 'ppppppp']],
605 \ [['1000-000', 'ppppppF'],
606 \ ['0000-000', 'ppppppp'],
607 \ ['0000-000', 'ppppppp']],
608 \ [['1100-100', 'ppFppFF'],
609 \ ['0000-000', 'ppppppp'],
610 \ ['0000-000', 'ppppppp']],
611 \ [['1110-110', 'pFFpFFF'],
612 \ ['0010-010', 'pFppFpp'],
613 \ ['0000-000', 'ppppppp']],
614 \ [['1111-111', 'FFFFFFF'],
615 \ ['0011-011', 'FFpFFpp'],
616 \ ['0000-000', 'ppppppp']]
617 \ ]
618 for depth in range(5)
619 for u in range(3)
620 VAR l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
621 exe "lockvar " .. depth .. " l"
622 if u == 1
623 exe "unlockvar l"
624 elseif u == 2
625 exe "unlockvar " .. depth .. " l"
626 endif
627 VAR ps = islocked("l") .. islocked("l[1]") .. islocked("l[1][1]") .. islocked("l[1][1][0]") .. '-' .. islocked("l[2]") .. islocked("l[2]['6']") .. islocked("l[2]['6'][7]")
628 call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
629 LET ps = ''
630 try
631 LET l[1][1][0] = 99
632 LET ps ..= 'p'
633 catch
634 LET ps ..= 'F'
635 endtry
636 try
637 LET l[1][1] = [99]
638 LET ps ..= 'p'
639 catch
640 LET ps ..= 'F'
641 endtry
642 try
643 LET l[1] = [99]
644 LET ps ..= 'p'
645 catch
646 LET ps ..= 'F'
647 endtry
648 try
649 LET l[2]['6'][7] = 99
650 LET ps ..= 'p'
651 catch
652 LET ps ..= 'F'
653 endtry
654 try
655 LET l[2][6] = {99: 99}
656 LET ps ..= 'p'
657 catch
658 LET ps ..= 'F'
659 endtry
660 try
661 LET l[2] = {99: 99}
662 LET ps ..= 'p'
663 catch
664 LET ps ..= 'F'
665 endtry
666 try
667 LET l = [99]
668 LET ps ..= 'p'
669 catch
670 LET ps ..= 'F'
671 endtry
672 call assert_equal(expected[depth][u][1], ps, 'depth: ' .. depth)
673 unlock! l
674 endfor
675 endfor
676 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000677 call v9.CheckTransLegacySuccess(lines)
678 call v9.CheckTransVim9Success(lines)
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200679
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100680 call assert_fails("let x=islocked('a b')", 'E488:')
681 let mylist = [1, 2, 3]
682 call assert_fails("let x = islocked('mylist[1:2]')", 'E786:')
683 let mydict = {'k' : 'v'}
684 call assert_fails("let x = islocked('mydict.a')", 'E716:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100685endfunc
686
687" Unletting locked variables
688func Test_list_locked_var_unlet()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200689 " Not tested with Vim9: script and local variables cannot be unlocked
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100690 let expected = [
Bram Moolenaara187c432020-09-16 21:08:28 +0200691 \ [['1000-000', 'ppppppp'],
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100692 \ ['0000-000', 'ppppppp'],
693 \ ['0000-000', 'ppppppp']],
694 \ [['1000-000', 'ppFppFp'],
695 \ ['0000-000', 'ppppppp'],
696 \ ['0000-000', 'ppppppp']],
697 \ [['1100-100', 'pFFpFFp'],
698 \ ['0000-000', 'ppppppp'],
699 \ ['0000-000', 'ppppppp']],
700 \ [['1110-110', 'FFFFFFp'],
701 \ ['0010-010', 'FppFppp'],
702 \ ['0000-000', 'ppppppp']],
703 \ [['1111-111', 'FFFFFFp'],
704 \ ['0011-011', 'FppFppp'],
705 \ ['0000-000', 'ppppppp']]
706 \ ]
707
708 for depth in range(5)
709 for u in range(3)
710 unlet! l
711 let l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
712 exe "lockvar " . depth . " l"
713 if u == 1
714 exe "unlockvar l"
715 elseif u == 2
716 exe "unlockvar " . depth . " l"
717 endif
718 let ps = islocked("l").islocked("l[1]").islocked("l[1][1]").islocked("l[1][1][0]").'-'.islocked("l[2]").islocked("l[2]['6']").islocked("l[2]['6'][7]")
Bram Moolenaara187c432020-09-16 21:08:28 +0200719 call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100720 let ps = ''
721 try
722 unlet l[2]['6'][7]
723 let ps .= 'p'
724 catch
725 let ps .= 'F'
726 endtry
727 try
728 unlet l[2][6]
729 let ps .= 'p'
730 catch
731 let ps .= 'F'
732 endtry
733 try
734 unlet l[2]
735 let ps .= 'p'
736 catch
737 let ps .= 'F'
738 endtry
739 try
740 unlet l[1][1][0]
741 let ps .= 'p'
742 catch
743 let ps .= 'F'
744 endtry
745 try
746 unlet l[1][1]
747 let ps .= 'p'
748 catch
749 let ps .= 'F'
750 endtry
751 try
752 unlet l[1]
753 let ps .= 'p'
754 catch
755 let ps .= 'F'
756 endtry
757 try
758 unlet l
759 let ps .= 'p'
760 catch
761 let ps .= 'F'
762 endtry
763 call assert_equal(expected[depth][u][1], ps)
764 endfor
765 endfor
Bram Moolenaar422085f2021-12-17 20:36:15 +0000766
Bram Moolenaar6b8c7ba2022-03-20 17:46:06 +0000767 " Deleting a list range with locked items works, but changing the items
768 " fails.
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +0200769 let l = [1, 2, 3, 4]
770 lockvar l[1:2]
Bram Moolenaar6b8c7ba2022-03-20 17:46:06 +0000771 call assert_fails('let l[1:2] = [8, 9]', 'E741:')
772 unlet l[1:2]
773 call assert_equal([1, 4], l)
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +0200774 unlet l
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100775endfunc
776
777" Locked variables and :unlet or list / dict functions
778
779" No :unlet after lock on dict:
780func Test_dict_lock_unlet()
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100781 let d = {'a': 99, 'b': 100}
782 lockvar 1 d
Bram Moolenaare2e40752020-09-04 21:18:46 +0200783 call assert_fails('unlet d.a', 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100784endfunc
785
786" unlet after lock on dict item
787func Test_dict_item_lock_unlet()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200788 let lines =<< trim END
789 VAR d = {'a': 99, 'b': 100}
790 lockvar d.a
791 unlet d.a
792 call assert_equal({'b': 100}, d)
793 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000794 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100795endfunc
796
797" filter() after lock on dict item
798func Test_dict_lock_filter()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200799 let lines =<< trim END
800 VAR d = {'a': 99, 'b': 100}
801 lockvar d.a
802 call filter(d, 'v:key != "a"')
803 call assert_equal({'b': 100}, d)
804 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000805 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100806endfunc
807
808" map() after lock on dict
809func Test_dict_lock_map()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200810 let lines =<< trim END
811 VAR d = {'a': 99, 'b': 100}
812 lockvar 1 d
813 call map(d, 'v:val + 200')
814 call assert_equal({'a': 299, 'b': 300}, d)
815 END
816 " This won't work in a :def function
Bram Moolenaar62aec932022-01-29 21:45:34 +0000817 call v9.CheckTransLegacySuccess(lines)
818 call v9.CheckTransVim9Success(lines)
Bram Moolenaar71b76852021-12-17 20:15:38 +0000819
820 " For a :def function use a global dict.
821 let lines =<< trim END
822 let g:thedict = {'a': 77, 'b': 88}
823 lockvar 1 g:thedict
824 def Delkey()
825 unlet g:thedict.a
826 enddef
827 call Delkey()
828 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000829 call v9.CheckScriptFailure(lines, 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100830endfunc
831
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000832" Lock one item in a list
833func Test_list_item_lock_map()
834 let lines =<< trim END
835 VAR l = [99, 100, 101]
836 lockvar l[1]
837 call assert_fails("echo map(l, 'v:val + 200')", 'E741:')
838 call assert_equal([299, 100, 101], l)
839 END
840 " This won't work in a :def function
Bram Moolenaar62aec932022-01-29 21:45:34 +0000841 call v9.CheckTransLegacySuccess(lines)
842 call v9.CheckTransVim9Success(lines)
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000843endfunc
844
845" Lock one item in a dict
846func Test_dict_item_lock_map()
847 let lines =<< trim END
848 VAR d = {'a': 99, 'b': 100, 'c': 101}
849 lockvar d.b
850 call assert_fails("echo map(d, 'v:val + 200')", 'E741:')
851 call assert_equal({'a': 299, 'b': 100, 'c': 101}, d)
852 END
853 " This won't work in a :def function
Bram Moolenaar62aec932022-01-29 21:45:34 +0000854 call v9.CheckTransLegacySuccess(lines)
855 call v9.CheckTransVim9Success(lines)
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000856endfunc
857
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100858" No extend() after lock on dict item
859func Test_dict_lock_extend()
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100860 let d = {'a': 99, 'b': 100}
861 lockvar d.a
Bram Moolenaare2e40752020-09-04 21:18:46 +0200862 call assert_fails("call extend(d, {'a' : 123})", 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100863 call assert_equal({'a': 99, 'b': 100}, d)
864endfunc
865
Bram Moolenaarf7b398c2020-04-23 15:46:35 +0200866" Cannot use += with a locked dict
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200867func Test_dict_lock_operator()
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200868 let d = {}
869 lockvar d
870 call assert_fails("let d += {'k' : 10}", 'E741:')
871 unlockvar d
872endfunc
873
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100874" No remove() of write-protected scope-level variable
Bram Moolenaar1e115362019-01-09 23:01:02 +0100875func Tfunc1(this_is_a_long_parameter_name)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200876 call assert_fails("call remove(a:, 'this_is_a_long_parameter_name')", 'E742:')
Bram Moolenaar1e115362019-01-09 23:01:02 +0100877endfunc
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100878func Test_dict_scope_var_remove()
Bram Moolenaar1e115362019-01-09 23:01:02 +0100879 call Tfunc1('testval')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100880endfunc
881
882" No extend() of write-protected scope-level variable
Bram Moolenaar31b81602019-02-10 22:14:27 +0100883func Test_dict_scope_var_extend()
Bram Moolenaare2e40752020-09-04 21:18:46 +0200884 call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100885endfunc
886
Bram Moolenaar1e115362019-01-09 23:01:02 +0100887func Tfunc2(this_is_a_long_parameter_name)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200888 call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100889endfunc
Bram Moolenaar31b81602019-02-10 22:14:27 +0100890func Test_dict_scope_var_extend_overwrite()
Bram Moolenaar1e115362019-01-09 23:01:02 +0100891 call Tfunc2('testval')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100892endfunc
893
894" No :unlet of variable in locked scope
895func Test_lock_var_unlet()
896 let b:testvar = 123
897 lockvar 1 b:
898 call assert_fails('unlet b:testvar', 'E741:')
899 unlockvar 1 b:
900 unlet! b:testvar
901endfunc
902
903" No :let += of locked list variable
904func Test_let_lock_list()
905 let l = ['a', 'b', 3]
906 lockvar 1 l
907 call assert_fails("let l += ['x']", 'E741:')
908 call assert_equal(['a', 'b', 3], l)
909
910 unlet l
911 let l = [1, 2, 3, 4]
912 lockvar! l
913 call assert_equal([1, 2, 3, 4], l)
914 unlockvar l[1]
915 call assert_fails('unlet l[0:1]', 'E741:')
916 call assert_equal([1, 2, 3, 4], l)
917 call assert_fails('unlet l[1:2]', 'E741:')
918 call assert_equal([1, 2, 3, 4], l)
919 unlockvar l[1]
920 call assert_fails('let l[0:1] = [0, 1]', 'E741:')
921 call assert_equal([1, 2, 3, 4], l)
922 call assert_fails('let l[1:2] = [0, 1]', 'E741:')
923 call assert_equal([1, 2, 3, 4], l)
924 unlet l
Bram Moolenaar422085f2021-12-17 20:36:15 +0000925
926 let lines =<< trim END
927 def TryUnletListItem(l: list<any>)
928 unlet l[0]
929 enddef
930 let l = [1, 2, 3, 4]
931 lockvar! l
932 call TryUnletListItem(l)
933 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000934 call v9.CheckScriptFailure(lines, 'E741:')
Bram Moolenaar422085f2021-12-17 20:36:15 +0000935 unlet g:l
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100936endfunc
937
Bram Moolenaar8dfcce32020-03-18 19:32:26 +0100938" Locking part of the list
939func Test_let_lock_list_items()
940 let l = [1, 2, 3, 4]
941 lockvar l[2:]
942 call assert_equal(0, islocked('l[0]'))
943 call assert_equal(1, islocked('l[2]'))
944 call assert_equal(1, islocked('l[3]'))
945 call assert_fails('let l[2] = 10', 'E741:')
946 call assert_fails('let l[3] = 20', 'E741:')
947 unlet l
948endfunc
949
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100950" lockvar/islocked() triggering script autoloading
951func Test_lockvar_script_autoload()
952 let old_rtp = &rtp
953 set rtp+=./sautest
954 lockvar g:footest#x
955 unlockvar g:footest#x
Bram Moolenaarf9f24ce2019-08-31 21:17:39 +0200956 call assert_equal(-1, 'g:footest#x'->islocked())
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100957 call assert_equal(0, exists('g:footest#x'))
958 call assert_equal(1, g:footest#x)
959 let &rtp = old_rtp
960endfunc
961
962" a:000 function argument test
963func s:arg_list_test(...)
964 call assert_fails('let a:000 = [1, 2]', 'E46:')
965 call assert_fails('let a:000[0] = 9', 'E742:')
966 call assert_fails('let a:000[2] = [9, 10]', 'E742:')
967 call assert_fails('let a:000[3] = {9 : 10}', 'E742:')
968
969 " now the tests that should pass
970 let a:000[2][1] = 9
971 call extend(a:000[2], [5, 6])
972 let a:000[3][5] = 8
973 let a:000[3]['a'] = 12
974 call assert_equal([1, 2, [3, 9, 5, 6], {'a': 12, '5': 8}], a:000)
975endfunc
976
977func Test_func_arg_list()
978 call s:arg_list_test(1, 2, [3, 4], {5: 6})
979endfunc
980
981" Tests for reverse(), sort(), uniq()
982func Test_reverse_sort_uniq()
Bram Moolenaaref982572021-08-12 19:27:57 +0200983 let lines =<< trim END
984 VAR l = ['-0', 'A11', 2, 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5]
985 call assert_equal(['-0', 'A11', 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5], uniq(copy(l)))
986 call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(l))
987 call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(reverse(l)))
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100988 call assert_equal(['-0', 'A11', 'foo', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 2, 4, [0, 1, 2], [0, 1, 2]], sort(l))
989 call assert_equal([[0, 1, 2], [0, 1, 2], 4, 2, 2, 1.5, 'xaaa', 'x8', 'foo6', 'foo', 'foo', 'A11', '-0'], reverse(sort(l)))
990 call assert_equal(['-0', 'A11', 'foo', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 2, 4, [0, 1, 2], [0, 1, 2]], sort(reverse(sort(l))))
991 call assert_equal(['-0', 'A11', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 4, [0, 1, 2]], uniq(sort(l)))
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100992
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100993 LET l = [7, 9, 'one', 18, 12, 22, 'two', 10.0e-16, -1, 'three', 0xff, 0.22, 'four']
994 call assert_equal([-1, 'one', 'two', 'three', 'four', 1.0e-15, 0.22, 7, 9, 12, 18, 22, 255], sort(copy(l), 'n'))
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100995
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100996 LET l = [7, 9, 18, 12, 22, 10.0e-16, -1, 0xff, 0, -0, 0.22, 'bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', {}, []]
997 call assert_equal(['bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', -1, 0, 0, 0.22, 1.0e-15, 12, 18, 22, 255, 7, 9, [], {}], sort(copy(l), 'i'))
998 call assert_equal(['bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', -1, 0, 0, 0.22, 1.0e-15, 12, 18, 22, 255, 7, 9, [], {}], sort(copy(l), 'i'))
999 call assert_equal(['BAR', 'Bar', 'FOO', 'FOOBAR', 'Foo', 'bar', 'foo', -1, 0, 0, 0.22, 1.0e-15, 12, 18, 22, 255, 7, 9, [], {}], sort(copy(l)))
Bram Moolenaaref982572021-08-12 19:27:57 +02001000 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001001 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01001002
Yegappan Lakshmananf9dc2782023-05-11 15:02:56 +01001003 call assert_fails('call reverse({})', 'E1252:')
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001004 call assert_fails('call uniq([1, 2], {x, y -> []})', 'E745:')
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +01001005 call assert_fails("call sort([1, 2], function('min'), 1)", "E1206:")
Bram Moolenaar08f41572020-04-20 16:50:00 +02001006 call assert_fails("call sort([1, 2], function('invalid_func'))", "E700:")
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001007 call assert_fails("call sort([1, 2], function('min'))", "E118:")
Bram Moolenaar2007dd42022-02-23 13:17:47 +00001008
1009 let lines =<< trim END
1010 call sort(['a', 'b'], 0)
1011 END
1012 call v9.CheckDefAndScriptFailure(lines, 'E1256: String or function required for argument 2')
1013
1014 let lines =<< trim END
1015 call sort(['a', 'b'], 1)
1016 END
1017 call v9.CheckDefAndScriptFailure(lines, 'E1256: String or function required for argument 2')
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001018endfunc
1019
rbtnn0ccb5842021-12-18 18:33:46 +00001020" reduce a list, blob or string
Bram Moolenaar85629982020-06-01 18:39:20 +02001021func Test_reduce()
Bram Moolenaaref982572021-08-12 19:27:57 +02001022 let lines =<< trim END
1023 call assert_equal(1, reduce([], LSTART acc, val LMIDDLE acc + val LEND, 1))
1024 call assert_equal(10, reduce([1, 3, 5], LSTART acc, val LMIDDLE acc + val LEND, 1))
1025 call assert_equal(2 * (2 * ((2 * 1) + 2) + 3) + 4, reduce([2, 3, 4], LSTART acc, val LMIDDLE 2 * acc + val LEND, 1))
1026 call assert_equal('a x y z', ['x', 'y', 'z']->reduce(LSTART acc, val LMIDDLE acc .. ' ' .. val LEND, 'a'))
1027 call assert_equal([0, 1, 2, 3], reduce([1, 2, 3], function('add'), [0]))
Bram Moolenaar85629982020-06-01 18:39:20 +02001028
Bram Moolenaaref982572021-08-12 19:27:57 +02001029 VAR l = ['x', 'y', 'z']
1030 call assert_equal(42, reduce(l, function('get'), {'x': {'y': {'z': 42 } } }))
1031 call assert_equal(['x', 'y', 'z'], l)
Bram Moolenaar85629982020-06-01 18:39:20 +02001032
Bram Moolenaaref982572021-08-12 19:27:57 +02001033 call assert_equal(1, reduce([1], LSTART acc, val LMIDDLE acc + val LEND))
1034 call assert_equal('x y z', reduce(['x', 'y', 'z'], LSTART acc, val LMIDDLE acc .. ' ' .. val LEND))
1035 call assert_equal(120, range(1, 5)->reduce(LSTART acc, val LMIDDLE acc * val LEND))
1036
Bram Moolenaar52df40e2022-09-28 13:22:59 +01001037 call assert_equal(0, range(1)->reduce(LSTART acc, val LMIDDLE acc + val LEND))
1038 call assert_equal(1, range(2)->reduce(LSTART acc, val LMIDDLE acc + val LEND))
1039 call assert_equal(3, range(3)->reduce(LSTART acc, val LMIDDLE acc + val LEND))
1040 call assert_equal(6, range(4)->reduce(LSTART acc, val LMIDDLE acc + val LEND))
1041 call assert_equal(10, range(5)->reduce(LSTART acc, val LMIDDLE acc + val LEND))
1042
Bram Moolenaaref982572021-08-12 19:27:57 +02001043 call assert_equal(1, reduce(0z, LSTART acc, val LMIDDLE acc + val LEND, 1))
1044 call assert_equal(1 + 0xaf + 0xbf + 0xcf, reduce(0zAFBFCF, LSTART acc, val LMIDDLE acc + val LEND, 1))
1045 call assert_equal(2 * (2 * 1 + 0xaf) + 0xbf, 0zAFBF->reduce(LSTART acc, val LMIDDLE 2 * acc + val LEND, 1))
1046
1047 call assert_equal(0xff, reduce(0zff, LSTART acc, val LMIDDLE acc + val LEND))
1048 call assert_equal(2 * (2 * 0xaf + 0xbf) + 0xcf, reduce(0zAFBFCF, LSTART acc, val LMIDDLE 2 * acc + val LEND))
rbtnn0ccb5842021-12-18 18:33:46 +00001049
1050 call assert_equal('x,y,z', 'xyz'->reduce(LSTART acc, val LMIDDLE acc .. ',' .. val LEND))
1051 call assert_equal('', ''->reduce(LSTART acc, val LMIDDLE acc .. ',' .. val LEND, ''))
1052 call assert_equal('あ,い,う,え,お,😊,💕', 'あいうえお😊💕'->reduce(LSTART acc, val LMIDDLE acc .. ',' .. val LEND))
1053 call assert_equal('😊,あ,い,う,え,お,💕', 'あいうえお💕'->reduce(LSTART acc, val LMIDDLE acc .. ',' .. val LEND, '😊'))
1054 call assert_equal('ऊ,ॠ,ॡ', reduce('ऊॠॡ', LSTART acc, val LMIDDLE acc .. ',' .. val LEND))
1055 call assert_equal('c,à,t', reduce('càt', LSTART acc, val LMIDDLE acc .. ',' .. val LEND))
1056 call assert_equal('Å,s,t,r,ö,m', reduce('Åström', LSTART acc, val LMIDDLE acc .. ',' .. val LEND))
1057 call assert_equal('Å,s,t,r,ö,m', reduce('Åström', LSTART acc, val LMIDDLE acc .. ',' .. val LEND))
1058 call assert_equal(',a,b,c', reduce('abc', LSTART acc, val LMIDDLE acc .. ',' .. val LEND, test_null_string()))
zeertzjqad0c4422023-08-17 22:15:47 +02001059
1060 call assert_equal(0x7d, reduce([0x30, 0x25, 0x08, 0x61], 'or'))
1061 call assert_equal(0x7d, reduce(0z30250861, 'or'))
1062 call assert_equal('β', reduce('ββββ', 'matchstr'))
Bram Moolenaaref982572021-08-12 19:27:57 +02001063 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001064 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaaref982572021-08-12 19:27:57 +02001065
1066 call assert_equal({'x': 1, 'y': 1, 'z': 1 }, ['x', 'y', 'z']->reduce({ acc, val -> extend(acc, { val: 1 }) }, {}))
1067 vim9 assert_equal({'x': 1, 'y': 1, 'z': 1 }, ['x', 'y', 'z']->reduce((acc, val) => extend(acc, {[val]: 1 }), {}))
1068
Bram Moolenaar85629982020-06-01 18:39:20 +02001069 call assert_fails("call reduce([], { acc, val -> acc + val })", 'E998: Reduce of an empty List with no initial value')
Bram Moolenaar52df40e2022-09-28 13:22:59 +01001070 call assert_fails("call reduce(range(0), { acc, val -> acc + val })", 'E998: Reduce of an empty List with no initial value')
Bram Moolenaar85629982020-06-01 18:39:20 +02001071 call assert_fails("call reduce(0z, { acc, val -> acc + val })", 'E998: Reduce of an empty Blob with no initial value')
Yegappan Lakshmanan885de442022-04-23 10:51:14 +01001072 call assert_fails("call reduce(test_null_blob(), { acc, val -> acc + val })", 'E998: Reduce of an empty Blob with no initial value')
rbtnn0ccb5842021-12-18 18:33:46 +00001073 call assert_fails("call reduce('', { acc, val -> acc + val })", 'E998: Reduce of an empty String with no initial value')
1074 call assert_fails("call reduce(test_null_string(), { acc, val -> acc + val })", 'E998: Reduce of an empty String with no initial value')
Bram Moolenaar85629982020-06-01 18:39:20 +02001075
rbtnn0ccb5842021-12-18 18:33:46 +00001076 call assert_fails("call reduce({}, { acc, val -> acc + val }, 1)", 'E1098:')
1077 call assert_fails("call reduce(0, { acc, val -> acc + val }, 1)", 'E1098:')
zeertzjqad0c4422023-08-17 22:15:47 +02001078 call assert_fails("call reduce([1, 2], 'Xdoes_not_exist')", 'E117:')
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001079 call assert_fails("echo reduce(0z01, { acc, val -> 2 * acc + val }, '')", 'E1210:')
Bram Moolenaarca275a02020-06-24 22:07:46 +02001080
rbtnn0ccb5842021-12-18 18:33:46 +00001081 call assert_fails("vim9 reduce(0, (acc, val) => (acc .. val), '')", 'E1252:')
1082 call assert_fails("vim9 reduce({}, (acc, val) => (acc .. val), '')", 'E1252:')
1083 call assert_fails("vim9 reduce(0.1, (acc, val) => (acc .. val), '')", 'E1252:')
1084 call assert_fails("vim9 reduce(function('tr'), (acc, val) => (acc .. val), '')", 'E1252:')
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001085 call assert_fails("call reduce('', { acc, val -> acc + val }, 1)", 'E1174:')
1086 call assert_fails("call reduce('', { acc, val -> acc + val }, {})", 'E1174:')
1087 call assert_fails("call reduce('', { acc, val -> acc + val }, 0.1)", 'E1174:')
1088 call assert_fails("call reduce('', { acc, val -> acc + val }, function('tr'))", 'E1174:')
Yegappan Lakshmanan389b7212021-12-19 10:35:15 +00001089 call assert_fails("call reduce('abc', { a, v -> a10}, '')", 'E121:')
Yegappan Lakshmanan885de442022-04-23 10:51:14 +01001090 call assert_fails("call reduce(0z0102, { a, v -> a10}, 1)", 'E121:')
1091 call assert_fails("call reduce([1, 2], { a, v -> a10}, '')", 'E121:')
rbtnn0ccb5842021-12-18 18:33:46 +00001092
Bram Moolenaarca275a02020-06-24 22:07:46 +02001093 let g:lut = [1, 2, 3, 4]
1094 func EvilRemove()
1095 call remove(g:lut, 1)
1096 return 1
1097 endfunc
1098 call assert_fails("call reduce(g:lut, { acc, val -> EvilRemove() }, 1)", 'E742:')
1099 unlet g:lut
1100 delfunc EvilRemove
Bram Moolenaarfda20c42020-06-29 20:09:36 +02001101
1102 call assert_equal(42, reduce(test_null_list(), function('add'), 42))
1103 call assert_equal(42, reduce(test_null_blob(), function('add'), 42))
Bram Moolenaar0d90e722020-11-03 18:20:19 +01001104
1105 " should not crash
1106 call assert_fails('echo reduce([1], test_null_function())', 'E1132:')
Dominique Pellefe8ebdb2021-05-13 14:55:55 +02001107 call assert_fails('echo reduce([1], test_null_partial())', 'E1132:')
Bram Moolenaar85629982020-06-01 18:39:20 +02001108endfunc
1109
Bram Moolenaar0e05de42020-03-25 22:23:46 +01001110" splitting a string to a List using split()
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001111func Test_str_split()
Bram Moolenaaref982572021-08-12 19:27:57 +02001112 let lines =<< trim END
1113 call assert_equal(['aa', 'bb'], split(' aa bb '))
1114 call assert_equal(['aa', 'bb'], split(' aa bb ', '\W\+', 0))
1115 call assert_equal(['', 'aa', 'bb', ''], split(' aa bb ', '\W\+', 1))
1116 call assert_equal(['', '', 'aa', '', 'bb', '', ''], split(' aa bb ', '\W', 1))
1117 call assert_equal(['aa', '', 'bb'], split(':aa::bb:', ':', 0))
1118 call assert_equal(['', 'aa', '', 'bb', ''], split(':aa::bb:', ':', 1))
1119 call assert_equal(['aa', '', 'bb', 'cc', ''], split('aa,,bb, cc,', ',\s*', 1))
1120 call assert_equal(['a', 'b', 'c'], split('abc', '\zs'))
1121 call assert_equal(['', 'a', '', 'b', '', 'c', ''], split('abc', '\zs', 1))
1122 call assert_equal(['abc'], split('abc', '\\%('))
1123 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001124 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaaref982572021-08-12 19:27:57 +02001125
Bram Moolenaar0e05de42020-03-25 22:23:46 +01001126 call assert_fails("call split('abc', [])", 'E730:')
1127 call assert_fails("call split('abc', 'b', [])", 'E745:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001128endfunc
1129
1130" compare recursively linked list and dict
1131func Test_listdict_compare()
Bram Moolenaaref982572021-08-12 19:27:57 +02001132 let lines =<< trim END
1133 VAR l = [1, 2, 3, '4']
1134 VAR d = {'1': 1, '2': l, '3': 3}
1135 LET l[1] = d
1136 call assert_true(l == l)
1137 call assert_true(d == d)
1138 call assert_false(l != deepcopy(l))
1139 call assert_false(d != deepcopy(d))
1140 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001141 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar8b633132020-03-20 18:20:51 +01001142
1143 " comparison errors
1144 call assert_fails('echo [1, 2] =~ {}', 'E691:')
1145 call assert_fails('echo [1, 2] =~ [1, 2]', 'E692:')
1146 call assert_fails('echo {} =~ 5', 'E735:')
1147 call assert_fails('echo {} =~ {}', 'E736:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001148endfunc
1149
Yinzuo Jiang7ccd1a22024-07-04 17:20:53 +02001150func Test_recursive_listdict_compare()
1151 let l1 = [0, 1]
1152 let l1[0] = l1
1153 let l2 = [0, 1]
1154 let l2[0] = l2
1155 call assert_true(l1 == l2)
1156 let d1 = {0: 0, 1: 1}
1157 let d1[0] = d1
1158 let d2 = {0: 0, 1: 1}
1159 let d2[0] = d2
1160 call assert_true(d1 == d2)
1161endfunc
1162
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001163 " compare complex recursively linked list and dict
1164func Test_listdict_compare_complex()
Bram Moolenaaref982572021-08-12 19:27:57 +02001165 let lines =<< trim END
1166 VAR l = []
1167 call add(l, l)
1168 VAR dict4 = {"l": l}
1169 call add(dict4.l, dict4)
1170 VAR lcopy = deepcopy(l)
1171 VAR dict4copy = deepcopy(dict4)
1172 call assert_true(l == lcopy)
1173 call assert_true(dict4 == dict4copy)
1174 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001175 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001176endfunc
1177
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001178" Test for extending lists and dictionaries
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001179func Test_listdict_extend()
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001180 " Test extend() with lists
1181
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001182 " Pass the same List to extend()
Bram Moolenaaref982572021-08-12 19:27:57 +02001183 let lines =<< trim END
1184 VAR l = [1, 2, 3]
1185 call assert_equal([1, 2, 3, 1, 2, 3], extend(l, l))
1186 call assert_equal([1, 2, 3, 1, 2, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001187
Bram Moolenaaref982572021-08-12 19:27:57 +02001188 LET l = [1, 2, 3]
1189 call assert_equal([1, 2, 3, 4, 5, 6], extend(l, [4, 5, 6]))
1190 call assert_equal([1, 2, 3, 4, 5, 6], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001191
Bram Moolenaaref982572021-08-12 19:27:57 +02001192 LET l = [1, 2, 3]
1193 call extend(l, [4, 5, 6], 0)
1194 call assert_equal([4, 5, 6, 1, 2, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001195
Bram Moolenaaref982572021-08-12 19:27:57 +02001196 LET l = [1, 2, 3]
1197 call extend(l, [4, 5, 6], 1)
1198 call assert_equal([1, 4, 5, 6, 2, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001199
Bram Moolenaaref982572021-08-12 19:27:57 +02001200 LET l = [1, 2, 3]
1201 call extend(l, [4, 5, 6], 3)
1202 call assert_equal([1, 2, 3, 4, 5, 6], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001203
Bram Moolenaaref982572021-08-12 19:27:57 +02001204 LET l = [1, 2, 3]
1205 call extend(l, [4, 5, 6], -1)
1206 call assert_equal([1, 2, 4, 5, 6, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001207
Bram Moolenaaref982572021-08-12 19:27:57 +02001208 LET l = [1, 2, 3]
1209 call extend(l, [4, 5, 6], -3)
1210 call assert_equal([4, 5, 6, 1, 2, 3], l)
1211 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001212 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001213
1214 let l = [1, 2, 3]
1215 call assert_fails("call extend(l, [4, 5, 6], 4)", 'E684:')
1216 call assert_fails("call extend(l, [4, 5, 6], -4)", 'E684:')
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01001217 call assert_fails("call extend(l, [4, 5, 6], 1.2)", 'E805:')
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001218
1219 " Test extend() with dictionaries.
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001220
1221 " Pass the same Dict to extend()
Bram Moolenaaref982572021-08-12 19:27:57 +02001222 let lines =<< trim END
1223 VAR d = {'a': {'b': 'B'}, 'x': 9}
1224 call extend(d, d)
1225 call assert_equal({'a': {'b': 'B'}, 'x': 9}, d)
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001226
Bram Moolenaaref982572021-08-12 19:27:57 +02001227 LET d = {'a': 'A', 'b': 9}
1228 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, extend(d, {'b': 0, 'c': 'C'}))
1229 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001230
Bram Moolenaaref982572021-08-12 19:27:57 +02001231 LET d = {'a': 'A', 'b': 9}
1232 call extend(d, {'a': 'A', 'b': 0, 'c': 'C'}, "force")
1233 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001234
Bram Moolenaaref982572021-08-12 19:27:57 +02001235 LET d = {'a': 'A', 'b': 9}
1236 call extend(d, {'b': 0, 'c': 'C'}, "keep")
1237 call assert_equal({'a': 'A', 'b': 9, 'c': 'C'}, d)
1238 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001239 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001240
1241 let d = {'a': 'A', 'b': 'B'}
1242 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'error')", 'E737:')
Yegappan Lakshmanan27708e62021-12-26 21:54:43 +00001243 call assert_fails("call extend(d, {'b': 0}, [])", 'E730:')
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001244 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'xxx')", 'E475:')
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01001245 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 1.2)", 'E475:')
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001246 call assert_equal({'a': 'A', 'b': 'B'}, d)
1247
1248 call assert_fails("call extend([1, 2], 1)", 'E712:')
1249 call assert_fails("call extend([1, 2], {})", 'E712:')
Bram Moolenaar08f41572020-04-20 16:50:00 +02001250
1251 " Extend g: dictionary with an invalid variable name
1252 call assert_fails("call extend(g:, {'-!' : 10})", 'E461:')
Bram Moolenaardcae51f2021-04-08 20:10:10 +02001253
1254 " Extend a list with itself.
Bram Moolenaaref982572021-08-12 19:27:57 +02001255 let lines =<< trim END
1256 VAR l = [1, 5, 7]
1257 call extend(l, l, 0)
1258 call assert_equal([1, 5, 7, 1, 5, 7], l)
1259 LET l = [1, 5, 7]
1260 call extend(l, l, 1)
1261 call assert_equal([1, 1, 5, 7, 5, 7], l)
1262 LET l = [1, 5, 7]
1263 call extend(l, l, 2)
1264 call assert_equal([1, 5, 1, 5, 7, 7], l)
1265 LET l = [1, 5, 7]
1266 call extend(l, l, 3)
1267 call assert_equal([1, 5, 7, 1, 5, 7], l)
1268 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001269 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001270endfunc
Bram Moolenaar31b81602019-02-10 22:14:27 +01001271
Bram Moolenaarb0e6b512021-01-12 20:23:40 +01001272func Test_listdict_extendnew()
1273 " Test extendnew() with lists
1274 let l = [1, 2, 3]
1275 call assert_equal([1, 2, 3, 4, 5], extendnew(l, [4, 5]))
1276 call assert_equal([1, 2, 3], l)
zeertzjq341f3872023-02-27 14:59:57 +00001277 lockvar l
1278 call assert_equal([1, 2, 3, 4, 5], extendnew(l, [4, 5]))
Bram Moolenaarb0e6b512021-01-12 20:23:40 +01001279
zeertzjq341f3872023-02-27 14:59:57 +00001280 " Test extendnew() with dictionaries.
Bram Moolenaarb0e6b512021-01-12 20:23:40 +01001281 let d = {'a': {'b': 'B'}}
1282 call assert_equal({'a': {'b': 'B'}, 'c': 'cc'}, extendnew(d, {'c': 'cc'}))
1283 call assert_equal({'a': {'b': 'B'}}, d)
zeertzjq341f3872023-02-27 14:59:57 +00001284 lockvar d
1285 call assert_equal({'a': {'b': 'B'}, 'c': 'cc'}, extendnew(d, {'c': 'cc'}))
Bram Moolenaarb0e6b512021-01-12 20:23:40 +01001286endfunc
1287
Bram Moolenaar31b81602019-02-10 22:14:27 +01001288func s:check_scope_dict(x, fixed)
1289 func s:gen_cmd(cmd, x)
1290 return substitute(a:cmd, '\<x\ze:', a:x, 'g')
1291 endfunc
1292
1293 let cmd = s:gen_cmd('let x:foo = 1', a:x)
1294 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +02001295 call assert_fails(cmd, 'E461:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001296 else
1297 exe cmd
1298 exe s:gen_cmd('call assert_equal(1, x:foo)', a:x)
1299 endif
1300
1301 let cmd = s:gen_cmd('let x:["bar"] = 2', a:x)
1302 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +02001303 call assert_fails(cmd, 'E461:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001304 else
1305 exe cmd
1306 exe s:gen_cmd('call assert_equal(2, x:bar)', a:x)
1307 endif
1308
1309 let cmd = s:gen_cmd('call extend(x:, {"baz": 3})', a:x)
1310 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +02001311 call assert_fails(cmd, 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001312 else
1313 exe cmd
1314 exe s:gen_cmd('call assert_equal(3, x:baz)', a:x)
1315 endif
1316
1317 if a:fixed
1318 if a:x ==# 'a'
Bram Moolenaare2e40752020-09-04 21:18:46 +02001319 call assert_fails('unlet a:x', 'E795:')
1320 call assert_fails('call remove(a:, "x")', 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001321 elseif a:x ==# 'v'
Bram Moolenaare2e40752020-09-04 21:18:46 +02001322 call assert_fails('unlet v:count', 'E795:')
1323 call assert_fails('call remove(v:, "count")', 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001324 endif
1325 else
1326 exe s:gen_cmd('unlet x:foo', a:x)
1327 exe s:gen_cmd('unlet x:bar', a:x)
1328 exe s:gen_cmd('call remove(x:, "baz")', a:x)
1329 endif
1330
1331 delfunc s:gen_cmd
1332endfunc
1333
1334func Test_scope_dict()
1335 " Test for g:
1336 call s:check_scope_dict('g', v:false)
1337
1338 " Test for s:
1339 call s:check_scope_dict('s', v:false)
1340
1341 " Test for l:
1342 call s:check_scope_dict('l', v:false)
1343
1344 " Test for a:
1345 call s:check_scope_dict('a', v:true)
1346
1347 " Test for b:
1348 call s:check_scope_dict('b', v:false)
1349
1350 " Test for w:
1351 call s:check_scope_dict('w', v:false)
1352
1353 " Test for t:
1354 call s:check_scope_dict('t', v:false)
1355
1356 " Test for v:
1357 call s:check_scope_dict('v', v:true)
1358endfunc
Bram Moolenaar8dfcce32020-03-18 19:32:26 +01001359
1360" Test for deep nesting of lists (> 100)
1361func Test_deep_nested_list()
1362 let deep_list = []
1363 let l = deep_list
1364 for i in range(102)
1365 let newlist = []
1366 call add(l, newlist)
1367 let l = newlist
1368 endfor
1369 call add(l, 102)
1370
1371 call assert_fails('let m = deepcopy(deep_list)', 'E698:')
1372 call assert_fails('lockvar 110 deep_list', 'E743:')
1373 call assert_fails('unlockvar 110 deep_list', 'E743:')
1374 call assert_fails('let x = execute("echo deep_list")', 'E724:')
1375 call test_garbagecollect_now()
1376 unlet deep_list
1377endfunc
1378
1379" Test for deep nesting of dicts (> 100)
1380func Test_deep_nested_dict()
1381 let deep_dict = {}
1382 let d = deep_dict
1383 for i in range(102)
1384 let newdict = {}
1385 let d.k = newdict
1386 let d = newdict
1387 endfor
1388 let d.k = 'v'
1389
1390 call assert_fails('let m = deepcopy(deep_dict)', 'E698:')
1391 call assert_fails('lockvar 110 deep_dict', 'E743:')
1392 call assert_fails('unlockvar 110 deep_dict', 'E743:')
1393 call assert_fails('let x = execute("echo deep_dict")', 'E724:')
1394 call test_garbagecollect_now()
1395 unlet deep_dict
1396endfunc
1397
Bram Moolenaar8b633132020-03-20 18:20:51 +01001398" List and dict indexing tests
1399func Test_listdict_index()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001400 call v9.CheckLegacyAndVim9Failure(['echo function("min")[0]'], 'E695:')
1401 call v9.CheckLegacyAndVim9Failure(['echo v:true[0]'], 'E909:')
1402 call v9.CheckLegacyAndVim9Failure(['echo v:null[0]'], 'E909:')
1403 call v9.CheckLegacyAndVim9Failure(['VAR d = {"k": 10}', 'echo d.'], ['E15:', 'E1127:', 'E15:'])
1404 call v9.CheckLegacyAndVim9Failure(['VAR d = {"k": 10}', 'echo d[1 : 2]'], 'E719:')
Bram Moolenaarf47c5a82021-12-19 15:17:21 +00001405
Bram Moolenaar8b633132020-03-20 18:20:51 +01001406 call assert_fails("let v = [4, 6][{-> 1}]", 'E729:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001407 call v9.CheckDefAndScriptFailure(['var v = [4, 6][() => 1]'], ['E1012', 'E703:'])
Bram Moolenaarf47c5a82021-12-19 15:17:21 +00001408
Bram Moolenaar62aec932022-01-29 21:45:34 +00001409 call v9.CheckLegacyAndVim9Failure(['VAR v = range(5)[2 : []]'], ['E730:', 'E1012:', 'E730:'])
Bram Moolenaar700e6b12021-12-19 17:27:06 +00001410
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001411 call assert_fails("let v = range(5)[2:{-> 2}(]", ['E15:', 'E116:'])
Bram Moolenaar62aec932022-01-29 21:45:34 +00001412 call v9.CheckDefAndScriptFailure(['var v = range(5)[2 : () => 2(]'], 'E15:')
Bram Moolenaar700e6b12021-12-19 17:27:06 +00001413
Bram Moolenaar62aec932022-01-29 21:45:34 +00001414 call v9.CheckLegacyAndVim9Failure(['VAR v = range(5)[2 : 3'], ['E111:', 'E1097:', 'E111:'])
1415 call v9.CheckLegacyAndVim9Failure(['VAR l = insert([1, 2, 3], 4, 10)'], 'E684:')
1416 call v9.CheckLegacyAndVim9Failure(['VAR l = insert([1, 2, 3], 4, -10)'], 'E684:')
1417 call v9.CheckLegacyAndVim9Failure(['VAR l = insert([1, 2, 3], 4, [])'], ['E745:', 'E1013:', 'E1210:'])
Bram Moolenaar700e6b12021-12-19 17:27:06 +00001418
Bram Moolenaar62aec932022-01-29 21:45:34 +00001419 call v9.CheckLegacyAndVim9Failure(['VAR l = [1, 2, 3]', 'LET l[i] = 3'], ['E121:', 'E1001:', 'E121:'])
1420 call v9.CheckLegacyAndVim9Failure(['VAR l = [1, 2, 3]', 'LET l[1.1] = 4'], ['E805:', 'E1012:', 'E805:'])
1421 call v9.CheckLegacyAndVim9Failure(['VAR l = [1, 2, 3]', 'LET l[: i] = [4, 5]'], ['E121:', 'E1001:', 'E121:'])
1422 call v9.CheckLegacyAndVim9Failure(['VAR l = [1, 2, 3]', 'LET l[: 3.2] = [4, 5]'], ['E805:', 'E1012:', 'E805:'])
Bram Moolenaar097c5372023-05-24 21:02:24 +01001423 call v9.CheckLegacyAndVim9Failure(['VAR t = test_unknown()', 'echo t[0]'], ['E340:', 'E909:', 'E340:', 'E685:'])
Bram Moolenaar08f41572020-04-20 16:50:00 +02001424endfunc
1425
1426" Test for a null list
1427func Test_null_list()
Bram Moolenaaref982572021-08-12 19:27:57 +02001428 let lines =<< trim END
1429 VAR l = test_null_list()
1430 call assert_equal('', join(test_null_list()))
1431 call assert_equal('', join(l))
1432 call assert_equal(0, len(l))
1433 call assert_equal(1, empty(l))
1434 call assert_equal([], split(test_null_string()))
1435 call assert_equal([], l[ : 2])
1436 call assert_true([] == l)
1437 call assert_equal('[]', string(l))
1438 call assert_equal([], sort(test_null_list()))
1439 call assert_equal([], sort(l))
1440 call assert_equal([], uniq(test_null_list()))
1441 call assert_equal([], uniq(l))
1442 VAR k = [] + l
1443 call assert_equal([], k)
1444 LET k = l + []
1445 call assert_equal([], k)
1446 call assert_equal(0, len(copy(l)))
1447 call assert_equal(0, count(l, 5))
1448 call assert_equal([], deepcopy(l))
1449 call assert_equal(5, get(l, 2, 5))
1450 call assert_equal(-1, index(l, 2, 5))
1451 call assert_equal(0, min(l))
1452 call assert_equal(0, max(l))
1453 call assert_equal(0, remove(test_null_list(), 0, 2))
1454 call assert_equal([], repeat(l, 2))
1455 call assert_equal([], reverse(test_null_list()))
1456 call assert_equal([], reverse(l))
1457 call assert_equal([], sort(test_null_list()))
1458 call assert_equal([], sort(l))
1459 call assert_equal('[]', string(l))
1460 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001461 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaaref982572021-08-12 19:27:57 +02001462
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001463 let l = test_null_list()
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001464 call assert_equal([], extend(l, l, 0))
Bram Moolenaaref982572021-08-12 19:27:57 +02001465 call assert_equal(0, insert(test_null_list(), 2, -1))
1466 call assert_fails('let s = join([1, 2], [])', 'E730:')
1467 call assert_fails('call remove(l, 0, 2)', 'E684:')
1468 call assert_fails('call insert(l, 2, -1)', 'E684:')
1469 call assert_fails('call extend(test_null_list(), test_null_list())', 'E1134:')
1470
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001471 lockvar l
1472 call assert_equal(1, islocked('l'))
1473 unlockvar l
Bram Moolenaar08f41572020-04-20 16:50:00 +02001474endfunc
1475
1476" Test for a null dict
1477func Test_null_dict()
Bram Moolenaaref982572021-08-12 19:27:57 +02001478 let lines =<< trim END
1479 call assert_equal(test_null_dict(), test_null_dict())
1480 VAR d = test_null_dict()
1481 call assert_equal({}, d)
1482 call assert_equal(0, len(d))
1483 call assert_equal(1, empty(d))
1484 call assert_equal([], items(test_null_dict()))
1485 call assert_equal([], items(d))
1486 call assert_equal([], keys(test_null_dict()))
1487 call assert_equal([], keys(d))
1488 call assert_equal([], values(test_null_dict()))
1489 call assert_equal([], values(d))
1490 call assert_false(has_key(d, 'k'))
1491 call assert_equal('{}', string(d))
1492 call assert_equal({}, {})
1493 call assert_equal(0, len(copy(d)))
1494 call assert_equal(0, count(d, 'k'))
1495 call assert_equal({}, deepcopy(d))
1496 call assert_equal(20, get(d, 'k', 20))
1497 call assert_equal(0, min(d))
1498 call assert_equal(0, max(d))
1499 call assert_equal(0, remove(test_null_dict(), 'k'))
1500 call assert_equal('{}', string(d))
1501 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001502 call v9.CheckLegacyAndVim9Success(lines)
Bram Moolenaaref982572021-08-12 19:27:57 +02001503
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001504 let d = test_null_dict()
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001505 call assert_equal({}, extend(d, d, 'keep'))
Bram Moolenaaref982572021-08-12 19:27:57 +02001506 call assert_fails("call remove(d, 'k')", 'E716:')
1507 call assert_fails('let x = d[10]', 'E716:')
1508 call assert_fails('call extend(test_null_dict(), test_null_dict())', 'E1133:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001509 lockvar d
1510 call assert_equal(1, islocked('d'))
1511 unlockvar d
Bram Moolenaar8b633132020-03-20 18:20:51 +01001512endfunc
1513
Yegappan Lakshmananb2186552022-08-13 13:09:20 +01001514" Test for the indexof() function
1515func Test_indexof()
1516 let l = [#{color: 'red'}, #{color: 'blue'}, #{color: 'green'}]
1517 call assert_equal(0, indexof(l, {i, v -> v.color == 'red'}))
1518 call assert_equal(2, indexof(l, {i, v -> v.color == 'green'}))
1519 call assert_equal(-1, indexof(l, {i, v -> v.color == 'grey'}))
1520 call assert_equal(1, indexof(l, "v:val.color == 'blue'"))
1521 call assert_equal(-1, indexof(l, "v:val.color == 'cyan'"))
1522
1523 let l = [#{n: 10}, #{n: 10}, #{n: 20}]
1524 call assert_equal(0, indexof(l, "v:val.n == 10", #{startidx: 0}))
1525 call assert_equal(1, indexof(l, "v:val.n == 10", #{startidx: -2}))
1526 call assert_equal(-1, indexof(l, "v:val.n == 10", #{startidx: 4}))
1527 call assert_equal(-1, indexof(l, "v:val.n == 10", #{startidx: -4}))
1528 call assert_equal(0, indexof(l, "v:val.n == 10", test_null_dict()))
1529
Yegappan Lakshmanan63acae12022-08-14 12:07:11 +01001530 let s = ["a", "b", "c"]
1531 call assert_equal(2, indexof(s, {_, v -> v == 'c'}))
1532 call assert_equal(-1, indexof(s, {_, v -> v == 'd'}))
1533 call assert_equal(-1, indexof(s, {_, v -> "v == 'd'"}))
1534
Yegappan Lakshmananb2186552022-08-13 13:09:20 +01001535 call assert_equal(-1, indexof([], {i, v -> v == 'a'}))
Yegappan Lakshmanan63acae12022-08-14 12:07:11 +01001536 call assert_equal(-1, indexof([1, 2, 3], {_, v -> "v == 2"}))
Yegappan Lakshmananb2186552022-08-13 13:09:20 +01001537 call assert_equal(-1, indexof(test_null_list(), {i, v -> v == 'a'}))
1538 call assert_equal(-1, indexof(l, test_null_string()))
1539 call assert_equal(-1, indexof(l, test_null_function()))
Yegappan Lakshmananfe424d12024-05-17 18:20:43 +02001540 call assert_equal(-1, indexof(l, ""))
1541 call assert_fails('let i = indexof(l, " ")', 'E15:')
Yegappan Lakshmananb2186552022-08-13 13:09:20 +01001542
1543 " failure cases
1544 call assert_fails('let i = indexof(l, "v:val == ''cyan''")', 'E735:')
1545 call assert_fails('let i = indexof(l, "color == ''cyan''")', 'E121:')
1546 call assert_fails('let i = indexof(l, {})', 'E1256:')
1547 call assert_fails('let i = indexof({}, "v:val == 2")', 'E1226:')
1548 call assert_fails('let i = indexof([], "v:val == 2", [])', 'E1206:')
1549
1550 func TestIdx(k, v)
1551 return a:v.n == 20
1552 endfunc
1553 call assert_equal(2, indexof(l, function("TestIdx")))
1554 delfunc TestIdx
1555 func TestIdx(k, v)
1556 return {}
1557 endfunc
1558 call assert_fails('let i = indexof(l, function("TestIdx"))', 'E728:')
1559 delfunc TestIdx
1560 func TestIdx(k, v)
1561 throw "IdxError"
1562 endfunc
1563 call assert_fails('let i = indexof(l, function("TestIdx"))', 'E605:')
1564 delfunc TestIdx
1565endfunc
1566
Christian Brabandt29269a72024-04-16 22:44:31 +02001567func Test_extendnew_leak()
1568 " This used to leak memory
1569 for i in range(100) | silent! call extendnew([], [], []) | endfor
1570 for i in range(100) | silent! call extendnew({}, {}, {}) | endfor
1571endfunc
1572
Yegappan Lakshmanan88bbdb02024-06-23 10:00:04 +02001573" Test for comparing deeply nested List/Dict values
1574func Test_deep_nested_listdict_compare()
1575 let lines =<< trim END
1576 def GetNestedList(sz: number): list<any>
1577 var l: list<any> = []
1578 var x: list<any> = l
1579 for i in range(sz)
1580 var y: list<any> = [1]
1581 add(x, y)
1582 x = y
1583 endfor
1584 return l
1585 enddef
1586
1587 VAR l1 = GetNestedList(1000)
1588 VAR l2 = GetNestedList(999)
1589 call assert_false(l1 == l2)
1590
1591 #" after 1000 nested items, the lists are considered to be equal
1592 VAR l3 = GetNestedList(1001)
1593 VAR l4 = GetNestedList(1002)
1594 call assert_true(l3 == l4)
1595 END
1596 call v9.CheckLegacyAndVim9Success(lines)
1597
1598 let lines =<< trim END
1599 def GetNestedDict(sz: number): dict<any>
1600 var d: dict<any> = {}
1601 var x: dict<any> = d
1602 for i in range(sz)
1603 var y: dict<any> = {}
1604 x['a'] = y
1605 x = y
1606 endfor
1607 return d
1608 enddef
1609
1610 VAR d1 = GetNestedDict(1000)
1611 VAR d2 = GetNestedDict(999)
1612 call assert_false(d1 == d2)
1613
1614 #" after 1000 nested items, the Dicts are considered to be equal
1615 VAR d3 = GetNestedDict(1001)
1616 VAR d4 = GetNestedDict(1002)
1617 call assert_true(d3 == d4)
1618 END
1619 call v9.CheckLegacyAndVim9Success(lines)
1620endfunc
1621
Ernie Raelc8e158b2024-07-09 18:39:52 +02001622" Test for using id()
1623def Test_id_with_dict()
zeertzjq8feed3a2024-09-29 10:37:47 +02001624 # demonstrate a way that "id(item)" differs from "string(item)"
Ernie Raelc8e158b2024-07-09 18:39:52 +02001625 var d1 = {one: 1}
1626 var d2 = {one: 1}
1627 var d3 = {one: 1}
1628 var idDict: dict<any>
1629 idDict[id(d1)] = d1
1630 idDict[id(d2)] = d2
1631 idDict[id(d3)] = d3
1632 assert_equal(3, idDict->len())
1633
1634 var stringDict: dict<any>
1635 stringDict[string(d1)] = d1
1636 stringDict[string(d2)] = d2
1637 stringDict[string(d3)] = d3
1638 assert_equal(1, stringDict->len())
1639
1640 assert_equal('', id(3))
1641
1642 assert_equal('', id(null))
1643 assert_equal('', id(null_blob))
1644 assert_equal('', id(null_dict))
1645 assert_equal('', id(null_function))
1646 assert_equal('', id(null_list))
1647 assert_equal('', id(null_partial))
1648 assert_equal('', id(null_string))
1649 assert_equal('', id(null_channel))
1650 assert_equal('', id(null_job))
1651enddef
Bram Moolenaar8dfcce32020-03-18 19:32:26 +01001652" vim: shiftwidth=2 sts=2 expandtab