blob: 1223ca96aadb134545dd3013deaa44292a4bf067 [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
Bram Moolenaar9f3685a2013-06-12 14:20:36 +02005:set noswapfile
Bram Moolenaardb913952012-06-29 12:54:53 +02006:if !has('python3') | e! test.ok | wq! test.out | endif
Bram Moolenaarc24c1ac2013-05-16 20:47:56 +02007:lang C
Bram Moolenaardb913952012-06-29 12:54:53 +02008:py3 import vim
9:fun Test()
10:let l = []
11:py3 l=vim.bindeval('l')
12:py3 f=vim.bindeval('function("strlen")')
13:" Extending List directly with different types
14:py3 l+=[1, "as'd", [1, 2, f, {'a': 1}]]
15:$put =string(l)
16:$put =string(l[-1])
17:try
18: $put =string(l[-4])
19:catch
20: $put =v:exception[:13]
21:endtry
22:" List assignment
23:py3 l[0]=0
24:$put =string(l)
25:py3 l[-2]=f
26:$put =string(l)
27:"
28:" Extending Dictionary directly with different types
29:let d = {}
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020030:fun d.f()
31: return 1
32:endfun
Bram Moolenaara9922d62013-05-30 13:01:18 +020033py3 << EOF
34d=vim.bindeval('d')
35d['1']='asd'
36d.update(b=[1, 2, f])
37d.update((('-1', {'a': 1}),))
38d.update({'0': -1})
39dk = d.keys()
40dv = d.values()
41di = d.items()
42dk.sort(key=repr)
43dv.sort(key=repr)
44di.sort(key=repr)
45EOF
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020046:$put =py3eval('d[''f''](self={})')
Bram Moolenaara9922d62013-05-30 13:01:18 +020047:$put =py3eval('repr(dk)')
48:$put =substitute(py3eval('repr(dv)'),'0x\x\+','','g')
49:$put =substitute(py3eval('repr(di)'),'0x\x\+','','g')
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020050:for [key, Val] in sort(items(d))
51: $put =string(key) . ' : ' . string(Val)
52: unlet key Val
Bram Moolenaardb913952012-06-29 12:54:53 +020053:endfor
54:"
55:" removing items with del
56:py3 del l[2]
57:$put =string(l)
58:let l = range(8)
59:py3 l=vim.bindeval('l')
60:try
61: py3 del l[:3]
62: py3 del l[1:]
63:catch
64: $put =v:exception
65:endtry
66:$put =string(l)
67:"
68:py3 del d['-1']
Bram Moolenaar355fd9b2013-05-30 13:14:13 +020069:py3 del d['f']
Bram Moolenaara9922d62013-05-30 13:01:18 +020070:$put =string(py3eval('d.get(''b'', 1)'))
71:$put =string(py3eval('d.pop(''b'')'))
72:$put =string(py3eval('d.get(''b'', 1)'))
73:$put =string(py3eval('d.pop(''1'', 2)'))
74:$put =string(py3eval('d.pop(''1'', 2)'))
75:$put =py3eval('repr(d.has_key(''0''))')
76:$put =py3eval('repr(d.has_key(''1''))')
77:$put =py3eval('repr(''0'' in d)')
78:$put =py3eval('repr(''1'' in d)')
79:$put =py3eval('repr(list(iter(d)))')
Bram Moolenaardb913952012-06-29 12:54:53 +020080:$put =string(d)
Bram Moolenaarde71b562013-06-02 17:41:54 +020081:$put =py3eval('repr(d.popitem())')
Bram Moolenaara9922d62013-05-30 13:01:18 +020082:$put =py3eval('repr(d.get(''0''))')
83:$put =py3eval('repr(list(iter(d)))')
Bram Moolenaardb913952012-06-29 12:54:53 +020084:"
85:" removing items out of range: silently skip items that don't exist
86:let l = [0, 1, 2, 3]
87:py3 l=vim.bindeval('l')
88:" The following two ranges delete nothing as they match empty list:
89:py3 del l[2:1]
90:$put =string(l)
91:py3 del l[2:2]
92:$put =string(l)
93:py3 del l[2:3]
94:$put =string(l)
95:let l = [0, 1, 2, 3]
96:py3 l=vim.bindeval('l')
97:py3 del l[2:4]
98:$put =string(l)
99:let l = [0, 1, 2, 3]
100:py3 l=vim.bindeval('l')
101:py3 del l[2:5]
102:$put =string(l)
103:let l = [0, 1, 2, 3]
104:py3 l=vim.bindeval('l')
105:py3 del l[2:6]
106:$put =string(l)
107:let l = [0, 1, 2, 3]
108:py3 l=vim.bindeval('l')
109:" The following two ranges delete nothing as they match empty list:
110:py3 del l[-1:2]
111:$put =string(l)
112:py3 del l[-2:2]
113:$put =string(l)
114:py3 del l[-3:2]
115:$put =string(l)
116:let l = [0, 1, 2, 3]
117:py3 l=vim.bindeval('l')
118:py3 del l[-4:2]
119:$put =string(l)
120:let l = [0, 1, 2, 3]
121:py3 l=vim.bindeval('l')
122:py3 del l[-5:2]
123:$put =string(l)
124:let l = [0, 1, 2, 3]
125:py3 l=vim.bindeval('l')
126:py3 del l[-6:2]
127:$put =string(l)
128:"
129:" Slice assignment to a list
130:let l = [0, 1, 2, 3]
131:py3 l=vim.bindeval('l')
132:py3 l[0:0]=['a']
133:$put =string(l)
134:let l = [0, 1, 2, 3]
135:py3 l=vim.bindeval('l')
136:py3 l[1:2]=['b']
137:$put =string(l)
138:let l = [0, 1, 2, 3]
139:py3 l=vim.bindeval('l')
140:py3 l[2:4]=['c']
141:$put =string(l)
142:let l = [0, 1, 2, 3]
143:py3 l=vim.bindeval('l')
144:py3 l[4:4]=['d']
145:$put =string(l)
146:let l = [0, 1, 2, 3]
147:py3 l=vim.bindeval('l')
148:py3 l[-1:2]=['e']
149:$put =string(l)
150:let l = [0, 1, 2, 3]
151:py3 l=vim.bindeval('l')
152:py3 l[-10:2]=['f']
153:$put =string(l)
154:let l = [0, 1, 2, 3]
155:py3 l=vim.bindeval('l')
156:py3 l[2:-10]=['g']
157:$put =string(l)
158:let l = []
159:py3 l=vim.bindeval('l')
160:py3 l[0:0]=['h']
161:$put =string(l)
162:"
163:" Locked variables
164:let l = [0, 1, 2, 3]
165:py3 l=vim.bindeval('l')
166:lockvar! l
167:py3 l[2]='i'
168:$put =string(l)
169:unlockvar! l
170:"
171:" Function calls
172:function New(...)
173:return ['NewStart']+a:000+['NewEnd']
174:endfunction
175:function DictNew(...) dict
176:return ['DictNewStart']+a:000+['DictNewEnd', self]
177:endfunction
178:let l=[function('New'), function('DictNew')]
179:py3 l=vim.bindeval('l')
180:py3 l.extend(list(l[0](1, 2, 3)))
181:$put =string(l)
182:py3 l.extend(list(l[1](1, 2, 3, self={'a': 'b'})))
183:$put =string(l)
184:py3 l+=[l[0].name]
185:$put =string(l)
186:try
187: py3 l[1](1, 2, 3)
188:catch
189: $put =v:exception[:13]
190:endtry
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200191:py3 f=l[0]
Bram Moolenaardb913952012-06-29 12:54:53 +0200192:delfunction New
193:try
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200194: py3 f(1, 2, 3)
Bram Moolenaardb913952012-06-29 12:54:53 +0200195:catch
196: $put =v:exception[:13]
197:endtry
198:if has('float')
199: let l=[0.0]
200: py3 l=vim.bindeval('l')
201: py3 l.extend([0.0])
202: $put =string(l)
203:else
204: $put ='[0.0, 0.0]'
205:endif
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200206:let messages=[]
207:py3 <<EOF
208d=vim.bindeval('{}')
209m=vim.bindeval('messages')
Bram Moolenaara9922d62013-05-30 13:01:18 +0200210def em(expr, g=globals(), l=locals()):
211 try:
212 exec(expr, g, l)
213 except Exception as e:
214 m.extend([e.__class__.__name__])
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200215
Bram Moolenaara9922d62013-05-30 13:01:18 +0200216em('d["abc"]')
217em('d["abc"]="\\0"')
218em('d["abc"]=vim')
219em('d[""]=1')
220em('d["a\\0b"]=1')
221em('d[b"a\\0b"]=1')
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200222
Bram Moolenaara9922d62013-05-30 13:01:18 +0200223em('d.pop("abc")')
Bram Moolenaarde71b562013-06-02 17:41:54 +0200224em('d.popitem()')
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200225EOF
226:$put =messages
Bram Moolenaar66b79852012-09-21 14:00:35 +0200227:unlet messages
228:" locked and scope attributes
229:let d={} | let dl={} | lockvar dl
230:for s in split("d dl v: g:")
231: let name=tr(s, ':', 's')
232: execute 'py3 '.name.'=vim.bindeval("'.s.'")'
233: let toput=s.' : '.join(map(['locked', 'scope'], 'v:val.":".py3eval(name.".".v:val)'), ';')
234: $put =toput
235:endfor
236:silent! let d.abc=1
237:silent! let dl.abc=1
238:py3 d.locked=True
239:py3 dl.locked=False
240:silent! let d.def=1
241:silent! let dl.def=1
242:put ='d:'.string(d)
243:put ='dl:'.string(dl)
244:unlet d dl
245:
246:let l=[] | let ll=[] | lockvar ll
247:for s in split("l ll")
248: let name=tr(s, ':', 's')
249: execute 'py3 '.name.'=vim.bindeval("'.s.'")'
250: let toput=s.' : locked:'.py3eval(name.'.locked')
251: $put =toput
252:endfor
253:silent! call extend(l, [0])
254:silent! call extend(ll, [0])
255:py3 l.locked=True
256:py3 ll.locked=False
257:silent! call extend(l, [1])
258:silent! call extend(ll, [1])
259:put ='l:'.string(l)
260:put ='ll:'.string(ll)
261:unlet l ll
Bram Moolenaardb913952012-06-29 12:54:53 +0200262:"
263:" py3eval()
264:let l=py3eval('[0, 1, 2]')
265:$put =string(l)
266:let d=py3eval('{"a": "b", "c": 1, "d": ["e"]}')
267:$put =sort(items(d))
Bram Moolenaardb913952012-06-29 12:54:53 +0200268:if has('float')
269: let f=py3eval('0.0')
270: $put =string(f)
271:else
272: $put ='0.0'
273:endif
Bram Moolenaarc11073c2012-09-05 19:17:42 +0200274:" Invalid values:
275:for e in ['"\0"', '{"\0": 1}', 'undefined_name', 'vim']
276: try
277: let v=py3eval(e)
278: catch
279: let toput=e.":\t".v:exception[:13]
280: $put =toput
281: endtry
282:endfor
Bram Moolenaar76d711c2013-02-13 14:17:08 +0100283:"
284:" threading
285:let l = [0]
286:py3 l=vim.bindeval('l')
287:py3 <<EOF
288import threading
289import time
290
291class T(threading.Thread):
292 def __init__(self):
293 threading.Thread.__init__(self)
294 self.t = 0
295 self.running = True
296
297 def run(self):
298 while self.running:
299 self.t += 1
300 time.sleep(0.1)
301
302t = T()
303t.start()
304EOF
305:sleep 1
306:py3 t.running = False
307:py3 t.join()
308:py3 l[0] = t.t > 8 # check if the background thread is working
309:$put =string(l)
310:"
311:" settrace
312:let l = []
313:py3 l=vim.bindeval('l')
314:py3 <<EOF
315import sys
316
317def traceit(frame, event, arg):
318 global l
319 if event == "line":
320 l += [frame.f_lineno]
321 return traceit
322
323def trace_main():
324 for i in range(5):
325 pass
326EOF
327:py3 sys.settrace(traceit)
328:py3 trace_main()
329:py3 sys.settrace(None)
330:$put =string(l)
Bram Moolenaar230bb3f2013-04-24 14:07:45 +0200331:"
332:" Vars
333:let g:foo = 'bac'
334:let w:abc = 'def'
335:let b:baz = 'bar'
Bram Moolenaara4720012013-05-15 16:27:37 +0200336:let t:bar = 'jkl'
Bram Moolenaar230bb3f2013-04-24 14:07:45 +0200337:try
338: throw "Abc"
339:catch
340: put =py3eval('vim.vvars[''exception'']')
341:endtry
342:put =py3eval('vim.vars[''foo'']')
343:put =py3eval('vim.current.window.vars[''abc'']')
344:put =py3eval('vim.current.buffer.vars[''baz'']')
Bram Moolenaara4720012013-05-15 16:27:37 +0200345:put =py3eval('vim.current.tabpage.vars[''bar'']')
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200346:"
347:" Options
348:" paste: boolean, global
349:" previewheight number, global
350:" operatorfunc: string, global
351:" number: boolean, window-local
352:" numberwidth: number, window-local
353:" colorcolumn: string, window-local
354:" statusline: string, window-local/global
355:" autoindent: boolean, buffer-local
Bram Moolenaar55b8ad32013-05-17 13:38:04 +0200356:" shiftwidth: number, buffer-local
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200357:" omnifunc: string, buffer-local
358:" preserveindent: boolean, buffer-local/global
359:" path: string, buffer-local/global
360:let g:bufs=[bufnr('%')]
361:new
362:let g:bufs+=[bufnr('%')]
363:vnew
364:let g:bufs+=[bufnr('%')]
365:wincmd j
366:vnew
367:let g:bufs+=[bufnr('%')]
368:wincmd l
369:fun RecVars(opt)
370: let gval =string(eval('&g:'.a:opt))
371: let wvals=join(map(range(1, 4), 'v:val.":".string(getwinvar(v:val, "&".a:opt))'))
372: let bvals=join(map(copy(g:bufs), 'v:val.":".string(getbufvar(v:val, "&".a:opt))'))
373: put =' G: '.gval
374: put =' W: '.wvals
375: put =' B: '.wvals
376:endfun
377py3 << EOF
378def e(s, g=globals(), l=locals()):
379 try:
380 exec(s, g, l)
381 except Exception as e:
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200382 vim.command('return ' + repr(e.__class__.__name__))
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200383
384def ev(s, g=globals(), l=locals()):
385 try:
386 return eval(s, g, l)
387 except Exception as e:
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200388 vim.command('let exc=' + repr(e.__class__.__name__))
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200389 return 0
390EOF
391:function E(s)
392: python3 e(vim.eval('a:s'))
393:endfunction
394:function Ev(s)
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200395: let r=py3eval('ev(vim.eval("a:s"))')
396: if exists('exc')
397: throw exc
398: endif
399: return r
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200400:endfunction
401:py3 gopts1=vim.options
402:py3 wopts1=vim.windows[2].options
403:py3 wopts2=vim.windows[0].options
404:py3 wopts3=vim.windows[1].options
405:py3 bopts1=vim.buffers[vim.bindeval("g:bufs")[2]].options
406:py3 bopts2=vim.buffers[vim.bindeval("g:bufs")[1]].options
407:py3 bopts3=vim.buffers[vim.bindeval("g:bufs")[0]].options
Bram Moolenaar04188112013-06-01 20:32:12 +0200408:set path=.,..,,
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200409:let lst=[]
410:let lst+=[['paste', 1, 0, 1, 2, 1, 1, 0 ]]
411:let lst+=[['previewheight', 5, 1, 6, 'a', 0, 1, 0 ]]
412:let lst+=[['operatorfunc', 'A', 'B', 'C', 2, 0, 1, 0 ]]
413:let lst+=[['number', 0, 1, 1, 0, 1, 0, 1 ]]
414:let lst+=[['numberwidth', 2, 3, 5, -100, 0, 0, 1 ]]
415:let lst+=[['colorcolumn', '+1', '+2', '+3', 'abc', 0, 0, 1 ]]
416:let lst+=[['statusline', '1', '2', '4', 0, 0, 1, 1 ]]
417:let lst+=[['autoindent', 0, 1, 1, 2, 1, 0, 2 ]]
Bram Moolenaar55b8ad32013-05-17 13:38:04 +0200418:let lst+=[['shiftwidth', 0, 2, 1, 3, 0, 0, 2 ]]
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200419:let lst+=[['omnifunc', 'A', 'B', 'C', 1, 0, 0, 2 ]]
420:let lst+=[['preserveindent', 0, 1, 1, 2, 1, 1, 2 ]]
421:let lst+=[['path', '.,,', ',,', '.', 0, 0, 1, 2 ]]
422:for [oname, oval1, oval2, oval3, invval, bool, global, local] in lst
423: py3 oname=vim.eval('oname')
424: py3 oval1=vim.bindeval('oval1')
425: py3 oval2=vim.bindeval('oval2')
426: py3 oval3=vim.bindeval('oval3')
427: if invval is 0 || invval is 1
428: py3 invval=bool(vim.bindeval('invval'))
429: else
430: py3 invval=vim.bindeval('invval')
431: endif
432: if bool
433: py3 oval1=bool(oval1)
434: py3 oval2=bool(oval2)
435: py3 oval3=bool(oval3)
436: endif
437: put ='>>> '.oname
438: for v in ['gopts1', 'wopts1', 'bopts1']
439: try
440: put =' p/'.v.': '.Ev('repr('.v.'['''.oname.'''])')
441: catch
442: put =' p/'.v.'! '.v:exception
443: endtry
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200444: let r=E(v.'['''.oname.''']=invval')
445: if r isnot 0
446: put =' inv: '.string(invval).'! '.r
447: endif
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200448: for vv in (v is# 'gopts1' ? [v] : [v, v[:-2].'2', v[:-2].'3'])
449: let val=substitute(vv, '^.opts', 'oval', '')
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200450: let r=E(vv.'['''.oname.''']='.val)
451: if r isnot 0
452: put =' '.vv.'! '.r
453: endif
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200454: endfor
455: endfor
456: call RecVars(oname)
457: for v in ['wopts3', 'bopts3']
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200458: let r=E('del '.v.'["'.oname.'"]')
459: if r isnot 0
460: put =' del '.v.'! '.r
461: endif
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200462: endfor
463: call RecVars(oname)
Bram Moolenaar84e0f6c2013-05-06 03:52:55 +0200464:endfor
465:only
Bram Moolenaarb6c589a2013-05-15 14:39:52 +0200466:for buf in g:bufs[1:]
467: execute 'bwipeout!' buf
468:endfor
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200469:"
470:" Test buffer object
471:vnew
472:put ='First line'
473:put ='Second line'
474:put ='Third line'
475:1 delete _
476:py3 b=vim.current.buffer
477:wincmd w
478:mark a
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200479:augroup BUFS
480: autocmd BufFilePost * python3 cb.append(vim.eval('expand("<abuf>")') + ':BufFilePost:' + vim.eval('bufnr("%")'))
481: autocmd BufFilePre * python3 cb.append(vim.eval('expand("<abuf>")') + ':BufFilePre:' + vim.eval('bufnr("%")'))
482:augroup END
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200483py3 << EOF
484cb = vim.current.buffer
485# Tests BufferAppend and BufferItem
486cb.append(b[0])
487# Tests BufferSlice and BufferAssSlice
488cb.append('abc') # Will be overwritten
489cb[-1:] = b[:-2]
490# Test BufferLength and BufferAssSlice
491cb.append('def') # Will not be overwritten
492cb[len(cb):] = b[:]
493# Test BufferAssItem and BufferMark
494cb.append('ghi') # Will be overwritten
495cb[-1] = repr((len(cb) - cb.mark('a')[0], cb.mark('a')[1]))
496# Test BufferRepr
497cb.append(repr(cb) + repr(b))
498# Modify foreign buffer
499b.append('foo')
500b[0]='bar'
501b[0:0]=['baz']
502vim.command('call append("$", getbufline(%i, 1, "$"))' % b.number)
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200503# Test assigning to name property
Bram Moolenaar04188112013-06-01 20:32:12 +0200504import os
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200505old_name = cb.name
506cb.name = 'foo'
Bram Moolenaar04188112013-06-01 20:32:12 +0200507cb.append(cb.name[-11:].replace(os.path.sep, '/'))
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200508b.name = 'bar'
Bram Moolenaar04188112013-06-01 20:32:12 +0200509cb.append(b.name[-11:].replace(os.path.sep, '/'))
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200510cb.name = old_name
Bram Moolenaar04188112013-06-01 20:32:12 +0200511cb.append(cb.name[-17:].replace(os.path.sep, '/'))
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200512# Test CheckBuffer
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200513for _b in vim.buffers:
514 if _b is not cb:
515 vim.command('bwipeout! ' + str(_b.number))
516del _b
Bram Moolenaar9e822c02013-05-29 22:15:30 +0200517cb.append('valid: b:%s, cb:%s' % (repr(b.valid), repr(cb.valid)))
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200518for expr in ('b[1]','b[:] = ["A", "B"]','b[:]','b.append("abc")'):
519 try:
520 exec(expr)
521 except vim.error:
522 pass
523 else:
524 # Usually a SEGV here
525 # Should not happen in any case
526 cb.append('No exception for ' + expr)
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200527vim.command('cd .')
Bram Moolenaarbd80f352013-05-12 21:16:23 +0200528EOF
Bram Moolenaarb6c589a2013-05-15 14:39:52 +0200529:"
530:" Test vim.buffers object
531:set hidden
532:edit a
533:buffer #
534:edit b
535:buffer #
536:edit c
537:buffer #
538py3 << EOF
539# Check GCing iterator that was not fully exhausted
540i = iter(vim.buffers)
541cb.append('i:' + str(next(i)))
542# and also check creating more then one iterator at a time
543i2 = iter(vim.buffers)
544cb.append('i2:' + str(next(i2)))
545cb.append('i:' + str(next(i)))
546# The following should trigger GC and not cause any problems
547del i
548del i2
549i3 = iter(vim.buffers)
550cb.append('i3:' + str(next(i3)))
551del i3
552
553prevnum = 0
554for b in vim.buffers:
555 # Check buffer order
556 if prevnum >= b.number:
557 cb.append('!!! Buffer numbers not in strictly ascending order')
558 # Check indexing: vim.buffers[number].number == number
559 cb.append(str(b.number) + ':' + repr(vim.buffers[b.number]) + '=' + repr(b))
560 prevnum = b.number
561
562cb.append(str(len(vim.buffers)))
563
564bnums = list(map(lambda b: b.number, vim.buffers))[1:]
565
566# Test wiping out buffer with existing iterator
567i4 = iter(vim.buffers)
568cb.append('i4:' + str(next(i4)))
569vim.command('bwipeout! ' + str(bnums.pop(0)))
570try:
571 next(i4)
572except vim.error:
573 pass
574else:
575 cb.append('!!!! No vim.error')
576i4 = iter(vim.buffers)
577vim.command('bwipeout! ' + str(bnums.pop(-1)))
578vim.command('bwipeout! ' + str(bnums.pop(-1)))
579cb.append('i4:' + str(next(i4)))
580try:
581 next(i4)
582except StopIteration:
583 cb.append('StopIteration')
584EOF
Bram Moolenaara4720012013-05-15 16:27:37 +0200585:"
586:" Test vim.{tabpage,window}list and vim.{tabpage,window} objects
587:tabnew 0
588:tabnew 1
589:vnew a.1
590:tabnew 2
591:vnew a.2
592:vnew b.2
593:vnew c.2
594py3 << EOF
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200595cb.append('Number of tabs: ' + str(len(vim.tabpages)))
596cb.append('Current tab pages:')
597
Bram Moolenaara4720012013-05-15 16:27:37 +0200598def W(w):
599 if '(unknown)' in repr(w):
600 return '<window object (unknown)>'
601 else:
602 return repr(w)
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200603
604def Cursor(w, start=len(cb)):
605 if w.buffer is cb:
606 return repr((start - w.cursor[0], w.cursor[1]))
607 else:
608 return repr(w.cursor)
609
Bram Moolenaara4720012013-05-15 16:27:37 +0200610for t in vim.tabpages:
611 cb.append(' ' + repr(t) + '(' + str(t.number) + ')' + ': ' + str(len(t.windows)) + ' windows, current is ' + W(t.window))
612 cb.append(' Windows:')
613 for w in t.windows:
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200614 cb.append(' ' + W(w) + '(' + str(w.number) + ')' + ': displays buffer ' + repr(w.buffer) + '; cursor is at ' + Cursor(w))
Bram Moolenaara4720012013-05-15 16:27:37 +0200615 # Other values depend on the size of the terminal, so they are checked partly:
616 for attr in ('height', 'row', 'width', 'col'):
617 try:
618 aval = getattr(w, attr)
619 if type(aval) is not int:
620 raise TypeError
621 if aval < 0:
622 raise ValueError
623 except Exception as e:
624 cb.append('!!!!!! Error while getting attribute ' + attr + ': ' + e.__class__.__name__)
625 w.cursor = (len(w.buffer), 0)
626cb.append('Number of windows in current tab page: ' + str(len(vim.windows)))
627if list(vim.windows) != list(vim.current.tabpage.windows):
628 cb.append('!!!!!! Windows differ')
629EOF
630:"
631:" Test vim.current
632py3 << EOF
633def H(o):
634 return repr(o)
635cb.append('Current tab page: ' + repr(vim.current.tabpage))
636cb.append('Current window: ' + repr(vim.current.window) + ': ' + H(vim.current.window) + ' is ' + H(vim.current.tabpage.window))
637cb.append('Current buffer: ' + repr(vim.current.buffer) + ': ' + H(vim.current.buffer) + ' is ' + H(vim.current.window.buffer)+ ' is ' + H(vim.current.tabpage.window.buffer))
638# Assigning: fails
639try:
640 vim.current.window = vim.tabpages[0].window
641except ValueError:
642 cb.append('ValueError at assigning foreign tab window')
643
644for attr in ('window', 'tabpage', 'buffer'):
645 try:
646 setattr(vim.current, attr, None)
647 except TypeError:
648 cb.append('Type error at assigning None to vim.current.' + attr)
649
650# Assigning: success
651vim.current.tabpage = vim.tabpages[-2]
652vim.current.buffer = cb
653vim.current.window = vim.windows[0]
654vim.current.window.cursor = (len(vim.current.buffer), 0)
655cb.append('Current tab page: ' + repr(vim.current.tabpage))
656cb.append('Current window: ' + repr(vim.current.window))
657cb.append('Current buffer: ' + repr(vim.current.buffer))
658cb.append('Current line: ' + repr(vim.current.line))
Bram Moolenaar9e822c02013-05-29 22:15:30 +0200659ws = list(vim.windows)
660ts = list(vim.tabpages)
Bram Moolenaara4720012013-05-15 16:27:37 +0200661for b in vim.buffers:
662 if b is not cb:
663 vim.command('bwipeout! ' + str(b.number))
Bram Moolenaar9e822c02013-05-29 22:15:30 +0200664cb.append('w.valid: ' + repr([w.valid for w in ws]))
665cb.append('t.valid: ' + repr([t.valid for t in ts]))
Bram Moolenaara4720012013-05-15 16:27:37 +0200666EOF
667:tabonly!
668:only!
Bram Moolenaarcac867a2013-05-21 19:50:34 +0200669:"
670:" Test types
671py3 << EOF
672for expr, attr in (
673 ('vim.vars', 'Dictionary'),
674 ('vim.options', 'Options'),
675 ('vim.bindeval("{}")', 'Dictionary'),
676 ('vim.bindeval("[]")', 'List'),
677 ('vim.bindeval("function(\'tr\')")', 'Function'),
678 ('vim.current.buffer', 'Buffer'),
679 ('vim.current.range', 'Range'),
680 ('vim.current.window', 'Window'),
681 ('vim.current.tabpage', 'TabPage'),
682):
683 cb.append(expr + ':' + attr + ':' + repr(type(eval(expr)) is getattr(vim, attr)))
684EOF
Bram Moolenaara7b64ce2013-05-21 20:40:40 +0200685:"
Bram Moolenaardd8aca62013-05-29 22:36:10 +0200686:" Test __dir__() method
687py3 << EOF
688for name, o in (
689 ('current', vim.current),
690 ('buffer', vim.current.buffer),
691 ('window', vim.current.window),
692 ('tabpage', vim.current.tabpage),
693 ('range', vim.current.range),
694 ('dictionary', vim.bindeval('{}')),
695 ('list', vim.bindeval('[]')),
696 ('function', vim.bindeval('function("tr")')),
697 ('output', sys.stdout),
698 ):
699 cb.append(name + ':' + ','.join(dir(o)))
700del name
701del o
702EOF
703:"
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200704:" Test vim.*.__new__
Bram Moolenaara9922d62013-05-30 13:01:18 +0200705:$put =string(py3eval('vim.Dictionary({})'))
706:$put =string(py3eval('vim.Dictionary(a=1)'))
707:$put =string(py3eval('vim.Dictionary(((''a'', 1),))'))
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200708:$put =string(py3eval('vim.List()'))
709:$put =string(py3eval('vim.List(iter(''abc''))'))
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200710:$put =string(py3eval('vim.Function(''tr'')'))
Bram Moolenaar01a7a722013-05-30 12:26:58 +0200711:"
712:" Test stdout/stderr
713:redir => messages
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200714:py3 sys.stdout.write('abc') ; sys.stdout.write('def')
715:py3 sys.stderr.write('abc') ; sys.stderr.write('def')
716:py3 sys.stdout.writelines(iter('abc'))
717:py3 sys.stderr.writelines(iter('abc'))
Bram Moolenaar01a7a722013-05-30 12:26:58 +0200718:redir END
719:$put =string(substitute(messages, '\d\+', '', 'g'))
Bram Moolenaara9922d62013-05-30 13:01:18 +0200720:" Test subclassing
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200721:fun Put(...)
722: $put =string(a:000)
723: return a:000
724:endfun
Bram Moolenaara9922d62013-05-30 13:01:18 +0200725py3 << EOF
726class DupDict(vim.Dictionary):
727 def __setitem__(self, key, value):
728 super(DupDict, self).__setitem__(key, value)
729 super(DupDict, self).__setitem__('dup_' + key, value)
730dd = DupDict()
731dd['a'] = 'b'
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200732
733class DupList(vim.List):
734 def __getitem__(self, idx):
735 return [super(DupList, self).__getitem__(idx)] * 2
736
737dl = DupList()
738dl2 = DupList(iter('abc'))
739dl.extend(dl2[0])
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200740
741class DupFun(vim.Function):
742 def __call__(self, arg):
743 return super(DupFun, self).__call__(arg, arg)
744
745df = DupFun('Put')
Bram Moolenaara9922d62013-05-30 13:01:18 +0200746EOF
747:$put =string(sort(keys(py3eval('dd'))))
Bram Moolenaar78cddbe2013-05-30 13:05:58 +0200748:$put =string(py3eval('dl'))
749:$put =string(py3eval('dl2'))
Bram Moolenaar355fd9b2013-05-30 13:14:13 +0200750:$put =string(py3eval('df(2)'))
Bram Moolenaar01a7a722013-05-30 12:26:58 +0200751:"
Bram Moolenaarf4258302013-06-02 18:20:17 +0200752:" Test chdir
753py3 << EOF
754import os
755fnamemodify = vim.Function('fnamemodify')
756cb.append(str(fnamemodify('.', ':p:h:t')))
757cb.append(vim.eval('@%'))
758os.chdir('..')
759cb.append(str(fnamemodify('.', ':p:h:t')))
760cb.append(vim.eval('@%').replace(os.path.sep, '/'))
761os.chdir('testdir')
762cb.append(str(fnamemodify('.', ':p:h:t')))
763cb.append(vim.eval('@%'))
764EOF
765:"
Bram Moolenaar8600e402013-05-30 13:28:41 +0200766:" Test errors
767:fun F() dict
768:endfun
769:fun D()
770:endfun
771py3 << EOF
772def ee(expr, g=globals(), l=locals()):
773 try:
774 try:
775 exec(expr, g, l)
776 except Exception as e:
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200777 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."):
778 cb.append(expr + ':' + repr((e.__class__, AttributeError(str(e)[str(e).rfind(" '") + 2:-1]))))
779 else:
780 cb.append(expr + ':' + repr((e.__class__, e)))
Bram Moolenaar8600e402013-05-30 13:28:41 +0200781 else:
782 cb.append(expr + ':NOT FAILED')
783 except Exception as e:
784 cb.append(expr + '::' + repr((e.__class__, e)))
785
786d = vim.Dictionary()
787ned = vim.Dictionary(foo='bar', baz='abc')
788dl = vim.Dictionary(a=1)
789dl.locked = True
790l = vim.List()
791ll = vim.List('abc')
792ll.locked = True
793f = vim.Function('string')
794fd = vim.Function('F')
795fdel = vim.Function('D')
796vim.command('delfunction D')
797
798def subexpr_test(expr, name, subexprs):
799 cb.append('>>> Testing %s using %s' % (name, expr))
800 for subexpr in subexprs:
801 ee(expr % subexpr)
802 cb.append('<<< Finished')
803
804def stringtochars_test(expr):
805 return subexpr_test(expr, 'StringToChars', (
806 '1', # Fail type checks
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200807 'b"\\0"', # Fail PyString_AsStringAndSize(object, , NULL) check
808 '"\\0"', # Fail PyString_AsStringAndSize(bytes, , NULL) check
Bram Moolenaar8600e402013-05-30 13:28:41 +0200809 ))
810
811class Mapping(object):
812 def __init__(self, d):
813 self.d = d
814
815 def __getitem__(self, key):
816 return self.d[key]
817
818 def keys(self):
819 return self.d.keys()
820
821 def items(self):
822 return self.d.items()
823
824def convertfrompyobject_test(expr, recurse=True):
825 # pydict_to_tv
826 stringtochars_test(expr % '{%s : 1}')
827 if recurse:
828 convertfrompyobject_test(expr % '{"abc" : %s}', False)
829 # pymap_to_tv
830 stringtochars_test(expr % 'Mapping({%s : 1})')
831 if recurse:
832 convertfrompyobject_test(expr % 'Mapping({"abc" : %s})', False)
833 # pyseq_to_tv
834 iter_test(expr)
835 return subexpr_test(expr, 'ConvertFromPyObject', (
836 'None', # Not conversible
Bram Moolenaar78b59572013-06-02 18:54:21 +0200837 '{b"": 1}', # Empty key not allowed
838 '{"": 1}', # Same, but with unicode object
Bram Moolenaar8600e402013-05-30 13:28:41 +0200839 'FailingMapping()', #
840 'FailingMappingKey()', #
841 ))
842
843def convertfrompymapping_test(expr):
844 convertfrompyobject_test(expr)
845 return subexpr_test(expr, 'ConvertFromPyMapping', (
846 '[]',
847 ))
848
849def iter_test(expr):
850 return subexpr_test(expr, '*Iter*', (
851 'FailingIter()',
852 'FailingIterNext()',
853 ))
854
855class FailingTrue(object):
856 def __bool__(self):
857 raise NotImplementedError
858
859class FailingIter(object):
860 def __iter__(self):
861 raise NotImplementedError
862
863class FailingIterNext(object):
864 def __iter__(self):
865 return self
866
867 def __next__(self):
868 raise NotImplementedError
869
870class FailingMappingKey(object):
871 def __getitem__(self, item):
872 raise NotImplementedError
873
874 def keys(self):
875 return list("abc")
876
877class FailingMapping(object):
878 def __getitem__(self):
879 raise NotImplementedError
880
881 def keys(self):
882 raise NotImplementedError
883
884class FailingList(list):
885 def __getitem__(self, idx):
886 if i == 2:
887 raise NotImplementedError
888 else:
889 return super(FailingList, self).__getitem__(idx)
890
891cb.append("> Output")
892cb.append(">> OutputSetattr")
893ee('del sys.stdout.softspace')
894ee('sys.stdout.softspace = []')
895ee('sys.stdout.attr = None')
896cb.append(">> OutputWrite")
897ee('sys.stdout.write(None)')
898cb.append(">> OutputWriteLines")
899ee('sys.stdout.writelines(None)')
900ee('sys.stdout.writelines([1])')
901iter_test('sys.stdout.writelines(%s)')
902cb.append("> VimCommand")
903ee('vim.command(1)')
904#! Not checked: vim->python exceptions translating: checked later
905cb.append("> VimToPython")
906#! Not checked: everything: needs errors in internal python functions
907cb.append("> VimEval")
908ee('vim.eval(1)')
909#! Not checked: everything: needs errors in internal python functions
910cb.append("> VimEvalPy")
911ee('vim.bindeval(1)')
912#! Not checked: vim->python exceptions translating: checked later
913cb.append("> VimStrwidth")
914ee('vim.strwidth(1)')
915cb.append("> Dictionary")
916cb.append(">> DictionaryConstructor")
917ee('vim.Dictionary("abc")')
918##! Not checked: py_dict_alloc failure
919cb.append(">> DictionarySetattr")
920ee('del d.locked')
921ee('d.locked = FailingTrue()')
922ee('vim.vvars.locked = False')
923ee('d.scope = True')
924ee('d.xxx = True')
925cb.append(">> _DictionaryItem")
926ee('d.get("a", 2, 3)')
927stringtochars_test('d.get(%s)')
928ee('d.pop("a")')
929ee('dl.pop("a")')
930cb.append(">> DictionaryIterNext")
931ee('for i in ned: ned["a"] = 1')
932cb.append(">> DictionaryAssItem")
933ee('dl["b"] = 1')
934stringtochars_test('d[%s] = 1')
935convertfrompyobject_test('d["a"] = %s')
936cb.append(">> DictionaryUpdate")
937cb.append(">>> kwargs")
938cb.append(">>> iter")
939ee('d.update(FailingMapping())')
940ee('d.update([FailingIterNext()])')
941iter_test('d.update(%s)')
942convertfrompyobject_test('d.update(%s)')
943stringtochars_test('d.update(((%s, 0),))')
944convertfrompyobject_test('d.update((("a", %s),))')
945cb.append(">> DictionaryPopItem")
946ee('d.popitem(1, 2)')
947cb.append(">> DictionaryHasKey")
948ee('d.has_key()')
949cb.append("> List")
950cb.append(">> ListConstructor")
951ee('vim.List(1, 2)')
952ee('vim.List(a=1)')
953iter_test('vim.List(%s)')
954convertfrompyobject_test('vim.List([%s])')
955cb.append(">> ListItem")
956ee('l[1000]')
957cb.append(">> ListAssItem")
958ee('ll[1] = 2')
959ee('l[1000] = 3')
960cb.append(">> ListAssSlice")
961ee('ll[1:100] = "abc"')
962iter_test('l[:] = %s')
963convertfrompyobject_test('l[:] = [%s]')
964cb.append(">> ListConcatInPlace")
965iter_test('l.extend(%s)')
966convertfrompyobject_test('l.extend([%s])')
967cb.append(">> ListSetattr")
968ee('del l.locked')
969ee('l.locked = FailingTrue()')
970ee('l.xxx = True')
971cb.append("> Function")
972cb.append(">> FunctionConstructor")
973ee('vim.Function("123")')
974ee('vim.Function("xxx_non_existent_function_xxx")')
975ee('vim.Function("xxx#non#existent#function#xxx")')
976cb.append(">> FunctionCall")
977convertfrompyobject_test('f(%s)')
978convertfrompymapping_test('fd(self=%s)')
979cb.append("> TabPage")
980cb.append(">> TabPageAttr")
981ee('vim.current.tabpage.xxx')
982cb.append("> TabList")
983cb.append(">> TabListItem")
984ee('vim.tabpages[1000]')
985cb.append("> Window")
986cb.append(">> WindowAttr")
987ee('vim.current.window.xxx')
988cb.append(">> WindowSetattr")
989ee('vim.current.window.buffer = 0')
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +0200990ee('vim.current.window.cursor = (100000000, 100000000)')
Bram Moolenaar8600e402013-05-30 13:28:41 +0200991ee('vim.current.window.cursor = True')
992ee('vim.current.window.height = "abc"')
993ee('vim.current.window.width = "abc"')
994ee('vim.current.window.xxxxxx = True')
995cb.append("> WinList")
996cb.append(">> WinListItem")
997ee('vim.windows[1000]')
998cb.append("> Buffer")
999cb.append(">> StringToLine (indirect)")
1000ee('vim.current.buffer[0] = "\\na"')
1001cb.append(">> SetBufferLine (indirect)")
1002ee('vim.current.buffer[0] = True')
1003cb.append(">> SetBufferLines (indirect)")
1004ee('vim.current.buffer[:] = True')
1005ee('vim.current.buffer[:] = ["\\na", "bc"]')
1006cb.append(">> InsertBufferLines (indirect)")
1007ee('vim.current.buffer.append(None)')
1008ee('vim.current.buffer.append(["\\na", "bc"])')
1009ee('vim.current.buffer.append("\\nbc")')
1010cb.append(">> RBItem")
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +02001011ee('vim.current.buffer[100000000]')
Bram Moolenaar8600e402013-05-30 13:28:41 +02001012cb.append(">> RBAsItem")
Bram Moolenaar96c7dfd2013-05-31 18:46:11 +02001013ee('vim.current.buffer[100000000] = ""')
Bram Moolenaar8600e402013-05-30 13:28:41 +02001014cb.append(">> BufferAttr")
1015ee('vim.current.buffer.xxx')
1016cb.append(">> BufferSetattr")
1017ee('vim.current.buffer.name = True')
1018ee('vim.current.buffer.xxx = True')
1019cb.append(">> BufferMark")
1020ee('vim.current.buffer.mark(0)')
1021ee('vim.current.buffer.mark("abc")')
1022ee('vim.current.buffer.mark("!")')
1023cb.append(">> BufferRange")
1024ee('vim.current.buffer.range(1, 2, 3)')
1025cb.append("> BufMap")
1026cb.append(">> BufMapItem")
1027ee('vim.buffers[None]')
1028ee('vim.buffers[100000000]')
1029cb.append("> Current")
1030cb.append(">> CurrentGetattr")
1031ee('vim.current.xxx')
1032cb.append(">> CurrentSetattr")
1033ee('vim.current.line = True')
1034ee('vim.current.buffer = True')
1035ee('vim.current.window = True')
1036ee('vim.current.tabpage = True')
1037ee('vim.current.xxx = True')
1038EOF
1039:"
Bram Moolenaara9f22202013-06-11 18:48:21 +02001040:" Test import
1041py3 << EOF
Bram Moolenaar9f3685a2013-06-12 14:20:36 +02001042sys.path.insert(0, os.path.join(os.getcwd(), 'python_before'))
1043sys.path.append(os.path.join(os.getcwd(), 'python_after'))
Bram Moolenaara9f22202013-06-11 18:48:21 +02001044vim.options['rtp'] = os.getcwd().replace(',', '\\,').replace('\\', '\\\\')
1045from module import dir as d
1046from modulex import ddir
1047cb.append(d + ',' + ddir)
Bram Moolenaar9f3685a2013-06-12 14:20:36 +02001048import before
1049cb.append(before.dir)
1050import after
1051cb.append(after.dir)
Bram Moolenaara9f22202013-06-11 18:48:21 +02001052EOF
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02001053:"
Bram Moolenaara7b64ce2013-05-21 20:40:40 +02001054:" Test exceptions
1055:fun Exe(e)
1056: execute a:e
1057:endfun
1058py3 << EOF
Bram Moolenaara7b64ce2013-05-21 20:40:40 +02001059Exe = vim.bindeval('function("Exe")')
1060ee('vim.command("throw \'abc\'")')
1061ee('Exe("throw \'def\'")')
1062ee('vim.eval("Exe(\'throw \'\'ghi\'\'\')")')
1063ee('vim.eval("Exe(\'echoerr \'\'jkl\'\'\')")')
1064ee('vim.eval("Exe(\'xxx_non_existent_command_xxx\')")')
1065ee('vim.bindeval("Exe(\'xxx_non_existent_command_xxx\')")')
1066EOF
Bram Moolenaardb913952012-06-29 12:54:53 +02001067:endfun
1068:"
1069:call Test()
1070:"
1071:delfunc Test
1072:call garbagecollect(1)
1073:"
1074:/^start:/,$wq! test.out
Bram Moolenaarb6c589a2013-05-15 14:39:52 +02001075:call getchar()
Bram Moolenaardb913952012-06-29 12:54:53 +02001076ENDTEST
1077
1078start: