mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 22:14:27 +08:00
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
import readline
|
|
|
|
from jedi import utils
|
|
from .helpers import TestCase
|
|
|
|
|
|
class TestSetupReadline(TestCase):
|
|
def __init__(self, *args, **kwargs):
|
|
super(type(self), self).__init__(*args, **kwargs)
|
|
self.namespace = dict()
|
|
utils.setup_readline(self.namespace)
|
|
|
|
def completions(self, text):
|
|
completer = readline.get_completer()
|
|
i = 0
|
|
completions = []
|
|
while True:
|
|
completion = completer(text, i)
|
|
if completion is None:
|
|
break
|
|
completions.append(completion)
|
|
i += 1
|
|
return completions
|
|
|
|
def test_simple(self):
|
|
assert self.completions('list') == ['list']
|
|
assert self.completions('importerror') == ['ImportError']
|
|
|
|
def test_nested(self):
|
|
assert self.completions('list.Insert') == ['list.insert']
|
|
assert self.completions('list().Insert') == ['list().insert']
|
|
|
|
def test_magic_methods(self):
|
|
assert self.completions('list.__getitem__') == ['list.__getitem__']
|
|
assert self.completions('list().__getitem__') == ['list().__getitem__']
|
|
|
|
def test_modules(self):
|
|
import sys
|
|
import os
|
|
self.namespace['sys'] = sys
|
|
self.namespace['os'] = os
|
|
|
|
assert self.completions('os.path.join') == ['os.path.join']
|
|
assert self.completions('os.path.join().upper') == ['os.path.join().upper']
|
|
|
|
c = set(['os.' + d for d in dir(os) if d.startswith('ch')])
|
|
assert set(self.completions('os.ch')) == set(c)
|
|
|
|
del self.namespace['sys']
|
|
del self.namespace['os']
|
|
|
|
def test_import(self):
|
|
s = 'from os.path import a'
|
|
assert set(self.completions(s)) == set([s + 'ltsep', s + 'bspath'])
|
|
assert self.completions('import keyword') == ['import keyword']
|
|
|
|
def test_colorama(self):
|
|
"""
|
|
Only test it if colorama library is available.
|
|
|
|
This module is being tested because it uses ``setattr`` at some point,
|
|
which Jedi doesn't understand, but it should still work in the REPL.
|
|
"""
|
|
try:
|
|
# if colorama is installed
|
|
import colorama
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
self.namespace['colorama'] = colorama
|
|
assert self.completions('colorama')
|
|
assert self.completions('colorama.Fore.BLACK') == ['colorama.Fore.BLACK']
|
|
del self.namespace['colorama']
|