| Tests for various python features. vim: set ft=vim : |
| |
| STARTTEST |
| :so small.vim |
| :if !has('python3') | e! test.ok | wq! test.out | endif |
| :py3 import vim |
| :fun Test() |
| :let l = [] |
| :py3 l=vim.bindeval('l') |
| :py3 f=vim.bindeval('function("strlen")') |
| :" Extending List directly with different types |
| :py3 l+=[1, "as'd", [1, 2, f, {'a': 1}]] |
| :$put =string(l) |
| :$put =string(l[-1]) |
| :try |
| : $put =string(l[-4]) |
| :catch |
| : $put =v:exception[:13] |
| :endtry |
| :" List assignment |
| :py3 l[0]=0 |
| :$put =string(l) |
| :py3 l[-2]=f |
| :$put =string(l) |
| :" |
| :" Extending Dictionary directly with different types |
| :let d = {} |
| :py3 d=vim.bindeval('d') |
| :py3 d['1']='asd' |
| :py3 d['b']=[1, 2, f] |
| :py3 d['-1']={'a': 1} |
| :let dkeys = [] |
| :py3 dk=vim.bindeval('dkeys') |
| :py3 dkeys=d.keys() |
| :py3 dkeys.sort() |
| :py3 dk+=dkeys |
| :$put =string(dkeys) |
| :for [key, val] in sort(items(d)) |
| : $put =string(key) . ' : ' . string(val) |
| : unlet key val |
| :endfor |
| :" |
| :" removing items with del |
| :py3 del l[2] |
| :$put =string(l) |
| :let l = range(8) |
| :py3 l=vim.bindeval('l') |
| :try |
| : py3 del l[:3] |
| : py3 del l[1:] |
| :catch |
| : $put =v:exception |
| :endtry |
| :$put =string(l) |
| :" |
| :py3 del d['-1'] |
| :$put =string(d) |
| :" |
| :" removing items out of range: silently skip items that don't exist |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :" The following two ranges delete nothing as they match empty list: |
| :py3 del l[2:1] |
| :$put =string(l) |
| :py3 del l[2:2] |
| :$put =string(l) |
| :py3 del l[2:3] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 del l[2:4] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 del l[2:5] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 del l[2:6] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :" The following two ranges delete nothing as they match empty list: |
| :py3 del l[-1:2] |
| :$put =string(l) |
| :py3 del l[-2:2] |
| :$put =string(l) |
| :py3 del l[-3:2] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 del l[-4:2] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 del l[-5:2] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 del l[-6:2] |
| :$put =string(l) |
| :" |
| :" Slice assignment to a list |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[0:0]=['a'] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[1:2]=['b'] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[2:4]=['c'] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[4:4]=['d'] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[-1:2]=['e'] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[-10:2]=['f'] |
| :$put =string(l) |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :py3 l[2:-10]=['g'] |
| :$put =string(l) |
| :let l = [] |
| :py3 l=vim.bindeval('l') |
| :py3 l[0:0]=['h'] |
| :$put =string(l) |
| :" |
| :" Locked variables |
| :let l = [0, 1, 2, 3] |
| :py3 l=vim.bindeval('l') |
| :lockvar! l |
| :py3 l[2]='i' |
| :$put =string(l) |
| :unlockvar! l |
| :" |
| :" Function calls |
| :function New(...) |
| :return ['NewStart']+a:000+['NewEnd'] |
| :endfunction |
| :function DictNew(...) dict |
| :return ['DictNewStart']+a:000+['DictNewEnd', self] |
| :endfunction |
| :let l=[function('New'), function('DictNew')] |
| :py3 l=vim.bindeval('l') |
| :py3 l.extend(list(l[0](1, 2, 3))) |
| :$put =string(l) |
| :py3 l.extend(list(l[1](1, 2, 3, self={'a': 'b'}))) |
| :$put =string(l) |
| :py3 l+=[l[0].name] |
| :$put =string(l) |
| :try |
| : py3 l[1](1, 2, 3) |
| :catch |
| : $put =v:exception[:13] |
| :endtry |
| :delfunction New |
| :try |
| : py3 l[0](1, 2, 3) |
| :catch |
| : $put =v:exception[:13] |
| :endtry |
| :if has('float') |
| : let l=[0.0] |
| : py3 l=vim.bindeval('l') |
| : py3 l.extend([0.0]) |
| : $put =string(l) |
| :else |
| : $put ='[0.0, 0.0]' |
| :endif |
| :let messages=[] |
| :py3 <<EOF |
| d=vim.bindeval('{}') |
| m=vim.bindeval('messages') |
| try: |
| d['abc'] |
| except Exception as e: |
| m.extend([e.__class__.__name__]) |
| |
| try: |
| d['abc']="\0" |
| except Exception as e: |
| m.extend([e.__class__.__name__]) |
| |
| try: |
| d['abc']=vim |
| except Exception as e: |
| m.extend([e.__class__.__name__]) |
| |
| try: |
| d['']=1 |
| except Exception as e: |
| m.extend([e.__class__.__name__]) |
| |
| try: |
| d['a\0b']=1 |
| except Exception as e: |
| m.extend([e.__class__.__name__]) |
| |
| try: |
| d[b'a\0b']=1 |
| except Exception as e: |
| m.extend([e.__class__.__name__]) |
| EOF |
| :$put =messages |
| :unlet messages |
| :" locked and scope attributes |
| :let d={} | let dl={} | lockvar dl |
| :for s in split("d dl v: g:") |
| : let name=tr(s, ':', 's') |
| : execute 'py3 '.name.'=vim.bindeval("'.s.'")' |
| : let toput=s.' : '.join(map(['locked', 'scope'], 'v:val.":".py3eval(name.".".v:val)'), ';') |
| : $put =toput |
| :endfor |
| :silent! let d.abc=1 |
| :silent! let dl.abc=1 |
| :py3 d.locked=True |
| :py3 dl.locked=False |
| :silent! let d.def=1 |
| :silent! let dl.def=1 |
| :put ='d:'.string(d) |
| :put ='dl:'.string(dl) |
| :unlet d dl |
| : |
| :let l=[] | let ll=[] | lockvar ll |
| :for s in split("l ll") |
| : let name=tr(s, ':', 's') |
| : execute 'py3 '.name.'=vim.bindeval("'.s.'")' |
| : let toput=s.' : locked:'.py3eval(name.'.locked') |
| : $put =toput |
| :endfor |
| :silent! call extend(l, [0]) |
| :silent! call extend(ll, [0]) |
| :py3 l.locked=True |
| :py3 ll.locked=False |
| :silent! call extend(l, [1]) |
| :silent! call extend(ll, [1]) |
| :put ='l:'.string(l) |
| :put ='ll:'.string(ll) |
| :unlet l ll |
| :" |
| :" py3eval() |
| :let l=py3eval('[0, 1, 2]') |
| :$put =string(l) |
| :let d=py3eval('{"a": "b", "c": 1, "d": ["e"]}') |
| :$put =sort(items(d)) |
| :if has('float') |
| : let f=py3eval('0.0') |
| : $put =string(f) |
| :else |
| : $put ='0.0' |
| :endif |
| :" Invalid values: |
| :for e in ['"\0"', '{"\0": 1}', 'undefined_name', 'vim'] |
| : try |
| : let v=py3eval(e) |
| : catch |
| : let toput=e.":\t".v:exception[:13] |
| : $put =toput |
| : endtry |
| :endfor |
| :" |
| :" threading |
| :let l = [0] |
| :py3 l=vim.bindeval('l') |
| :py3 <<EOF |
| import threading |
| import time |
| |
| class T(threading.Thread): |
| def __init__(self): |
| threading.Thread.__init__(self) |
| self.t = 0 |
| self.running = True |
| |
| def run(self): |
| while self.running: |
| self.t += 1 |
| time.sleep(0.1) |
| |
| t = T() |
| t.start() |
| EOF |
| :sleep 1 |
| :py3 t.running = False |
| :py3 t.join() |
| :py3 l[0] = t.t > 8 # check if the background thread is working |
| :$put =string(l) |
| :" |
| :" settrace |
| :let l = [] |
| :py3 l=vim.bindeval('l') |
| :py3 <<EOF |
| import sys |
| |
| def traceit(frame, event, arg): |
| global l |
| if event == "line": |
| l += [frame.f_lineno] |
| return traceit |
| |
| def trace_main(): |
| for i in range(5): |
| pass |
| EOF |
| :py3 sys.settrace(traceit) |
| :py3 trace_main() |
| :py3 sys.settrace(None) |
| :$put =string(l) |
| :" |
| :" Vars |
| :let g:foo = 'bac' |
| :let w:abc = 'def' |
| :let b:baz = 'bar' |
| :let t:bar = 'jkl' |
| :try |
| : throw "Abc" |
| :catch |
| : put =py3eval('vim.vvars[''exception'']') |
| :endtry |
| :put =py3eval('vim.vars[''foo'']') |
| :put =py3eval('vim.current.window.vars[''abc'']') |
| :put =py3eval('vim.current.buffer.vars[''baz'']') |
| :put =py3eval('vim.current.tabpage.vars[''bar'']') |
| :" |
| :" Options |
| :" paste: boolean, global |
| :" previewheight number, global |
| :" operatorfunc: string, global |
| :" number: boolean, window-local |
| :" numberwidth: number, window-local |
| :" colorcolumn: string, window-local |
| :" statusline: string, window-local/global |
| :" autoindent: boolean, buffer-local |
| :" iminsert: number, buffer-local |
| :" omnifunc: string, buffer-local |
| :" preserveindent: boolean, buffer-local/global |
| :" path: string, buffer-local/global |
| :let g:bufs=[bufnr('%')] |
| :new |
| :let g:bufs+=[bufnr('%')] |
| :vnew |
| :let g:bufs+=[bufnr('%')] |
| :wincmd j |
| :vnew |
| :let g:bufs+=[bufnr('%')] |
| :wincmd l |
| :fun RecVars(opt) |
| : let gval =string(eval('&g:'.a:opt)) |
| : let wvals=join(map(range(1, 4), 'v:val.":".string(getwinvar(v:val, "&".a:opt))')) |
| : let bvals=join(map(copy(g:bufs), 'v:val.":".string(getbufvar(v:val, "&".a:opt))')) |
| : put =' G: '.gval |
| : put =' W: '.wvals |
| : put =' B: '.wvals |
| :endfun |
| py3 << EOF |
| def e(s, g=globals(), l=locals()): |
| try: |
| exec(s, g, l) |
| except Exception as e: |
| vim.command('throw ' + repr(e.__class__.__name__)) |
| |
| def ev(s, g=globals(), l=locals()): |
| try: |
| return eval(s, g, l) |
| except Exception as e: |
| vim.command('throw ' + repr(e.__class__.__name__)) |
| return 0 |
| EOF |
| :function E(s) |
| : python3 e(vim.eval('a:s')) |
| :endfunction |
| :function Ev(s) |
| : return py3eval('ev(vim.eval("a:s"))') |
| :endfunction |
| :py3 gopts1=vim.options |
| :py3 wopts1=vim.windows[2].options |
| :py3 wopts2=vim.windows[0].options |
| :py3 wopts3=vim.windows[1].options |
| :py3 bopts1=vim.buffers[vim.bindeval("g:bufs")[2]].options |
| :py3 bopts2=vim.buffers[vim.bindeval("g:bufs")[1]].options |
| :py3 bopts3=vim.buffers[vim.bindeval("g:bufs")[0]].options |
| :let lst=[] |
| :let lst+=[['paste', 1, 0, 1, 2, 1, 1, 0 ]] |
| :let lst+=[['previewheight', 5, 1, 6, 'a', 0, 1, 0 ]] |
| :let lst+=[['operatorfunc', 'A', 'B', 'C', 2, 0, 1, 0 ]] |
| :let lst+=[['number', 0, 1, 1, 0, 1, 0, 1 ]] |
| :let lst+=[['numberwidth', 2, 3, 5, -100, 0, 0, 1 ]] |
| :let lst+=[['colorcolumn', '+1', '+2', '+3', 'abc', 0, 0, 1 ]] |
| :let lst+=[['statusline', '1', '2', '4', 0, 0, 1, 1 ]] |
| :let lst+=[['autoindent', 0, 1, 1, 2, 1, 0, 2 ]] |
| :let lst+=[['iminsert', 0, 2, 1, 3, 0, 0, 2 ]] |
| :let lst+=[['omnifunc', 'A', 'B', 'C', 1, 0, 0, 2 ]] |
| :let lst+=[['preserveindent', 0, 1, 1, 2, 1, 1, 2 ]] |
| :let lst+=[['path', '.,,', ',,', '.', 0, 0, 1, 2 ]] |
| :for [oname, oval1, oval2, oval3, invval, bool, global, local] in lst |
| : py3 oname=vim.eval('oname') |
| : py3 oval1=vim.bindeval('oval1') |
| : py3 oval2=vim.bindeval('oval2') |
| : py3 oval3=vim.bindeval('oval3') |
| : if invval is 0 || invval is 1 |
| : py3 invval=bool(vim.bindeval('invval')) |
| : else |
| : py3 invval=vim.bindeval('invval') |
| : endif |
| : if bool |
| : py3 oval1=bool(oval1) |
| : py3 oval2=bool(oval2) |
| : py3 oval3=bool(oval3) |
| : endif |
| : put ='>>> '.oname |
| : for v in ['gopts1', 'wopts1', 'bopts1'] |
| : try |
| : put =' p/'.v.': '.Ev('repr('.v.'['''.oname.'''])') |
| : catch |
| : put =' p/'.v.'! '.v:exception |
| : endtry |
| : try |
| : call E(v.'["'.oname.'"]=invval') |
| : catch |
| : put =' inv: '.string(invval).'! '.v:exception |
| : endtry |
| : for vv in (v is# 'gopts1' ? [v] : [v, v[:-2].'2', v[:-2].'3']) |
| : let val=substitute(vv, '^.opts', 'oval', '') |
| : try |
| : call E(vv.'["'.oname.'"]='.val) |
| : catch |
| : put =' '.vv.'! '.v:exception |
| : endtry |
| : endfor |
| : endfor |
| : call RecVars(oname) |
| : for v in ['wopts3', 'bopts3'] |
| : try |
| : call E('del '.v.'["'.oname.'"]') |
| : catch |
| : put =' del '.v.'! '.v:exception |
| : endtry |
| : endfor |
| : call RecVars(oname) |
| :endfor |
| :only |
| :for buf in g:bufs[1:] |
| : execute 'bwipeout!' buf |
| :endfor |
| :" |
| :" Test buffer object |
| :vnew |
| :put ='First line' |
| :put ='Second line' |
| :put ='Third line' |
| :1 delete _ |
| :py3 b=vim.current.buffer |
| :wincmd w |
| :mark a |
| py3 << EOF |
| cb = vim.current.buffer |
| # Tests BufferAppend and BufferItem |
| cb.append(b[0]) |
| # Tests BufferSlice and BufferAssSlice |
| cb.append('abc') # Will be overwritten |
| cb[-1:] = b[:-2] |
| # Test BufferLength and BufferAssSlice |
| cb.append('def') # Will not be overwritten |
| cb[len(cb):] = b[:] |
| # Test BufferAssItem and BufferMark |
| cb.append('ghi') # Will be overwritten |
| cb[-1] = repr((len(cb) - cb.mark('a')[0], cb.mark('a')[1])) |
| # Test BufferRepr |
| cb.append(repr(cb) + repr(b)) |
| # Modify foreign buffer |
| b.append('foo') |
| b[0]='bar' |
| b[0:0]=['baz'] |
| vim.command('call append("$", getbufline(%i, 1, "$"))' % b.number) |
| # Test CheckBuffer |
| vim.command('bwipeout! ' + str(b.number)) |
| for expr in ('b[1]','b[:] = ["A", "B"]','b[:]','b.append("abc")'): |
| try: |
| exec(expr) |
| except vim.error: |
| pass |
| else: |
| # Usually a SEGV here |
| # Should not happen in any case |
| cb.append('No exception for ' + expr) |
| EOF |
| :" |
| :" Test vim.buffers object |
| :set hidden |
| :edit a |
| :buffer # |
| :edit b |
| :buffer # |
| :edit c |
| :buffer # |
| py3 << EOF |
| # Check GCing iterator that was not fully exhausted |
| i = iter(vim.buffers) |
| cb.append('i:' + str(next(i))) |
| # and also check creating more then one iterator at a time |
| i2 = iter(vim.buffers) |
| cb.append('i2:' + str(next(i2))) |
| cb.append('i:' + str(next(i))) |
| # The following should trigger GC and not cause any problems |
| del i |
| del i2 |
| i3 = iter(vim.buffers) |
| cb.append('i3:' + str(next(i3))) |
| del i3 |
| |
| prevnum = 0 |
| for b in vim.buffers: |
| # Check buffer order |
| if prevnum >= b.number: |
| cb.append('!!! Buffer numbers not in strictly ascending order') |
| # Check indexing: vim.buffers[number].number == number |
| cb.append(str(b.number) + ':' + repr(vim.buffers[b.number]) + '=' + repr(b)) |
| prevnum = b.number |
| |
| cb.append(str(len(vim.buffers))) |
| |
| bnums = list(map(lambda b: b.number, vim.buffers))[1:] |
| |
| # Test wiping out buffer with existing iterator |
| i4 = iter(vim.buffers) |
| cb.append('i4:' + str(next(i4))) |
| vim.command('bwipeout! ' + str(bnums.pop(0))) |
| try: |
| next(i4) |
| except vim.error: |
| pass |
| else: |
| cb.append('!!!! No vim.error') |
| i4 = iter(vim.buffers) |
| vim.command('bwipeout! ' + str(bnums.pop(-1))) |
| vim.command('bwipeout! ' + str(bnums.pop(-1))) |
| cb.append('i4:' + str(next(i4))) |
| try: |
| next(i4) |
| except StopIteration: |
| cb.append('StopIteration') |
| EOF |
| :" |
| :" Test vim.{tabpage,window}list and vim.{tabpage,window} objects |
| :tabnew 0 |
| :tabnew 1 |
| :vnew a.1 |
| :tabnew 2 |
| :vnew a.2 |
| :vnew b.2 |
| :vnew c.2 |
| py3 << EOF |
| def W(w): |
| if '(unknown)' in repr(w): |
| return '<window object (unknown)>' |
| else: |
| return repr(w) |
| cb.append('Number of tabs: ' + str(len(vim.tabpages))) |
| cb.append('Current tab pages:') |
| for t in vim.tabpages: |
| cb.append(' ' + repr(t) + '(' + str(t.number) + ')' + ': ' + str(len(t.windows)) + ' windows, current is ' + W(t.window)) |
| cb.append(' Windows:') |
| for w in t.windows: |
| cb.append(' ' + W(w) + '(' + str(w.number) + ')' + ': displays buffer ' + repr(w.buffer) + '; cursor is at ' + repr(w.cursor)) |
| # Other values depend on the size of the terminal, so they are checked partly: |
| for attr in ('height', 'row', 'width', 'col'): |
| try: |
| aval = getattr(w, attr) |
| if type(aval) is not int: |
| raise TypeError |
| if aval < 0: |
| raise ValueError |
| except Exception as e: |
| cb.append('!!!!!! Error while getting attribute ' + attr + ': ' + e.__class__.__name__) |
| w.cursor = (len(w.buffer), 0) |
| cb.append('Number of windows in current tab page: ' + str(len(vim.windows))) |
| if list(vim.windows) != list(vim.current.tabpage.windows): |
| cb.append('!!!!!! Windows differ') |
| EOF |
| :" |
| :" Test vim.current |
| py3 << EOF |
| def H(o): |
| return repr(o) |
| cb.append('Current tab page: ' + repr(vim.current.tabpage)) |
| cb.append('Current window: ' + repr(vim.current.window) + ': ' + H(vim.current.window) + ' is ' + H(vim.current.tabpage.window)) |
| cb.append('Current buffer: ' + repr(vim.current.buffer) + ': ' + H(vim.current.buffer) + ' is ' + H(vim.current.window.buffer)+ ' is ' + H(vim.current.tabpage.window.buffer)) |
| # Assigning: fails |
| try: |
| vim.current.window = vim.tabpages[0].window |
| except ValueError: |
| cb.append('ValueError at assigning foreign tab window') |
| |
| for attr in ('window', 'tabpage', 'buffer'): |
| try: |
| setattr(vim.current, attr, None) |
| except TypeError: |
| cb.append('Type error at assigning None to vim.current.' + attr) |
| |
| # Assigning: success |
| vim.current.tabpage = vim.tabpages[-2] |
| vim.current.buffer = cb |
| vim.current.window = vim.windows[0] |
| vim.current.window.cursor = (len(vim.current.buffer), 0) |
| cb.append('Current tab page: ' + repr(vim.current.tabpage)) |
| cb.append('Current window: ' + repr(vim.current.window)) |
| cb.append('Current buffer: ' + repr(vim.current.buffer)) |
| cb.append('Current line: ' + repr(vim.current.line)) |
| for b in vim.buffers: |
| if b is not cb: |
| vim.command('bwipeout! ' + str(b.number)) |
| EOF |
| :tabonly! |
| :only! |
| :endfun |
| :" |
| :call Test() |
| :" |
| :delfunc Test |
| :call garbagecollect(1) |
| :" |
| :/^start:/,$wq! test.out |
| :call getchar() |
| ENDTEST |
| |
| start: |