blob: 45a4b8e40107320553f196e9edef75a82921e280 [file] [log] [blame]
Bram Moolenaardb913952012-06-29 12:54:53 +02001Tests for various python features. vim: set ft=vim :
2
3STARTTEST
4:so small.vim
5:if !has('python3') | e! test.ok | wq! test.out | endif
Bram Moolenaarc24c1ac2013-05-16 20:47:56 +02006:lang C
Bram Moolenaardb913952012-06-29 12:54:53 +02007:py3 import vim
8:fun Test()
9:let l = []
10:py3 l=vim.bindeval('l')
11:py3 f=vim.bindeval('function("strlen")')
12:" Extending List directly with different types
13:py3 l+=[1, "as'd", [1, 2, f, {'a': 1}]]
14:$put =string(l)
15:$put =string(l[-1])
16:try
17: $put =string(l[-4])
18:catch
19: $put =v:exception[:13]
20:endtry
21:" List assignment
22:py3 l[0]=0
23:$put =string(l)
24:py3 l[-2]=f
25:$put =string(l)
26:"
27:" Extending Dictionary directly with different types
28:let d = {}
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020029:fun d.f()
30: return 1
31:endfun
Bram Moolenaara9922d62013-05-30 13:01:18 +020032py3 << EOF
33d=vim.bindeval('d')
34d['1']='asd'
35d.update(b=[1, 2, f])
36d.update((('-1', {'a': 1}),))
37d.update({'0': -1})
38dk = d.keys()
39dv = d.values()
40di = d.items()
41dk.sort(key=repr)
42dv.sort(key=repr)
43di.sort(key=repr)
44EOF
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020045:$put =py3eval('d[''f''](self={})')
Bram Moolenaara9922d62013-05-30 13:01:18 +020046:$put =py3eval('repr(dk)')
47:$put =substitute(py3eval('repr(dv)'),'0x\x\+','','g')
48:$put =substitute(py3eval('repr(di)'),'0x\x\+','','g')
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020049:for [key, Val] in sort(items(d))
50: $put =string(key) . ' : ' . string(Val)
51: unlet key Val
Bram Moolenaardb913952012-06-29 12:54:53 +020052:endfor
53:"
54:" removing items with del
55:py3 del l[2]
56:$put =string(l)
57:let l = range(8)
58:py3 l=vim.bindeval('l')
59:try
60: py3 del l[:3]
61: py3 del l[1:]
62:catch
63: $put =v:exception
64:endtry
65:$put =string(l)
66:"
67:py3 del d['-1']
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020068:py3 del d['f']
Bram Moolenaara9922d62013-05-30 13:01:18 +020069:$put =string(py3eval('d.get(''b'', 1)'))
70:$put =string(py3eval('d.pop(''b'')'))
71:$put =string(py3eval('d.get(''b'', 1)'))
72:$put =string(py3eval('d.pop(''1'', 2)'))
73:$put =string(py3eval('d.pop(''1'', 2)'))
74:$put =py3eval('repr(d.has_key(''0''))')
75:$put =py3eval('repr(d.has_key(''1''))')
76:$put =py3eval('repr(''0'' in d)')
77:$put =py3eval('repr(''1'' in d)')
78:$put =py3eval('repr(list(iter(d)))')
Bram Moolenaardb913952012-06-29 12:54:53 +020079:$put =string(d)
Bram Moolenaarde71b562013-06-02 17:41:54 +020080:$put =py3eval('repr(d.popitem())')
Bram Moolenaara9922d62013-05-30 13:01:18 +020081:$put =py3eval('repr(d.get(''0''))')
82:$put =py3eval('repr(list(iter(d)))')
Bram Moolenaardb913952012-06-29 12:54:53 +020083:"
84:" removing items out of range: silently skip items that don't exist
85:let l = [0, 1, 2, 3]
86:py3 l=vim.bindeval('l')
87:" The following two ranges delete nothing as they match empty list:
88:py3 del l[2:1]
89:$put =string(l)
90:py3 del l[2:2]
91:$put =string(l)
92:py3 del l[2:3]
93:$put =string(l)
94:let l = [0, 1, 2, 3]
95:py3 l=vim.bindeval('l')
96:py3 del l[2:4]
97:$put =string(l)
98:let l = [0, 1, 2, 3]
99:py3 l=vim.bindeval('l')
100:py3 del l[2:5]
101:$put =string(l)
102:let l = [0, 1, 2, 3]
103:py3 l=vim.bindeval('l')
104:py3 del l[2:6]
105:$put =string(l)
106:let l = [0, 1, 2, 3]
107:py3 l=vim.bindeval('l')
108:" The following two ranges delete nothing as they match empty list:
109:py3 del l[-1:2]
110:$put =string(l)
111:py3 del l[-2:2]
112:$put =string(l)
113:py3 del l[-3:2]
114:$put =string(l)
115:let l = [0, 1, 2, 3]
116:py3 l=vim.bindeval('l')
117:py3 del l[-4:2]
118:$put =string(l)
119:let l = [0, 1, 2, 3]
120:py3 l=vim.bindeval('l')
121:py3 del l[-5:2]
122:$put =string(l)
123:let l = [0, 1, 2, 3]
124:py3 l=vim.bindeval('l')
125:py3 del l[-6:2]
126:$put =string(l)
127:"
128:" Slice assignment to a list
129:let l = [0, 1, 2, 3]
130:py3 l=vim.bindeval('l')
131:py3 l[0:0]=['a']
132:$put =string(l)
133:let l = [0, 1, 2, 3]
134:py3 l=vim.bindeval('l')
135:py3 l[1:2]=['b']
136:$put =string(l)
137:let l = [0, 1, 2, 3]
138:py3 l=vim.bindeval('l')
139:py3 l[2:4]=['c']
140:$put =string(l)
141:let l = [0, 1, 2, 3]
142:py3 l=vim.bindeval('l')
143:py3 l[4:4]=['d']
144:$put =string(l)
145:let l = [0, 1, 2, 3]
146:py3 l=vim.bindeval('l')
147:py3 l[-1:2]=['e']
148:$put =string(l)
149:let l = [0, 1, 2, 3]
150:py3 l=vim.bindeval('l')
151:py3 l[-10:2]=['f']
152:$put =string(l)
153:let l = [0, 1, 2, 3]
154:py3 l=vim.bindeval('l')
155:py3 l[2:-10]=['g']
156:$put =string(l)
157:let l = []
158:py3 l=vim.bindeval('l')
159:py3 l[0:0]=['h']
160:$put =string(l)
161:"
162:" Locked variables
163:let l = [0, 1, 2, 3]
164:py3 l=vim.bindeval('l')
165:lockvar! l
166:py3 l[2]='i'
167:$put =string(l)
168:unlockvar! l
169:"
170:" Function calls
171:function New(...)
172:return ['NewStart']+a:000+['NewEnd']
173:endfunction
174:function DictNew(...) dict
175:return ['DictNewStart']+a:000+['DictNewEnd', self]
176:endfunction
177:let l=[function('New'), function('DictNew')]
178:py3 l=vim.bindeval('l')
179:py3 l.extend(list(l[0](1, 2, 3)))
180:$put =string(l)
181:py3 l.extend(list(l[1](1, 2, 3, self={'a': 'b'})))
182:$put =string(l)
183:py3 l+=[l[0].name]
184:$put =string(l)
185:try
186: py3 l[1](1, 2, 3)
187:catch
188: $put =v:exception[:13]
189:endtry
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200190:py3 f=l[0]
Bram Moolenaardb913952012-06-29 12:54:53 +0200191:delfunction New
192:try
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200193: py3 f(1, 2, 3)
Bram Moolenaardb913952012-06-29 12:54:53 +0200194:catch
195: $put =v:exception[:13]
196:endtry
197:if has('float')
198: let l=[0.0]
199: py3 l=vim.bindeval('l')
200: py3 l.extend([0.0])
201: $put =string(l)
202:else
203: $put ='[0.0, 0.0]'
204:endif
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200205:let messages=[]
206:py3 <<EOF
207d=vim.bindeval('{}')
208m=vim.bindeval('messages')
Bram Moolenaara9922d62013-05-30 13:01:18 +0200209def em(expr, g=globals(), l=locals()):
210 try:
211 exec(expr, g, l)
212 except Exception as e:
213 m.extend([e.__class__.__name__])
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200214
Bram Moolenaara9922d62013-05-30 13:01:18 +0200215em('d["abc"]')
216em('d["abc"]="\\0"')
217em('d["abc"]=vim')
218em('d[""]=1')
219em('d["a\\0b"]=1')
220em('d[b"a\\0b"]=1')
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200221
Bram Moolenaara9922d62013-05-30 13:01:18 +0200222em('d.pop("abc")')
Bram Moolenaarde71b562013-06-02 17:41:54 +0200223em('d.popitem()')
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200224EOF
225:$put =messages
Bram Moolenaar66b79852012-09-21 14:00:35 +0200226:unlet messages
227:" locked and scope attributes
228:let d={} | let dl={} | lockvar dl
229:for s in split("d dl v: g:")
230: let name=tr(s, ':', 's')
231: execute 'py3 '.name.'=vim.bindeval("'.s.'")'
232: let toput=s.' : '.join(map(['locked', 'scope'], 'v:val.":".py3eval(name.".".v:val)'), ';')
233: $put =toput
234:endfor
235:silent! let d.abc=1
236:silent! let dl.abc=1
237:py3 d.locked=True
238:py3 dl.locked=False
239:silent! let d.def=1
240:silent! let dl.def=1
241:put ='d:'.string(d)
242:put ='dl:'.string(dl)
243:unlet d dl
244:
245:let l=[] | let ll=[] | lockvar ll
246:for s in split("l ll")
247: let name=tr(s, ':', 's')
248: execute 'py3 '.name.'=vim.bindeval("'.s.'")'
249: let toput=s.' : locked:'.py3eval(name.'.locked')
250: $put =toput
251:endfor
252:silent! call extend(l, [0])
253:silent! call extend(ll, [0])
254:py3 l.locked=True
255:py3 ll.locked=False
256:silent! call extend(l, [1])
257:silent! call extend(ll, [1])
258:put ='l:'.string(l)
259:put ='ll:'.string(ll)
260:unlet l ll
Bram Moolenaardb913952012-06-29 12:54:53 +0200261:"
262:" py3eval()
263:let l=py3eval('[0, 1, 2]')
264:$put =string(l)
265:let d=py3eval('{"a": "b", "c": 1, "d": ["e"]}')
266:$put =sort(items(d))
Bram Moolenaardb913952012-06-29 12:54:53 +0200267:if has('float')
268: let f=py3eval('0.0')
269: $put =string(f)
270:else
271: $put ='0.0'
272:endif
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200273:" Invalid values:
274:for e in ['"\0"', '{"\0": 1}', 'undefined_name', 'vim']
275: try
276: let v=py3eval(e)
277: catch
278: let toput=e.":\t".v:exception[:13]
279: $put =toput
280: endtry
281:endfor
Bram Moolenaar76d711c2013-02-13 14:17:08 +0100282:"
283:" threading
284:let l = [0]
285:py3 l=vim.bindeval('l')
286:py3 <<EOF
287import threading
288import time
289
290class T(threading.Thread):
291 def __init__(self):
292 threading.Thread.__init__(self)
293 self.t = 0
294 self.running = True
295
296 def run(self):
297 while self.running:
298 self.t += 1
299 time.sleep(0.1)
300
301t = T()
302t.start()
303EOF
304:sleep 1
305:py3 t.running = False
306:py3 t.join()
307:py3 l[0] = t.t > 8 # check if the background thread is working
308:$put =string(l)
309:"
310:" settrace
311:let l = []
312:py3 l=vim.bindeval('l')
313:py3 <<EOF
314import sys
315
316def traceit(frame, event, arg):
317 global l
318 if event == "line":
319 l += [frame.f_lineno]
320 return traceit
321
322def trace_main():
323 for i in range(5):
324 pass
325EOF
326:py3 sys.settrace(traceit)
327:py3 trace_main()
328:py3 sys.settrace(None)
329:$put =string(l)
Bram Moolenaar230bb3f2013-04-24 14:07:45 +0200330:"
331:" Vars
332:let g:foo = 'bac'
333:let w:abc = 'def'
334:let b:baz = 'bar'
Bram Moolenaara4720012013-05-15 16:27:37 +0200335:let t:bar = 'jkl'
Bram Moolenaar230bb3f2013-04-24 14:07:45 +0200336:try
337: throw "Abc"
338:catch
339: put =py3eval('vim.vvars[''exception'']')
340:endtry
341:put =py3eval('vim.vars[''foo'']')
342:put =py3eval('vim.current.window.vars[''abc'']')
343:put =py3eval('vim.current.buffer.vars[''baz'']')
Bram Moolenaara4720012013-05-15 16:27:37 +0200344:put =py3eval('vim.current.tabpage.vars[''bar'']')
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200345:"
346:" Options
347:" paste: boolean, global
348:" previewheight number, global
349:" operatorfunc: string, global
350:" number: boolean, window-local
351:" numberwidth: number, window-local
352:" colorcolumn: string, window-local
353:" statusline: string, window-local/global
354:" autoindent: boolean, buffer-local
Bram Moolenaar55b8ad32013-05-17 13:38:04 +0200355:" shiftwidth: number, buffer-local
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200356:" omnifunc: string, buffer-local
357:" preserveindent: boolean, buffer-local/global
358:" path: string, buffer-local/global
359:let g:bufs=[bufnr('%')]
360:new
361:let g:bufs+=[bufnr('%')]
362:vnew
363:let g:bufs+=[bufnr('%')]
364:wincmd j
365:vnew
366:let g:bufs+=[bufnr('%')]
367:wincmd l
368:fun RecVars(opt)
369: let gval =string(eval('&g:'.a:opt))
370: let wvals=join(map(range(1, 4), 'v:val.":".string(getwinvar(v:val, "&".a:opt))'))
371: let bvals=join(map(copy(g:bufs), 'v:val.":".string(getbufvar(v:val, "&".a:opt))'))
372: put =' G: '.gval
373: put =' W: '.wvals
374: put =' B: '.wvals
375:endfun
376py3 << EOF
377def e(s, g=globals(), l=locals()):
378 try:
379 exec(s, g, l)
380 except Exception as e:
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200381 vim.command('return ' + repr(e.__class__.__name__))
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200382
383def ev(s, g=globals(), l=locals()):
384 try:
385 return eval(s, g, l)
386 except Exception as e:
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200387 vim.command('let exc=' + repr(e.__class__.__name__))
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200388 return 0
389EOF
390:function E(s)
391: python3 e(vim.eval('a:s'))
392:endfunction
393:function Ev(s)
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200394: let r=py3eval('ev(vim.eval("a:s"))')
395: if exists('exc')
396: throw exc
397: endif
398: return r
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200399:endfunction
400:py3 gopts1=vim.options
401:py3 wopts1=vim.windows[2].options
402:py3 wopts2=vim.windows[0].options
403:py3 wopts3=vim.windows[1].options
404:py3 bopts1=vim.buffers[vim.bindeval("g:bufs")[2]].options
405:py3 bopts2=vim.buffers[vim.bindeval("g:bufs")[1]].options
406:py3 bopts3=vim.buffers[vim.bindeval("g:bufs")[0]].options
Bram Moolenaar04188112013-06-01 20:32:12 +0200407:set path=.,..,,
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200408:let lst=[]
409:let lst+=[['paste', 1, 0, 1, 2, 1, 1, 0 ]]
410:let lst+=[['previewheight', 5, 1, 6, 'a', 0, 1, 0 ]]
411:let lst+=[['operatorfunc', 'A', 'B', 'C', 2, 0, 1, 0 ]]
412:let lst+=[['number', 0, 1, 1, 0, 1, 0, 1 ]]
413:let lst+=[['numberwidth', 2, 3, 5, -100, 0, 0, 1 ]]
414:let lst+=[['colorcolumn', '+1', '+2', '+3', 'abc', 0, 0, 1 ]]
415:let lst+=[['statusline', '1', '2', '4', 0, 0, 1, 1 ]]
416:let lst+=[['autoindent', 0, 1, 1, 2, 1, 0, 2 ]]
Bram Moolenaar55b8ad32013-05-17 13:38:04 +0200417:let lst+=[['shiftwidth', 0, 2, 1, 3, 0, 0, 2 ]]
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200418:let lst+=[['omnifunc', 'A', 'B', 'C', 1, 0, 0, 2 ]]
419:let lst+=[['preserveindent', 0, 1, 1, 2, 1, 1, 2 ]]
420:let lst+=[['path', '.,,', ',,', '.', 0, 0, 1, 2 ]]
421:for [oname, oval1, oval2, oval3, invval, bool, global, local] in lst
422: py3 oname=vim.eval('oname')
423: py3 oval1=vim.bindeval('oval1')
424: py3 oval2=vim.bindeval('oval2')
425: py3 oval3=vim.bindeval('oval3')
426: if invval is 0 || invval is 1
427: py3 invval=bool(vim.bindeval('invval'))
428: else
429: py3 invval=vim.bindeval('invval')
430: endif
431: if bool
432: py3 oval1=bool(oval1)
433: py3 oval2=bool(oval2)
434: py3 oval3=bool(oval3)
435: endif
436: put ='>>> '.oname
437: for v in ['gopts1', 'wopts1', 'bopts1']
438: try
439: put =' p/'.v.': '.Ev('repr('.v.'['''.oname.'''])')
440: catch
441: put =' p/'.v.'! '.v:exception
442: endtry
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200443: let r=E(v.'['''.oname.''']=invval')
444: if r isnot 0
445: put =' inv: '.string(invval).'! '.r
446: endif
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200447: for vv in (v is# 'gopts1' ? [v] : [v, v[:-2].'2', v[:-2].'3'])
448: let val=substitute(vv, '^.opts', 'oval', '')
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200449: let r=E(vv.'['''.oname.''']='.val)
450: if r isnot 0
451: put =' '.vv.'! '.r
452: endif
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200453: endfor
454: endfor
455: call RecVars(oname)
456: for v in ['wopts3', 'bopts3']
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200457: let r=E('del '.v.'["'.oname.'"]')
458: if r isnot 0
459: put =' del '.v.'! '.r
460: endif
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200461: endfor
462: call RecVars(oname)
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200463:endfor
464:only
Bram Moolenaarb6c589a2013-05-15 14:39:52 +0200465:for buf in g:bufs[1:]
466: execute 'bwipeout!' buf
467:endfor
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200468:"
469:" Test buffer object
470:vnew
471:put ='First line'
472:put ='Second line'
473:put ='Third line'
474:1 delete _
475:py3 b=vim.current.buffer
476:wincmd w
477:mark a
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200478:augroup BUFS
479: autocmd BufFilePost * python3 cb.append(vim.eval('expand("<abuf>")') + ':BufFilePost:' + vim.eval('bufnr("%")'))
480: autocmd BufFilePre * python3 cb.append(vim.eval('expand("<abuf>")') + ':BufFilePre:' + vim.eval('bufnr("%")'))
481:augroup END
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200482py3 << EOF
483cb = vim.current.buffer
484# Tests BufferAppend and BufferItem
485cb.append(b[0])
486# Tests BufferSlice and BufferAssSlice
487cb.append('abc') # Will be overwritten
488cb[-1:] = b[:-2]
489# Test BufferLength and BufferAssSlice
490cb.append('def') # Will not be overwritten
491cb[len(cb):] = b[:]
492# Test BufferAssItem and BufferMark
493cb.append('ghi') # Will be overwritten
494cb[-1] = repr((len(cb) - cb.mark('a')[0], cb.mark('a')[1]))
495# Test BufferRepr
496cb.append(repr(cb) + repr(b))
497# Modify foreign buffer
498b.append('foo')
499b[0]='bar'
500b[0:0]=['baz']
501vim.command('call append("$", getbufline(%i, 1, "$"))' % b.number)
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200502# Test assigning to name property
Bram Moolenaar04188112013-06-01 20:32:12 +0200503import os
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200504old_name = cb.name
505cb.name = 'foo'
Bram Moolenaar04188112013-06-01 20:32:12 +0200506cb.append(cb.name[-11:].replace(os.path.sep, '/'))
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200507b.name = 'bar'
Bram Moolenaar04188112013-06-01 20:32:12 +0200508cb.append(b.name[-11:].replace(os.path.sep, '/'))
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200509cb.name = old_name
Bram Moolenaar04188112013-06-01 20:32:12 +0200510cb.append(cb.name[-17:].replace(os.path.sep, '/'))
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200511# Test CheckBuffer
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200512for _b in vim.buffers:
513 if _b is not cb:
514 vim.command('bwipeout! ' + str(_b.number))
515del _b
Bram Moolenaar9e822c02013-05-29 22:15:30 +0200516cb.append('valid: b:%s, cb:%s' % (repr(b.valid), repr(cb.valid)))
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200517for expr in ('b[1]','b[:] = ["A", "B"]','b[:]','b.append("abc")'):
518 try:
519 exec(expr)
520 except vim.error:
521 pass
522 else:
523 # Usually a SEGV here
524 # Should not happen in any case
525 cb.append('No exception for ' + expr)
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200526vim.command('cd .')
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200527EOF
Bram Moolenaarb6c589a2013-05-15 14:39:52 +0200528:"
529:" Test vim.buffers object
530:set hidden
531:edit a
532:buffer #
533:edit b
534:buffer #
535:edit c
536:buffer #
537py3 << EOF
538# Check GCing iterator that was not fully exhausted
539i = iter(vim.buffers)
540cb.append('i:' + str(next(i)))
541# and also check creating more then one iterator at a time
542i2 = iter(vim.buffers)
543cb.append('i2:' + str(next(i2)))
544cb.append('i:' + str(next(i)))
545# The following should trigger GC and not cause any problems
546del i
547del i2
548i3 = iter(vim.buffers)
549cb.append('i3:' + str(next(i3)))
550del i3
551
552prevnum = 0
553for b in vim.buffers:
554 # Check buffer order
555 if prevnum >= b.number:
556 cb.append('!!! Buffer numbers not in strictly ascending order')
557 # Check indexing: vim.buffers[number].number == number
558 cb.append(str(b.number) + ':' + repr(vim.buffers[b.number]) + '=' + repr(b))
559 prevnum = b.number
560
561cb.append(str(len(vim.buffers)))
562
563bnums = list(map(lambda b: b.number, vim.buffers))[1:]
564
565# Test wiping out buffer with existing iterator
566i4 = iter(vim.buffers)
567cb.append('i4:' + str(next(i4)))
568vim.command('bwipeout! ' + str(bnums.pop(0)))
569try:
570 next(i4)
571except vim.error:
572 pass
573else:
574 cb.append('!!!! No vim.error')
575i4 = iter(vim.buffers)
576vim.command('bwipeout! ' + str(bnums.pop(-1)))
577vim.command('bwipeout! ' + str(bnums.pop(-1)))
578cb.append('i4:' + str(next(i4)))
579try:
580 next(i4)
581except StopIteration:
582 cb.append('StopIteration')
583EOF
Bram Moolenaara4720012013-05-15 16:27:37 +0200584:"
585:" Test vim.{tabpage,window}list and vim.{tabpage,window} objects
586:tabnew 0
587:tabnew 1
588:vnew a.1
589:tabnew 2
590:vnew a.2
591:vnew b.2
592:vnew c.2
593py3 << EOF
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200594cb.append('Number of tabs: ' + str(len(vim.tabpages)))
595cb.append('Current tab pages:')
596
Bram Moolenaara4720012013-05-15 16:27:37 +0200597def W(w):
598 if '(unknown)' in repr(w):
599 return '<window object (unknown)>'
600 else:
601 return repr(w)
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200602
603def Cursor(w, start=len(cb)):
604 if w.buffer is cb:
605 return repr((start - w.cursor[0], w.cursor[1]))
606 else:
607 return repr(w.cursor)
608
Bram Moolenaara4720012013-05-15 16:27:37 +0200609for t in vim.tabpages:
610 cb.append(' ' + repr(t) + '(' + str(t.number) + ')' + ': ' + str(len(t.windows)) + ' windows, current is ' + W(t.window))
611 cb.append(' Windows:')
612 for w in t.windows:
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200613 cb.append(' ' + W(w) + '(' + str(w.number) + ')' + ': displays buffer ' + repr(w.buffer) + '; cursor is at ' + Cursor(w))
Bram Moolenaara4720012013-05-15 16:27:37 +0200614 # Other values depend on the size of the terminal, so they are checked partly:
615 for attr in ('height', 'row', 'width', 'col'):
616 try:
617 aval = getattr(w, attr)
618 if type(aval) is not int:
619 raise TypeError
620 if aval < 0:
621 raise ValueError
622 except Exception as e:
623 cb.append('!!!!!! Error while getting attribute ' + attr + ': ' + e.__class__.__name__)
624 w.cursor = (len(w.buffer), 0)
625cb.append('Number of windows in current tab page: ' + str(len(vim.windows)))
626if list(vim.windows) != list(vim.current.tabpage.windows):
627 cb.append('!!!!!! Windows differ')
628EOF
629:"
630:" Test vim.current
631py3 << EOF
632def H(o):
633 return repr(o)
634cb.append('Current tab page: ' + repr(vim.current.tabpage))
635cb.append('Current window: ' + repr(vim.current.window) + ': ' + H(vim.current.window) + ' is ' + H(vim.current.tabpage.window))
636cb.append('Current buffer: ' + repr(vim.current.buffer) + ': ' + H(vim.current.buffer) + ' is ' + H(vim.current.window.buffer)+ ' is ' + H(vim.current.tabpage.window.buffer))
637# Assigning: fails
638try:
639 vim.current.window = vim.tabpages[0].window
640except ValueError:
641 cb.append('ValueError at assigning foreign tab window')
642
643for attr in ('window', 'tabpage', 'buffer'):
644 try:
645 setattr(vim.current, attr, None)
646 except TypeError:
647 cb.append('Type error at assigning None to vim.current.' + attr)
648
649# Assigning: success
650vim.current.tabpage = vim.tabpages[-2]
651vim.current.buffer = cb
652vim.current.window = vim.windows[0]
653vim.current.window.cursor = (len(vim.current.buffer), 0)
654cb.append('Current tab page: ' + repr(vim.current.tabpage))
655cb.append('Current window: ' + repr(vim.current.window))
656cb.append('Current buffer: ' + repr(vim.current.buffer))
657cb.append('Current line: ' + repr(vim.current.line))
Bram Moolenaar9e822c02013-05-29 22:15:30 +0200658ws = list(vim.windows)
659ts = list(vim.tabpages)
Bram Moolenaara4720012013-05-15 16:27:37 +0200660for b in vim.buffers:
661 if b is not cb:
662 vim.command('bwipeout! ' + str(b.number))
Bram Moolenaar9e822c02013-05-29 22:15:30 +0200663cb.append('w.valid: ' + repr([w.valid for w in ws]))
664cb.append('t.valid: ' + repr([t.valid for t in ts]))
Bram Moolenaara4720012013-05-15 16:27:37 +0200665EOF
666:tabonly!
667:only!
Bram Moolenaarcac867a2013-05-21 19:50:34 +0200668:"
669:" Test types
670py3 << EOF
671for expr, attr in (
672 ('vim.vars', 'Dictionary'),
673 ('vim.options', 'Options'),
674 ('vim.bindeval("{}")', 'Dictionary'),
675 ('vim.bindeval("[]")', 'List'),
676 ('vim.bindeval("function(\'tr\')")', 'Function'),
677 ('vim.current.buffer', 'Buffer'),
678 ('vim.current.range', 'Range'),
679 ('vim.current.window', 'Window'),
680 ('vim.current.tabpage', 'TabPage'),
681):
682 cb.append(expr + ':' + attr + ':' + repr(type(eval(expr)) is getattr(vim, attr)))
683EOF
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200684:"
Bram Moolenaardd8aca62013-05-29 22:36:10 +0200685:" Test __dir__() method
686py3 << EOF
687for name, o in (
688 ('current', vim.current),
689 ('buffer', vim.current.buffer),
690 ('window', vim.current.window),
691 ('tabpage', vim.current.tabpage),
692 ('range', vim.current.range),
693 ('dictionary', vim.bindeval('{}')),
694 ('list', vim.bindeval('[]')),
695 ('function', vim.bindeval('function("tr")')),
696 ('output', sys.stdout),
697 ):
698 cb.append(name + ':' + ','.join(dir(o)))
699del name
700del o
701EOF
702:"
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200703:" Test vim.*.__new__
Bram Moolenaara9922d62013-05-30 13:01:18 +0200704:$put =string(py3eval('vim.Dictionary({})'))
705:$put =string(py3eval('vim.Dictionary(a=1)'))
706:$put =string(py3eval('vim.Dictionary(((''a'', 1),))'))
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200707:$put =string(py3eval('vim.List()'))
708:$put =string(py3eval('vim.List(iter(''abc''))'))
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200709:$put =string(py3eval('vim.Function(''tr'')'))
Bram Moolenaar01a7a722013-05-30 12:26:58 +0200710:"
711:" Test stdout/stderr
712:redir => messages
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200713:py3 sys.stdout.write('abc') ; sys.stdout.write('def')
714:py3 sys.stderr.write('abc') ; sys.stderr.write('def')
715:py3 sys.stdout.writelines(iter('abc'))
716:py3 sys.stderr.writelines(iter('abc'))
Bram Moolenaar01a7a722013-05-30 12:26:58 +0200717:redir END
718:$put =string(substitute(messages, '\d\+', '', 'g'))
Bram Moolenaara9922d62013-05-30 13:01:18 +0200719:" Test subclassing
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200720:fun Put(...)
721: $put =string(a:000)
722: return a:000
723:endfun
Bram Moolenaara9922d62013-05-30 13:01:18 +0200724py3 << EOF
725class DupDict(vim.Dictionary):
726 def __setitem__(self, key, value):
727 super(DupDict, self).__setitem__(key, value)
728 super(DupDict, self).__setitem__('dup_' + key, value)
729dd = DupDict()
730dd['a'] = 'b'
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200731
732class DupList(vim.List):
733 def __getitem__(self, idx):
734 return [super(DupList, self).__getitem__(idx)] * 2
735
736dl = DupList()
737dl2 = DupList(iter('abc'))
738dl.extend(dl2[0])
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200739
740class DupFun(vim.Function):
741 def __call__(self, arg):
742 return super(DupFun, self).__call__(arg, arg)
743
744df = DupFun('Put')
Bram Moolenaara9922d62013-05-30 13:01:18 +0200745EOF
746:$put =string(sort(keys(py3eval('dd'))))
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200747:$put =string(py3eval('dl'))
748:$put =string(py3eval('dl2'))
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200749:$put =string(py3eval('df(2)'))
Bram Moolenaar01a7a722013-05-30 12:26:58 +0200750:"
Bram Moolenaar8600e402013-05-30 13:28:41 +0200751:" Test errors
752:fun F() dict
753:endfun
754:fun D()
755:endfun
756py3 << EOF
757def ee(expr, g=globals(), l=locals()):
758 try:
759 try:
760 exec(expr, g, l)
761 except Exception as e:
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200762 if sys.version_info >= (3, 3) and e.__class__ is AttributeError and str(e).find('has no attribute')>=0 and not str(e).startswith("'vim."):
763 cb.append(expr + ':' + repr((e.__class__, AttributeError(str(e)[str(e).rfind(" '") + 2:-1]))))
764 else:
765 cb.append(expr + ':' + repr((e.__class__, e)))
Bram Moolenaar8600e402013-05-30 13:28:41 +0200766 else:
767 cb.append(expr + ':NOT FAILED')
768 except Exception as e:
769 cb.append(expr + '::' + repr((e.__class__, e)))
770
771d = vim.Dictionary()
772ned = vim.Dictionary(foo='bar', baz='abc')
773dl = vim.Dictionary(a=1)
774dl.locked = True
775l = vim.List()
776ll = vim.List('abc')
777ll.locked = True
778f = vim.Function('string')
779fd = vim.Function('F')
780fdel = vim.Function('D')
781vim.command('delfunction D')
782
783def subexpr_test(expr, name, subexprs):
784 cb.append('>>> Testing %s using %s' % (name, expr))
785 for subexpr in subexprs:
786 ee(expr % subexpr)
787 cb.append('<<< Finished')
788
789def stringtochars_test(expr):
790 return subexpr_test(expr, 'StringToChars', (
791 '1', # Fail type checks
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200792 'b"\\0"', # Fail PyString_AsStringAndSize(object, , NULL) check
793 '"\\0"', # Fail PyString_AsStringAndSize(bytes, , NULL) check
Bram Moolenaar8600e402013-05-30 13:28:41 +0200794 ))
795
796class Mapping(object):
797 def __init__(self, d):
798 self.d = d
799
800 def __getitem__(self, key):
801 return self.d[key]
802
803 def keys(self):
804 return self.d.keys()
805
806 def items(self):
807 return self.d.items()
808
809def convertfrompyobject_test(expr, recurse=True):
810 # pydict_to_tv
811 stringtochars_test(expr % '{%s : 1}')
812 if recurse:
813 convertfrompyobject_test(expr % '{"abc" : %s}', False)
814 # pymap_to_tv
815 stringtochars_test(expr % 'Mapping({%s : 1})')
816 if recurse:
817 convertfrompyobject_test(expr % 'Mapping({"abc" : %s})', False)
818 # pyseq_to_tv
819 iter_test(expr)
820 return subexpr_test(expr, 'ConvertFromPyObject', (
821 'None', # Not conversible
822 '{"": 1}', # Empty key not allowed
823 'FailingMapping()', #
824 'FailingMappingKey()', #
825 ))
826
827def convertfrompymapping_test(expr):
828 convertfrompyobject_test(expr)
829 return subexpr_test(expr, 'ConvertFromPyMapping', (
830 '[]',
831 ))
832
833def iter_test(expr):
834 return subexpr_test(expr, '*Iter*', (
835 'FailingIter()',
836 'FailingIterNext()',
837 ))
838
839class FailingTrue(object):
840 def __bool__(self):
841 raise NotImplementedError
842
843class FailingIter(object):
844 def __iter__(self):
845 raise NotImplementedError
846
847class FailingIterNext(object):
848 def __iter__(self):
849 return self
850
851 def __next__(self):
852 raise NotImplementedError
853
854class FailingMappingKey(object):
855 def __getitem__(self, item):
856 raise NotImplementedError
857
858 def keys(self):
859 return list("abc")
860
861class FailingMapping(object):
862 def __getitem__(self):
863 raise NotImplementedError
864
865 def keys(self):
866 raise NotImplementedError
867
868class FailingList(list):
869 def __getitem__(self, idx):
870 if i == 2:
871 raise NotImplementedError
872 else:
873 return super(FailingList, self).__getitem__(idx)
874
875cb.append("> Output")
876cb.append(">> OutputSetattr")
877ee('del sys.stdout.softspace')
878ee('sys.stdout.softspace = []')
879ee('sys.stdout.attr = None')
880cb.append(">> OutputWrite")
881ee('sys.stdout.write(None)')
882cb.append(">> OutputWriteLines")
883ee('sys.stdout.writelines(None)')
884ee('sys.stdout.writelines([1])')
885iter_test('sys.stdout.writelines(%s)')
886cb.append("> VimCommand")
887ee('vim.command(1)')
888#! Not checked: vim->python exceptions translating: checked later
889cb.append("> VimToPython")
890#! Not checked: everything: needs errors in internal python functions
891cb.append("> VimEval")
892ee('vim.eval(1)')
893#! Not checked: everything: needs errors in internal python functions
894cb.append("> VimEvalPy")
895ee('vim.bindeval(1)')
896#! Not checked: vim->python exceptions translating: checked later
897cb.append("> VimStrwidth")
898ee('vim.strwidth(1)')
899cb.append("> Dictionary")
900cb.append(">> DictionaryConstructor")
901ee('vim.Dictionary("abc")')
902##! Not checked: py_dict_alloc failure
903cb.append(">> DictionarySetattr")
904ee('del d.locked')
905ee('d.locked = FailingTrue()')
906ee('vim.vvars.locked = False')
907ee('d.scope = True')
908ee('d.xxx = True')
909cb.append(">> _DictionaryItem")
910ee('d.get("a", 2, 3)')
911stringtochars_test('d.get(%s)')
912ee('d.pop("a")')
913ee('dl.pop("a")')
914cb.append(">> DictionaryIterNext")
915ee('for i in ned: ned["a"] = 1')
916cb.append(">> DictionaryAssItem")
917ee('dl["b"] = 1')
918stringtochars_test('d[%s] = 1')
919convertfrompyobject_test('d["a"] = %s')
920cb.append(">> DictionaryUpdate")
921cb.append(">>> kwargs")
922cb.append(">>> iter")
923ee('d.update(FailingMapping())')
924ee('d.update([FailingIterNext()])')
925iter_test('d.update(%s)')
926convertfrompyobject_test('d.update(%s)')
927stringtochars_test('d.update(((%s, 0),))')
928convertfrompyobject_test('d.update((("a", %s),))')
929cb.append(">> DictionaryPopItem")
930ee('d.popitem(1, 2)')
931cb.append(">> DictionaryHasKey")
932ee('d.has_key()')
933cb.append("> List")
934cb.append(">> ListConstructor")
935ee('vim.List(1, 2)')
936ee('vim.List(a=1)')
937iter_test('vim.List(%s)')
938convertfrompyobject_test('vim.List([%s])')
939cb.append(">> ListItem")
940ee('l[1000]')
941cb.append(">> ListAssItem")
942ee('ll[1] = 2')
943ee('l[1000] = 3')
944cb.append(">> ListAssSlice")
945ee('ll[1:100] = "abc"')
946iter_test('l[:] = %s')
947convertfrompyobject_test('l[:] = [%s]')
948cb.append(">> ListConcatInPlace")
949iter_test('l.extend(%s)')
950convertfrompyobject_test('l.extend([%s])')
951cb.append(">> ListSetattr")
952ee('del l.locked')
953ee('l.locked = FailingTrue()')
954ee('l.xxx = True')
955cb.append("> Function")
956cb.append(">> FunctionConstructor")
957ee('vim.Function("123")')
958ee('vim.Function("xxx_non_existent_function_xxx")')
959ee('vim.Function("xxx#non#existent#function#xxx")')
960cb.append(">> FunctionCall")
961convertfrompyobject_test('f(%s)')
962convertfrompymapping_test('fd(self=%s)')
963cb.append("> TabPage")
964cb.append(">> TabPageAttr")
965ee('vim.current.tabpage.xxx')
966cb.append("> TabList")
967cb.append(">> TabListItem")
968ee('vim.tabpages[1000]')
969cb.append("> Window")
970cb.append(">> WindowAttr")
971ee('vim.current.window.xxx')
972cb.append(">> WindowSetattr")
973ee('vim.current.window.buffer = 0')
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200974ee('vim.current.window.cursor = (100000000, 100000000)')
Bram Moolenaar8600e402013-05-30 13:28:41 +0200975ee('vim.current.window.cursor = True')
976ee('vim.current.window.height = "abc"')
977ee('vim.current.window.width = "abc"')
978ee('vim.current.window.xxxxxx = True')
979cb.append("> WinList")
980cb.append(">> WinListItem")
981ee('vim.windows[1000]')
982cb.append("> Buffer")
983cb.append(">> StringToLine (indirect)")
984ee('vim.current.buffer[0] = "\\na"')
985cb.append(">> SetBufferLine (indirect)")
986ee('vim.current.buffer[0] = True')
987cb.append(">> SetBufferLines (indirect)")
988ee('vim.current.buffer[:] = True')
989ee('vim.current.buffer[:] = ["\\na", "bc"]')
990cb.append(">> InsertBufferLines (indirect)")
991ee('vim.current.buffer.append(None)')
992ee('vim.current.buffer.append(["\\na", "bc"])')
993ee('vim.current.buffer.append("\\nbc")')
994cb.append(">> RBItem")
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200995ee('vim.current.buffer[100000000]')
Bram Moolenaar8600e402013-05-30 13:28:41 +0200996cb.append(">> RBAsItem")
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200997ee('vim.current.buffer[100000000] = ""')
Bram Moolenaar8600e402013-05-30 13:28:41 +0200998cb.append(">> BufferAttr")
999ee('vim.current.buffer.xxx')
1000cb.append(">> BufferSetattr")
1001ee('vim.current.buffer.name = True')
1002ee('vim.current.buffer.xxx = True')
1003cb.append(">> BufferMark")
1004ee('vim.current.buffer.mark(0)')
1005ee('vim.current.buffer.mark("abc")')
1006ee('vim.current.buffer.mark("!")')
1007cb.append(">> BufferRange")
1008ee('vim.current.buffer.range(1, 2, 3)')
1009cb.append("> BufMap")
1010cb.append(">> BufMapItem")
1011ee('vim.buffers[None]')
1012ee('vim.buffers[100000000]')
1013cb.append("> Current")
1014cb.append(">> CurrentGetattr")
1015ee('vim.current.xxx')
1016cb.append(">> CurrentSetattr")
1017ee('vim.current.line = True')
1018ee('vim.current.buffer = True')
1019ee('vim.current.window = True')
1020ee('vim.current.tabpage = True')
1021ee('vim.current.xxx = True')
1022EOF
1023:"
Bram Moolenaara7b64ce2013-05-21 20:40:40 +02001024:" Test exceptions
1025:fun Exe(e)
1026: execute a:e
1027:endfun
1028py3 << EOF
Bram Moolenaara7b64ce2013-05-21 20:40:40 +02001029Exe = vim.bindeval('function("Exe")')
1030ee('vim.command("throw \'abc\'")')
1031ee('Exe("throw \'def\'")')
1032ee('vim.eval("Exe(\'throw \'\'ghi\'\'\')")')
1033ee('vim.eval("Exe(\'echoerr \'\'jkl\'\'\')")')
1034ee('vim.eval("Exe(\'xxx_non_existent_command_xxx\')")')
1035ee('vim.bindeval("Exe(\'xxx_non_existent_command_xxx\')")')
1036EOF
Bram Moolenaardb913952012-06-29 12:54:53 +02001037:endfun
1038:"
1039:call Test()
1040:"
1041:delfunc Test
1042:call garbagecollect(1)
1043:"
1044:/^start:/,$wq! test.out
Bram Moolenaarb6c589a2013-05-15 14:39:52 +02001045:call getchar()
Bram Moolenaardb913952012-06-29 12:54:53 +02001046ENDTEST
1047
1048start: