##// END OF EJS Templates
Add test for kwarg stripping
Matthias Bussonnier -
Show More
@@ -1,1029 +1,1049 b''
1 # encoding: utf-8
1 # encoding: utf-8
2 """Tests for the IPython tab-completion machinery."""
2 """Tests for the IPython tab-completion machinery."""
3
3
4 # Copyright (c) IPython Development Team.
4 # Copyright (c) IPython Development Team.
5 # Distributed under the terms of the Modified BSD License.
5 # Distributed under the terms of the Modified BSD License.
6
6
7 import os
7 import os
8 import sys
8 import sys
9 import textwrap
9 import textwrap
10 import unittest
10 import unittest
11
11
12 from contextlib import contextmanager
12 from contextlib import contextmanager
13
13
14 import nose.tools as nt
14 import nose.tools as nt
15
15
16 from traitlets.config.loader import Config
16 from traitlets.config.loader import Config
17 from IPython import get_ipython
17 from IPython import get_ipython
18 from IPython.core import completer
18 from IPython.core import completer
19 from IPython.external.decorators import knownfailureif
19 from IPython.external.decorators import knownfailureif
20 from IPython.utils.tempdir import TemporaryDirectory, TemporaryWorkingDirectory
20 from IPython.utils.tempdir import TemporaryDirectory, TemporaryWorkingDirectory
21 from IPython.utils.generics import complete_object
21 from IPython.utils.generics import complete_object
22 from IPython.testing import decorators as dec
22 from IPython.testing import decorators as dec
23
23
24 from IPython.core.completer import (
24 from IPython.core.completer import (
25 Completion, provisionalcompleter, match_dict_keys, _deduplicate_completions)
25 Completion, provisionalcompleter, match_dict_keys, _deduplicate_completions)
26 from nose.tools import assert_in, assert_not_in
26 from nose.tools import assert_in, assert_not_in
27
27
28 #-----------------------------------------------------------------------------
28 #-----------------------------------------------------------------------------
29 # Test functions
29 # Test functions
30 #-----------------------------------------------------------------------------
30 #-----------------------------------------------------------------------------
31
31
32 @contextmanager
32 @contextmanager
33 def greedy_completion():
33 def greedy_completion():
34 ip = get_ipython()
34 ip = get_ipython()
35 greedy_original = ip.Completer.greedy
35 greedy_original = ip.Completer.greedy
36 try:
36 try:
37 ip.Completer.greedy = True
37 ip.Completer.greedy = True
38 yield
38 yield
39 finally:
39 finally:
40 ip.Completer.greedy = greedy_original
40 ip.Completer.greedy = greedy_original
41
41
42 def test_protect_filename():
42 def test_protect_filename():
43 if sys.platform == 'win32':
43 if sys.platform == 'win32':
44 pairs = [('abc','abc'),
44 pairs = [('abc','abc'),
45 (' abc','" abc"'),
45 (' abc','" abc"'),
46 ('a bc','"a bc"'),
46 ('a bc','"a bc"'),
47 ('a bc','"a bc"'),
47 ('a bc','"a bc"'),
48 (' bc','" bc"'),
48 (' bc','" bc"'),
49 ]
49 ]
50 else:
50 else:
51 pairs = [('abc','abc'),
51 pairs = [('abc','abc'),
52 (' abc',r'\ abc'),
52 (' abc',r'\ abc'),
53 ('a bc',r'a\ bc'),
53 ('a bc',r'a\ bc'),
54 ('a bc',r'a\ \ bc'),
54 ('a bc',r'a\ \ bc'),
55 (' bc',r'\ \ bc'),
55 (' bc',r'\ \ bc'),
56 # On posix, we also protect parens and other special characters.
56 # On posix, we also protect parens and other special characters.
57 ('a(bc',r'a\(bc'),
57 ('a(bc',r'a\(bc'),
58 ('a)bc',r'a\)bc'),
58 ('a)bc',r'a\)bc'),
59 ('a( )bc',r'a\(\ \)bc'),
59 ('a( )bc',r'a\(\ \)bc'),
60 ('a[1]bc', r'a\[1\]bc'),
60 ('a[1]bc', r'a\[1\]bc'),
61 ('a{1}bc', r'a\{1\}bc'),
61 ('a{1}bc', r'a\{1\}bc'),
62 ('a#bc', r'a\#bc'),
62 ('a#bc', r'a\#bc'),
63 ('a?bc', r'a\?bc'),
63 ('a?bc', r'a\?bc'),
64 ('a=bc', r'a\=bc'),
64 ('a=bc', r'a\=bc'),
65 ('a\\bc', r'a\\bc'),
65 ('a\\bc', r'a\\bc'),
66 ('a|bc', r'a\|bc'),
66 ('a|bc', r'a\|bc'),
67 ('a;bc', r'a\;bc'),
67 ('a;bc', r'a\;bc'),
68 ('a:bc', r'a\:bc'),
68 ('a:bc', r'a\:bc'),
69 ("a'bc", r"a\'bc"),
69 ("a'bc", r"a\'bc"),
70 ('a*bc', r'a\*bc'),
70 ('a*bc', r'a\*bc'),
71 ('a"bc', r'a\"bc'),
71 ('a"bc', r'a\"bc'),
72 ('a^bc', r'a\^bc'),
72 ('a^bc', r'a\^bc'),
73 ('a&bc', r'a\&bc'),
73 ('a&bc', r'a\&bc'),
74 ]
74 ]
75 # run the actual tests
75 # run the actual tests
76 for s1, s2 in pairs:
76 for s1, s2 in pairs:
77 s1p = completer.protect_filename(s1)
77 s1p = completer.protect_filename(s1)
78 nt.assert_equal(s1p, s2)
78 nt.assert_equal(s1p, s2)
79
79
80
80
81 def check_line_split(splitter, test_specs):
81 def check_line_split(splitter, test_specs):
82 for part1, part2, split in test_specs:
82 for part1, part2, split in test_specs:
83 cursor_pos = len(part1)
83 cursor_pos = len(part1)
84 line = part1+part2
84 line = part1+part2
85 out = splitter.split_line(line, cursor_pos)
85 out = splitter.split_line(line, cursor_pos)
86 nt.assert_equal(out, split)
86 nt.assert_equal(out, split)
87
87
88
88
89 def test_line_split():
89 def test_line_split():
90 """Basic line splitter test with default specs."""
90 """Basic line splitter test with default specs."""
91 sp = completer.CompletionSplitter()
91 sp = completer.CompletionSplitter()
92 # The format of the test specs is: part1, part2, expected answer. Parts 1
92 # The format of the test specs is: part1, part2, expected answer. Parts 1
93 # and 2 are joined into the 'line' sent to the splitter, as if the cursor
93 # and 2 are joined into the 'line' sent to the splitter, as if the cursor
94 # was at the end of part1. So an empty part2 represents someone hitting
94 # was at the end of part1. So an empty part2 represents someone hitting
95 # tab at the end of the line, the most common case.
95 # tab at the end of the line, the most common case.
96 t = [('run some/scrip', '', 'some/scrip'),
96 t = [('run some/scrip', '', 'some/scrip'),
97 ('run scripts/er', 'ror.py foo', 'scripts/er'),
97 ('run scripts/er', 'ror.py foo', 'scripts/er'),
98 ('echo $HOM', '', 'HOM'),
98 ('echo $HOM', '', 'HOM'),
99 ('print sys.pa', '', 'sys.pa'),
99 ('print sys.pa', '', 'sys.pa'),
100 ('print(sys.pa', '', 'sys.pa'),
100 ('print(sys.pa', '', 'sys.pa'),
101 ("execfile('scripts/er", '', 'scripts/er'),
101 ("execfile('scripts/er", '', 'scripts/er'),
102 ('a[x.', '', 'x.'),
102 ('a[x.', '', 'x.'),
103 ('a[x.', 'y', 'x.'),
103 ('a[x.', 'y', 'x.'),
104 ('cd "some_file/', '', 'some_file/'),
104 ('cd "some_file/', '', 'some_file/'),
105 ]
105 ]
106 check_line_split(sp, t)
106 check_line_split(sp, t)
107 # Ensure splitting works OK with unicode by re-running the tests with
107 # Ensure splitting works OK with unicode by re-running the tests with
108 # all inputs turned into unicode
108 # all inputs turned into unicode
109 check_line_split(sp, [ map(str, p) for p in t] )
109 check_line_split(sp, [ map(str, p) for p in t] )
110
110
111
111
112 def test_custom_completion_error():
112 def test_custom_completion_error():
113 """Test that errors from custom attribute completers are silenced."""
113 """Test that errors from custom attribute completers are silenced."""
114 ip = get_ipython()
114 ip = get_ipython()
115 class A(object): pass
115 class A(object): pass
116 ip.user_ns['a'] = A()
116 ip.user_ns['a'] = A()
117
117
118 @complete_object.when_type(A)
118 @complete_object.when_type(A)
119 def complete_A(a, existing_completions):
119 def complete_A(a, existing_completions):
120 raise TypeError("this should be silenced")
120 raise TypeError("this should be silenced")
121
121
122 ip.complete("a.")
122 ip.complete("a.")
123
123
124
124
125 def test_unicode_completions():
125 def test_unicode_completions():
126 ip = get_ipython()
126 ip = get_ipython()
127 # Some strings that trigger different types of completion. Check them both
127 # Some strings that trigger different types of completion. Check them both
128 # in str and unicode forms
128 # in str and unicode forms
129 s = ['ru', '%ru', 'cd /', 'floa', 'float(x)/']
129 s = ['ru', '%ru', 'cd /', 'floa', 'float(x)/']
130 for t in s + list(map(str, s)):
130 for t in s + list(map(str, s)):
131 # We don't need to check exact completion values (they may change
131 # We don't need to check exact completion values (they may change
132 # depending on the state of the namespace, but at least no exceptions
132 # depending on the state of the namespace, but at least no exceptions
133 # should be thrown and the return value should be a pair of text, list
133 # should be thrown and the return value should be a pair of text, list
134 # values.
134 # values.
135 text, matches = ip.complete(t)
135 text, matches = ip.complete(t)
136 nt.assert_true(isinstance(text, str))
136 nt.assert_true(isinstance(text, str))
137 nt.assert_true(isinstance(matches, list))
137 nt.assert_true(isinstance(matches, list))
138
138
139 def test_latex_completions():
139 def test_latex_completions():
140 from IPython.core.latex_symbols import latex_symbols
140 from IPython.core.latex_symbols import latex_symbols
141 import random
141 import random
142 ip = get_ipython()
142 ip = get_ipython()
143 # Test some random unicode symbols
143 # Test some random unicode symbols
144 keys = random.sample(latex_symbols.keys(), 10)
144 keys = random.sample(latex_symbols.keys(), 10)
145 for k in keys:
145 for k in keys:
146 text, matches = ip.complete(k)
146 text, matches = ip.complete(k)
147 nt.assert_equal(len(matches),1)
147 nt.assert_equal(len(matches),1)
148 nt.assert_equal(text, k)
148 nt.assert_equal(text, k)
149 nt.assert_equal(matches[0], latex_symbols[k])
149 nt.assert_equal(matches[0], latex_symbols[k])
150 # Test a more complex line
150 # Test a more complex line
151 text, matches = ip.complete(u'print(\\alpha')
151 text, matches = ip.complete(u'print(\\alpha')
152 nt.assert_equal(text, u'\\alpha')
152 nt.assert_equal(text, u'\\alpha')
153 nt.assert_equal(matches[0], latex_symbols['\\alpha'])
153 nt.assert_equal(matches[0], latex_symbols['\\alpha'])
154 # Test multiple matching latex symbols
154 # Test multiple matching latex symbols
155 text, matches = ip.complete(u'\\al')
155 text, matches = ip.complete(u'\\al')
156 nt.assert_in('\\alpha', matches)
156 nt.assert_in('\\alpha', matches)
157 nt.assert_in('\\aleph', matches)
157 nt.assert_in('\\aleph', matches)
158
158
159
159
160
160
161
161
162 def test_back_latex_completion():
162 def test_back_latex_completion():
163 ip = get_ipython()
163 ip = get_ipython()
164
164
165 # do not return more than 1 matches fro \beta, only the latex one.
165 # do not return more than 1 matches fro \beta, only the latex one.
166 name, matches = ip.complete('\\Ξ²')
166 name, matches = ip.complete('\\Ξ²')
167 nt.assert_equal(len(matches), 1)
167 nt.assert_equal(len(matches), 1)
168 nt.assert_equal(matches[0], '\\beta')
168 nt.assert_equal(matches[0], '\\beta')
169
169
170 def test_back_unicode_completion():
170 def test_back_unicode_completion():
171 ip = get_ipython()
171 ip = get_ipython()
172
172
173 name, matches = ip.complete('\\β…€')
173 name, matches = ip.complete('\\β…€')
174 nt.assert_equal(len(matches), 1)
174 nt.assert_equal(len(matches), 1)
175 nt.assert_equal(matches[0], '\\ROMAN NUMERAL FIVE')
175 nt.assert_equal(matches[0], '\\ROMAN NUMERAL FIVE')
176
176
177
177
178 def test_forward_unicode_completion():
178 def test_forward_unicode_completion():
179 ip = get_ipython()
179 ip = get_ipython()
180
180
181 name, matches = ip.complete('\\ROMAN NUMERAL FIVE')
181 name, matches = ip.complete('\\ROMAN NUMERAL FIVE')
182 nt.assert_equal(len(matches), 1)
182 nt.assert_equal(len(matches), 1)
183 nt.assert_equal(matches[0], 'β…€')
183 nt.assert_equal(matches[0], 'β…€')
184
184
185 @dec.knownfailureif(sys.platform == 'win32', 'Fails if there is a C:\\j... path')
185 @dec.knownfailureif(sys.platform == 'win32', 'Fails if there is a C:\\j... path')
186 def test_no_ascii_back_completion():
186 def test_no_ascii_back_completion():
187 ip = get_ipython()
187 ip = get_ipython()
188 with TemporaryWorkingDirectory(): # Avoid any filename completions
188 with TemporaryWorkingDirectory(): # Avoid any filename completions
189 # single ascii letter that don't have yet completions
189 # single ascii letter that don't have yet completions
190 for letter in 'jJ' :
190 for letter in 'jJ' :
191 name, matches = ip.complete('\\'+letter)
191 name, matches = ip.complete('\\'+letter)
192 nt.assert_equal(matches, [])
192 nt.assert_equal(matches, [])
193
193
194
194
195
195
196
196
197 class CompletionSplitterTestCase(unittest.TestCase):
197 class CompletionSplitterTestCase(unittest.TestCase):
198 def setUp(self):
198 def setUp(self):
199 self.sp = completer.CompletionSplitter()
199 self.sp = completer.CompletionSplitter()
200
200
201 def test_delim_setting(self):
201 def test_delim_setting(self):
202 self.sp.delims = ' '
202 self.sp.delims = ' '
203 nt.assert_equal(self.sp.delims, ' ')
203 nt.assert_equal(self.sp.delims, ' ')
204 nt.assert_equal(self.sp._delim_expr, '[\ ]')
204 nt.assert_equal(self.sp._delim_expr, '[\ ]')
205
205
206 def test_spaces(self):
206 def test_spaces(self):
207 """Test with only spaces as split chars."""
207 """Test with only spaces as split chars."""
208 self.sp.delims = ' '
208 self.sp.delims = ' '
209 t = [('foo', '', 'foo'),
209 t = [('foo', '', 'foo'),
210 ('run foo', '', 'foo'),
210 ('run foo', '', 'foo'),
211 ('run foo', 'bar', 'foo'),
211 ('run foo', 'bar', 'foo'),
212 ]
212 ]
213 check_line_split(self.sp, t)
213 check_line_split(self.sp, t)
214
214
215
215
216 def test_has_open_quotes1():
216 def test_has_open_quotes1():
217 for s in ["'", "'''", "'hi' '"]:
217 for s in ["'", "'''", "'hi' '"]:
218 nt.assert_equal(completer.has_open_quotes(s), "'")
218 nt.assert_equal(completer.has_open_quotes(s), "'")
219
219
220
220
221 def test_has_open_quotes2():
221 def test_has_open_quotes2():
222 for s in ['"', '"""', '"hi" "']:
222 for s in ['"', '"""', '"hi" "']:
223 nt.assert_equal(completer.has_open_quotes(s), '"')
223 nt.assert_equal(completer.has_open_quotes(s), '"')
224
224
225
225
226 def test_has_open_quotes3():
226 def test_has_open_quotes3():
227 for s in ["''", "''' '''", "'hi' 'ipython'"]:
227 for s in ["''", "''' '''", "'hi' 'ipython'"]:
228 nt.assert_false(completer.has_open_quotes(s))
228 nt.assert_false(completer.has_open_quotes(s))
229
229
230
230
231 def test_has_open_quotes4():
231 def test_has_open_quotes4():
232 for s in ['""', '""" """', '"hi" "ipython"']:
232 for s in ['""', '""" """', '"hi" "ipython"']:
233 nt.assert_false(completer.has_open_quotes(s))
233 nt.assert_false(completer.has_open_quotes(s))
234
234
235
235
236 @knownfailureif(sys.platform == 'win32', "abspath completions fail on Windows")
236 @knownfailureif(sys.platform == 'win32', "abspath completions fail on Windows")
237 def test_abspath_file_completions():
237 def test_abspath_file_completions():
238 ip = get_ipython()
238 ip = get_ipython()
239 with TemporaryDirectory() as tmpdir:
239 with TemporaryDirectory() as tmpdir:
240 prefix = os.path.join(tmpdir, 'foo')
240 prefix = os.path.join(tmpdir, 'foo')
241 suffixes = ['1', '2']
241 suffixes = ['1', '2']
242 names = [prefix+s for s in suffixes]
242 names = [prefix+s for s in suffixes]
243 for n in names:
243 for n in names:
244 open(n, 'w').close()
244 open(n, 'w').close()
245
245
246 # Check simple completion
246 # Check simple completion
247 c = ip.complete(prefix)[1]
247 c = ip.complete(prefix)[1]
248 nt.assert_equal(c, names)
248 nt.assert_equal(c, names)
249
249
250 # Now check with a function call
250 # Now check with a function call
251 cmd = 'a = f("%s' % prefix
251 cmd = 'a = f("%s' % prefix
252 c = ip.complete(prefix, cmd)[1]
252 c = ip.complete(prefix, cmd)[1]
253 comp = [prefix+s for s in suffixes]
253 comp = [prefix+s for s in suffixes]
254 nt.assert_equal(c, comp)
254 nt.assert_equal(c, comp)
255
255
256
256
257 def test_local_file_completions():
257 def test_local_file_completions():
258 ip = get_ipython()
258 ip = get_ipython()
259 with TemporaryWorkingDirectory():
259 with TemporaryWorkingDirectory():
260 prefix = './foo'
260 prefix = './foo'
261 suffixes = ['1', '2']
261 suffixes = ['1', '2']
262 names = [prefix+s for s in suffixes]
262 names = [prefix+s for s in suffixes]
263 for n in names:
263 for n in names:
264 open(n, 'w').close()
264 open(n, 'w').close()
265
265
266 # Check simple completion
266 # Check simple completion
267 c = ip.complete(prefix)[1]
267 c = ip.complete(prefix)[1]
268 nt.assert_equal(c, names)
268 nt.assert_equal(c, names)
269
269
270 # Now check with a function call
270 # Now check with a function call
271 cmd = 'a = f("%s' % prefix
271 cmd = 'a = f("%s' % prefix
272 c = ip.complete(prefix, cmd)[1]
272 c = ip.complete(prefix, cmd)[1]
273 comp = set(prefix+s for s in suffixes)
273 comp = set(prefix+s for s in suffixes)
274 nt.assert_true(comp.issubset(set(c)))
274 nt.assert_true(comp.issubset(set(c)))
275
275
276
276
277 def test_quoted_file_completions():
277 def test_quoted_file_completions():
278 ip = get_ipython()
278 ip = get_ipython()
279 with TemporaryWorkingDirectory():
279 with TemporaryWorkingDirectory():
280 name = "foo'bar"
280 name = "foo'bar"
281 open(name, 'w').close()
281 open(name, 'w').close()
282
282
283 # Don't escape Windows
283 # Don't escape Windows
284 escaped = name if sys.platform == "win32" else "foo\\'bar"
284 escaped = name if sys.platform == "win32" else "foo\\'bar"
285
285
286 # Single quote matches embedded single quote
286 # Single quote matches embedded single quote
287 text = "open('foo"
287 text = "open('foo"
288 c = ip.Completer._complete(cursor_line=0,
288 c = ip.Completer._complete(cursor_line=0,
289 cursor_pos=len(text),
289 cursor_pos=len(text),
290 full_text=text)[1]
290 full_text=text)[1]
291 nt.assert_equal(c, [escaped])
291 nt.assert_equal(c, [escaped])
292
292
293 # Double quote requires no escape
293 # Double quote requires no escape
294 text = 'open("foo'
294 text = 'open("foo'
295 c = ip.Completer._complete(cursor_line=0,
295 c = ip.Completer._complete(cursor_line=0,
296 cursor_pos=len(text),
296 cursor_pos=len(text),
297 full_text=text)[1]
297 full_text=text)[1]
298 nt.assert_equal(c, [name])
298 nt.assert_equal(c, [name])
299
299
300 # No quote requires an escape
300 # No quote requires an escape
301 text = '%ls foo'
301 text = '%ls foo'
302 c = ip.Completer._complete(cursor_line=0,
302 c = ip.Completer._complete(cursor_line=0,
303 cursor_pos=len(text),
303 cursor_pos=len(text),
304 full_text=text)[1]
304 full_text=text)[1]
305 nt.assert_equal(c, [escaped])
305 nt.assert_equal(c, [escaped])
306
306
307
307
308 def test_jedi():
308 def test_jedi():
309 """
309 """
310 A couple of issue we had with Jedi
310 A couple of issue we had with Jedi
311 """
311 """
312 ip = get_ipython()
312 ip = get_ipython()
313
313
314 def _test_complete(reason, s, comp, start=None, end=None):
314 def _test_complete(reason, s, comp, start=None, end=None):
315 l = len(s)
315 l = len(s)
316 start = start if start is not None else l
316 start = start if start is not None else l
317 end = end if end is not None else l
317 end = end if end is not None else l
318 with provisionalcompleter():
318 with provisionalcompleter():
319 ip.Completer.use_jedi = True
319 ip.Completer.use_jedi = True
320 completions = set(ip.Completer.completions(s, l))
320 completions = set(ip.Completer.completions(s, l))
321 ip.Completer.use_jedi = False
321 ip.Completer.use_jedi = False
322 assert_in(Completion(start, end, comp), completions, reason)
322 assert_in(Completion(start, end, comp), completions, reason)
323
323
324 def _test_not_complete(reason, s, comp):
324 def _test_not_complete(reason, s, comp):
325 l = len(s)
325 l = len(s)
326 with provisionalcompleter():
326 with provisionalcompleter():
327 ip.Completer.use_jedi = True
327 ip.Completer.use_jedi = True
328 completions = set(ip.Completer.completions(s, l))
328 completions = set(ip.Completer.completions(s, l))
329 ip.Completer.use_jedi = False
329 ip.Completer.use_jedi = False
330 assert_not_in(Completion(l, l, comp), completions, reason)
330 assert_not_in(Completion(l, l, comp), completions, reason)
331
331
332 import jedi
332 import jedi
333 jedi_version = tuple(int(i) for i in jedi.__version__.split('.')[:3])
333 jedi_version = tuple(int(i) for i in jedi.__version__.split('.')[:3])
334 if jedi_version > (0, 10):
334 if jedi_version > (0, 10):
335 yield _test_complete, 'jedi >0.9 should complete and not crash', 'a=1;a.', 'real'
335 yield _test_complete, 'jedi >0.9 should complete and not crash', 'a=1;a.', 'real'
336 yield _test_complete, 'can infer first argument', 'a=(1,"foo");a[0].', 'real'
336 yield _test_complete, 'can infer first argument', 'a=(1,"foo");a[0].', 'real'
337 yield _test_complete, 'can infer second argument', 'a=(1,"foo");a[1].', 'capitalize'
337 yield _test_complete, 'can infer second argument', 'a=(1,"foo");a[1].', 'capitalize'
338 yield _test_complete, 'cover duplicate completions', 'im', 'import', 0, 2
338 yield _test_complete, 'cover duplicate completions', 'im', 'import', 0, 2
339
339
340 yield _test_not_complete, 'does not mix types', 'a=(1,"foo");a[0].', 'capitalize'
340 yield _test_not_complete, 'does not mix types', 'a=(1,"foo");a[0].', 'capitalize'
341
341
342 def test_completion_have_signature():
342 def test_completion_have_signature():
343 """
343 """
344 Lets make sure jedi is capable of pulling out the signature of the function we are completing.
344 Lets make sure jedi is capable of pulling out the signature of the function we are completing.
345 """
345 """
346 ip = get_ipython()
346 ip = get_ipython()
347 with provisionalcompleter():
347 with provisionalcompleter():
348 ip.Completer.use_jedi = True
348 ip.Completer.use_jedi = True
349 completions = ip.Completer.completions('ope', 3)
349 completions = ip.Completer.completions('ope', 3)
350 c = next(completions) # should be `open`
350 c = next(completions) # should be `open`
351 ip.Completer.use_jedi = False
351 ip.Completer.use_jedi = False
352 assert 'file' in c.signature, "Signature of function was not found by completer"
352 assert 'file' in c.signature, "Signature of function was not found by completer"
353 assert 'encoding' in c.signature, "Signature of function was not found by completer"
353 assert 'encoding' in c.signature, "Signature of function was not found by completer"
354
354
355
355
356 def test_deduplicate_completions():
356 def test_deduplicate_completions():
357 """
357 """
358 Test that completions are correctly deduplicated (even if ranges are not the same)
358 Test that completions are correctly deduplicated (even if ranges are not the same)
359 """
359 """
360 ip = get_ipython()
360 ip = get_ipython()
361 ip.ex(textwrap.dedent('''
361 ip.ex(textwrap.dedent('''
362 class Z:
362 class Z:
363 zoo = 1
363 zoo = 1
364 '''))
364 '''))
365 with provisionalcompleter():
365 with provisionalcompleter():
366 ip.Completer.use_jedi = True
366 ip.Completer.use_jedi = True
367 l = list(_deduplicate_completions('Z.z', ip.Completer.completions('Z.z', 3)))
367 l = list(_deduplicate_completions('Z.z', ip.Completer.completions('Z.z', 3)))
368 ip.Completer.use_jedi = False
368 ip.Completer.use_jedi = False
369
369
370 assert len(l) == 1, 'Completions (Z.z<tab>) correctly deduplicate: %s ' % l
370 assert len(l) == 1, 'Completions (Z.z<tab>) correctly deduplicate: %s ' % l
371 assert l[0].text == 'zoo' # and not `it.accumulate`
371 assert l[0].text == 'zoo' # and not `it.accumulate`
372
372
373
373
374 def test_greedy_completions():
374 def test_greedy_completions():
375 """
375 """
376 Test the capability of the Greedy completer.
376 Test the capability of the Greedy completer.
377
377
378 Most of the test here does not really show off the greedy completer, for proof
378 Most of the test here does not really show off the greedy completer, for proof
379 each of the text below now pass with Jedi. The greedy completer is capable of more.
379 each of the text below now pass with Jedi. The greedy completer is capable of more.
380
380
381 See the :any:`test_dict_key_completion_contexts`
381 See the :any:`test_dict_key_completion_contexts`
382
382
383 """
383 """
384 ip = get_ipython()
384 ip = get_ipython()
385 ip.ex('a=list(range(5))')
385 ip.ex('a=list(range(5))')
386 _,c = ip.complete('.',line='a[0].')
386 _,c = ip.complete('.',line='a[0].')
387 nt.assert_false('.real' in c,
387 nt.assert_false('.real' in c,
388 "Shouldn't have completed on a[0]: %s"%c)
388 "Shouldn't have completed on a[0]: %s"%c)
389 with greedy_completion(), provisionalcompleter():
389 with greedy_completion(), provisionalcompleter():
390 def _(line, cursor_pos, expect, message, completion):
390 def _(line, cursor_pos, expect, message, completion):
391 ip.Completer.use_jedi = False
391 ip.Completer.use_jedi = False
392 _,c = ip.complete('.', line=line, cursor_pos=cursor_pos)
392 _,c = ip.complete('.', line=line, cursor_pos=cursor_pos)
393 nt.assert_in(expect, c, message % c)
393 nt.assert_in(expect, c, message % c)
394
394
395 ip.Completer.use_jedi = True
395 ip.Completer.use_jedi = True
396 with provisionalcompleter():
396 with provisionalcompleter():
397 completions = ip.Completer.completions(line, cursor_pos)
397 completions = ip.Completer.completions(line, cursor_pos)
398 nt.assert_in(completion, completions)
398 nt.assert_in(completion, completions)
399
399
400 yield _, 'a[0].', 5, 'a[0].real', "Should have completed on a[0].: %s", Completion(5,5, 'real')
400 yield _, 'a[0].', 5, 'a[0].real', "Should have completed on a[0].: %s", Completion(5,5, 'real')
401 yield _, 'a[0].r', 6, 'a[0].real', "Should have completed on a[0].r: %s", Completion(5,6, 'real')
401 yield _, 'a[0].r', 6, 'a[0].real', "Should have completed on a[0].r: %s", Completion(5,6, 'real')
402
402
403 if sys.version_info > (3, 4):
403 if sys.version_info > (3, 4):
404 yield _, 'a[0].from_', 10, 'a[0].from_bytes', "Should have completed on a[0].from_: %s", Completion(5, 10, 'from_bytes')
404 yield _, 'a[0].from_', 10, 'a[0].from_bytes', "Should have completed on a[0].from_: %s", Completion(5, 10, 'from_bytes')
405
405
406
406
407 def test_omit__names():
407 def test_omit__names():
408 # also happens to test IPCompleter as a configurable
408 # also happens to test IPCompleter as a configurable
409 ip = get_ipython()
409 ip = get_ipython()
410 ip._hidden_attr = 1
410 ip._hidden_attr = 1
411 ip._x = {}
411 ip._x = {}
412 c = ip.Completer
412 c = ip.Completer
413 ip.ex('ip=get_ipython()')
413 ip.ex('ip=get_ipython()')
414 cfg = Config()
414 cfg = Config()
415 cfg.IPCompleter.omit__names = 0
415 cfg.IPCompleter.omit__names = 0
416 c.update_config(cfg)
416 c.update_config(cfg)
417 with provisionalcompleter():
417 with provisionalcompleter():
418 c.use_jedi = False
418 c.use_jedi = False
419 s,matches = c.complete('ip.')
419 s,matches = c.complete('ip.')
420 nt.assert_in('ip.__str__', matches)
420 nt.assert_in('ip.__str__', matches)
421 nt.assert_in('ip._hidden_attr', matches)
421 nt.assert_in('ip._hidden_attr', matches)
422
422
423 # c.use_jedi = True
423 # c.use_jedi = True
424 # completions = set(c.completions('ip.', 3))
424 # completions = set(c.completions('ip.', 3))
425 # nt.assert_in(Completion(3, 3, '__str__'), completions)
425 # nt.assert_in(Completion(3, 3, '__str__'), completions)
426 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
426 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
427
427
428
428
429 cfg = Config()
429 cfg = Config()
430 cfg.IPCompleter.omit__names = 1
430 cfg.IPCompleter.omit__names = 1
431 c.update_config(cfg)
431 c.update_config(cfg)
432 with provisionalcompleter():
432 with provisionalcompleter():
433 c.use_jedi = False
433 c.use_jedi = False
434 s,matches = c.complete('ip.')
434 s,matches = c.complete('ip.')
435 nt.assert_not_in('ip.__str__', matches)
435 nt.assert_not_in('ip.__str__', matches)
436 # nt.assert_in('ip._hidden_attr', matches)
436 # nt.assert_in('ip._hidden_attr', matches)
437
437
438 # c.use_jedi = True
438 # c.use_jedi = True
439 # completions = set(c.completions('ip.', 3))
439 # completions = set(c.completions('ip.', 3))
440 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
440 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
441 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
441 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
442
442
443 cfg = Config()
443 cfg = Config()
444 cfg.IPCompleter.omit__names = 2
444 cfg.IPCompleter.omit__names = 2
445 c.update_config(cfg)
445 c.update_config(cfg)
446 with provisionalcompleter():
446 with provisionalcompleter():
447 c.use_jedi = False
447 c.use_jedi = False
448 s,matches = c.complete('ip.')
448 s,matches = c.complete('ip.')
449 nt.assert_not_in('ip.__str__', matches)
449 nt.assert_not_in('ip.__str__', matches)
450 nt.assert_not_in('ip._hidden_attr', matches)
450 nt.assert_not_in('ip._hidden_attr', matches)
451
451
452 # c.use_jedi = True
452 # c.use_jedi = True
453 # completions = set(c.completions('ip.', 3))
453 # completions = set(c.completions('ip.', 3))
454 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
454 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
455 # nt.assert_not_in(Completion(3,3, "_hidden_attr"), completions)
455 # nt.assert_not_in(Completion(3,3, "_hidden_attr"), completions)
456
456
457 with provisionalcompleter():
457 with provisionalcompleter():
458 c.use_jedi = False
458 c.use_jedi = False
459 s,matches = c.complete('ip._x.')
459 s,matches = c.complete('ip._x.')
460 nt.assert_in('ip._x.keys', matches)
460 nt.assert_in('ip._x.keys', matches)
461
461
462 # c.use_jedi = True
462 # c.use_jedi = True
463 # completions = set(c.completions('ip._x.', 6))
463 # completions = set(c.completions('ip._x.', 6))
464 # nt.assert_in(Completion(6,6, "keys"), completions)
464 # nt.assert_in(Completion(6,6, "keys"), completions)
465
465
466 del ip._hidden_attr
466 del ip._hidden_attr
467 del ip._x
467 del ip._x
468
468
469
469
470 def test_limit_to__all__False_ok():
470 def test_limit_to__all__False_ok():
471 """
471 """
472 Limit to all is deprecated, once we remove it this test can go away.
472 Limit to all is deprecated, once we remove it this test can go away.
473 """
473 """
474 ip = get_ipython()
474 ip = get_ipython()
475 c = ip.Completer
475 c = ip.Completer
476 c.use_jedi = False
476 c.use_jedi = False
477 ip.ex('class D: x=24')
477 ip.ex('class D: x=24')
478 ip.ex('d=D()')
478 ip.ex('d=D()')
479 cfg = Config()
479 cfg = Config()
480 cfg.IPCompleter.limit_to__all__ = False
480 cfg.IPCompleter.limit_to__all__ = False
481 c.update_config(cfg)
481 c.update_config(cfg)
482 s, matches = c.complete('d.')
482 s, matches = c.complete('d.')
483 nt.assert_in('d.x', matches)
483 nt.assert_in('d.x', matches)
484
484
485
485
486 def test_get__all__entries_ok():
486 def test_get__all__entries_ok():
487 class A(object):
487 class A(object):
488 __all__ = ['x', 1]
488 __all__ = ['x', 1]
489 words = completer.get__all__entries(A())
489 words = completer.get__all__entries(A())
490 nt.assert_equal(words, ['x'])
490 nt.assert_equal(words, ['x'])
491
491
492
492
493 def test_get__all__entries_no__all__ok():
493 def test_get__all__entries_no__all__ok():
494 class A(object):
494 class A(object):
495 pass
495 pass
496 words = completer.get__all__entries(A())
496 words = completer.get__all__entries(A())
497 nt.assert_equal(words, [])
497 nt.assert_equal(words, [])
498
498
499
499
500 def test_func_kw_completions():
500 def test_func_kw_completions():
501 ip = get_ipython()
501 ip = get_ipython()
502 c = ip.Completer
502 c = ip.Completer
503 c.use_jedi = False
503 c.use_jedi = False
504 ip.ex('def myfunc(a=1,b=2): return a+b')
504 ip.ex('def myfunc(a=1,b=2): return a+b')
505 s, matches = c.complete(None, 'myfunc(1,b')
505 s, matches = c.complete(None, 'myfunc(1,b')
506 nt.assert_in('b=', matches)
506 nt.assert_in('b=', matches)
507 # Simulate completing with cursor right after b (pos==10):
507 # Simulate completing with cursor right after b (pos==10):
508 s, matches = c.complete(None, 'myfunc(1,b)', 10)
508 s, matches = c.complete(None, 'myfunc(1,b)', 10)
509 nt.assert_in('b=', matches)
509 nt.assert_in('b=', matches)
510 s, matches = c.complete(None, 'myfunc(a="escaped\\")string",b')
510 s, matches = c.complete(None, 'myfunc(a="escaped\\")string",b')
511 nt.assert_in('b=', matches)
511 nt.assert_in('b=', matches)
512 #builtin function
512 #builtin function
513 s, matches = c.complete(None, 'min(k, k')
513 s, matches = c.complete(None, 'min(k, k')
514 nt.assert_in('key=', matches)
514 nt.assert_in('key=', matches)
515
515
516
516
517 def test_default_arguments_from_docstring():
517 def test_default_arguments_from_docstring():
518 ip = get_ipython()
518 ip = get_ipython()
519 c = ip.Completer
519 c = ip.Completer
520 kwd = c._default_arguments_from_docstring(
520 kwd = c._default_arguments_from_docstring(
521 'min(iterable[, key=func]) -> value')
521 'min(iterable[, key=func]) -> value')
522 nt.assert_equal(kwd, ['key'])
522 nt.assert_equal(kwd, ['key'])
523 #with cython type etc
523 #with cython type etc
524 kwd = c._default_arguments_from_docstring(
524 kwd = c._default_arguments_from_docstring(
525 'Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
525 'Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
526 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
526 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
527 #white spaces
527 #white spaces
528 kwd = c._default_arguments_from_docstring(
528 kwd = c._default_arguments_from_docstring(
529 '\n Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
529 '\n Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
530 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
530 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
531
531
532 def test_line_magics():
532 def test_line_magics():
533 ip = get_ipython()
533 ip = get_ipython()
534 c = ip.Completer
534 c = ip.Completer
535 s, matches = c.complete(None, 'lsmag')
535 s, matches = c.complete(None, 'lsmag')
536 nt.assert_in('%lsmagic', matches)
536 nt.assert_in('%lsmagic', matches)
537 s, matches = c.complete(None, '%lsmag')
537 s, matches = c.complete(None, '%lsmag')
538 nt.assert_in('%lsmagic', matches)
538 nt.assert_in('%lsmagic', matches)
539
539
540
540
541 def test_cell_magics():
541 def test_cell_magics():
542 from IPython.core.magic import register_cell_magic
542 from IPython.core.magic import register_cell_magic
543
543
544 @register_cell_magic
544 @register_cell_magic
545 def _foo_cellm(line, cell):
545 def _foo_cellm(line, cell):
546 pass
546 pass
547
547
548 ip = get_ipython()
548 ip = get_ipython()
549 c = ip.Completer
549 c = ip.Completer
550
550
551 s, matches = c.complete(None, '_foo_ce')
551 s, matches = c.complete(None, '_foo_ce')
552 nt.assert_in('%%_foo_cellm', matches)
552 nt.assert_in('%%_foo_cellm', matches)
553 s, matches = c.complete(None, '%%_foo_ce')
553 s, matches = c.complete(None, '%%_foo_ce')
554 nt.assert_in('%%_foo_cellm', matches)
554 nt.assert_in('%%_foo_cellm', matches)
555
555
556
556
557 def test_line_cell_magics():
557 def test_line_cell_magics():
558 from IPython.core.magic import register_line_cell_magic
558 from IPython.core.magic import register_line_cell_magic
559
559
560 @register_line_cell_magic
560 @register_line_cell_magic
561 def _bar_cellm(line, cell):
561 def _bar_cellm(line, cell):
562 pass
562 pass
563
563
564 ip = get_ipython()
564 ip = get_ipython()
565 c = ip.Completer
565 c = ip.Completer
566
566
567 # The policy here is trickier, see comments in completion code. The
567 # The policy here is trickier, see comments in completion code. The
568 # returned values depend on whether the user passes %% or not explicitly,
568 # returned values depend on whether the user passes %% or not explicitly,
569 # and this will show a difference if the same name is both a line and cell
569 # and this will show a difference if the same name is both a line and cell
570 # magic.
570 # magic.
571 s, matches = c.complete(None, '_bar_ce')
571 s, matches = c.complete(None, '_bar_ce')
572 nt.assert_in('%_bar_cellm', matches)
572 nt.assert_in('%_bar_cellm', matches)
573 nt.assert_in('%%_bar_cellm', matches)
573 nt.assert_in('%%_bar_cellm', matches)
574 s, matches = c.complete(None, '%_bar_ce')
574 s, matches = c.complete(None, '%_bar_ce')
575 nt.assert_in('%_bar_cellm', matches)
575 nt.assert_in('%_bar_cellm', matches)
576 nt.assert_in('%%_bar_cellm', matches)
576 nt.assert_in('%%_bar_cellm', matches)
577 s, matches = c.complete(None, '%%_bar_ce')
577 s, matches = c.complete(None, '%%_bar_ce')
578 nt.assert_not_in('%_bar_cellm', matches)
578 nt.assert_not_in('%_bar_cellm', matches)
579 nt.assert_in('%%_bar_cellm', matches)
579 nt.assert_in('%%_bar_cellm', matches)
580
580
581
581
582 def test_magic_completion_order():
582 def test_magic_completion_order():
583 ip = get_ipython()
583 ip = get_ipython()
584 c = ip.Completer
584 c = ip.Completer
585
585
586 # Test ordering of line and cell magics.
586 # Test ordering of line and cell magics.
587 text, matches = c.complete("timeit")
587 text, matches = c.complete("timeit")
588 nt.assert_equal(matches, ["%timeit", "%%timeit"])
588 nt.assert_equal(matches, ["%timeit", "%%timeit"])
589
589
590
590
591 def test_magic_completion_shadowing():
591 def test_magic_completion_shadowing():
592 ip = get_ipython()
592 ip = get_ipython()
593 c = ip.Completer
593 c = ip.Completer
594 c.use_jedi = False
594 c.use_jedi = False
595
595
596 # Before importing matplotlib, %matplotlib magic should be the only option.
596 # Before importing matplotlib, %matplotlib magic should be the only option.
597 text, matches = c.complete("mat")
597 text, matches = c.complete("mat")
598 nt.assert_equal(matches, ["%matplotlib"])
598 nt.assert_equal(matches, ["%matplotlib"])
599
599
600 # The newly introduced name should shadow the magic.
600 # The newly introduced name should shadow the magic.
601 ip.run_cell("matplotlib = 1")
601 ip.run_cell("matplotlib = 1")
602 text, matches = c.complete("mat")
602 text, matches = c.complete("mat")
603 nt.assert_equal(matches, ["matplotlib"])
603 nt.assert_equal(matches, ["matplotlib"])
604
604
605 # After removing matplotlib from namespace, the magic should again be
605 # After removing matplotlib from namespace, the magic should again be
606 # the only option.
606 # the only option.
607 del ip.user_ns["matplotlib"]
607 del ip.user_ns["matplotlib"]
608 text, matches = c.complete("mat")
608 text, matches = c.complete("mat")
609 nt.assert_equal(matches, ["%matplotlib"])
609 nt.assert_equal(matches, ["%matplotlib"])
610
610
611 def test_magic_completion_shadowing_explicit():
611 def test_magic_completion_shadowing_explicit():
612 """
612 """
613 If the user try to complete a shadowed magic, and explicit % start should
613 If the user try to complete a shadowed magic, and explicit % start should
614 still return the completions.
614 still return the completions.
615 """
615 """
616 ip = get_ipython()
616 ip = get_ipython()
617 c = ip.Completer
617 c = ip.Completer
618
618
619 # Before importing matplotlib, %matplotlib magic should be the only option.
619 # Before importing matplotlib, %matplotlib magic should be the only option.
620 text, matches = c.complete("%mat")
620 text, matches = c.complete("%mat")
621 nt.assert_equal(matches, ["%matplotlib"])
621 nt.assert_equal(matches, ["%matplotlib"])
622
622
623 ip.run_cell("matplotlib = 1")
623 ip.run_cell("matplotlib = 1")
624
624
625 # After removing matplotlib from namespace, the magic should still be
625 # After removing matplotlib from namespace, the magic should still be
626 # the only option.
626 # the only option.
627 text, matches = c.complete("%mat")
627 text, matches = c.complete("%mat")
628 nt.assert_equal(matches, ["%matplotlib"])
628 nt.assert_equal(matches, ["%matplotlib"])
629
629
630 def test_magic_config():
630 def test_magic_config():
631 ip = get_ipython()
631 ip = get_ipython()
632 c = ip.Completer
632 c = ip.Completer
633
633
634 s, matches = c.complete(None, 'conf')
634 s, matches = c.complete(None, 'conf')
635 nt.assert_in('%config', matches)
635 nt.assert_in('%config', matches)
636 s, matches = c.complete(None, 'conf')
636 s, matches = c.complete(None, 'conf')
637 nt.assert_not_in('AliasManager', matches)
637 nt.assert_not_in('AliasManager', matches)
638 s, matches = c.complete(None, 'config ')
638 s, matches = c.complete(None, 'config ')
639 nt.assert_in('AliasManager', matches)
639 nt.assert_in('AliasManager', matches)
640 s, matches = c.complete(None, '%config ')
640 s, matches = c.complete(None, '%config ')
641 nt.assert_in('AliasManager', matches)
641 nt.assert_in('AliasManager', matches)
642 s, matches = c.complete(None, 'config Ali')
642 s, matches = c.complete(None, 'config Ali')
643 nt.assert_list_equal(['AliasManager'], matches)
643 nt.assert_list_equal(['AliasManager'], matches)
644 s, matches = c.complete(None, '%config Ali')
644 s, matches = c.complete(None, '%config Ali')
645 nt.assert_list_equal(['AliasManager'], matches)
645 nt.assert_list_equal(['AliasManager'], matches)
646 s, matches = c.complete(None, 'config AliasManager')
646 s, matches = c.complete(None, 'config AliasManager')
647 nt.assert_list_equal(['AliasManager'], matches)
647 nt.assert_list_equal(['AliasManager'], matches)
648 s, matches = c.complete(None, '%config AliasManager')
648 s, matches = c.complete(None, '%config AliasManager')
649 nt.assert_list_equal(['AliasManager'], matches)
649 nt.assert_list_equal(['AliasManager'], matches)
650 s, matches = c.complete(None, 'config AliasManager.')
650 s, matches = c.complete(None, 'config AliasManager.')
651 nt.assert_in('AliasManager.default_aliases', matches)
651 nt.assert_in('AliasManager.default_aliases', matches)
652 s, matches = c.complete(None, '%config AliasManager.')
652 s, matches = c.complete(None, '%config AliasManager.')
653 nt.assert_in('AliasManager.default_aliases', matches)
653 nt.assert_in('AliasManager.default_aliases', matches)
654 s, matches = c.complete(None, 'config AliasManager.de')
654 s, matches = c.complete(None, 'config AliasManager.de')
655 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
655 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
656 s, matches = c.complete(None, 'config AliasManager.de')
656 s, matches = c.complete(None, 'config AliasManager.de')
657 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
657 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
658
658
659
659
660 def test_magic_color():
660 def test_magic_color():
661 ip = get_ipython()
661 ip = get_ipython()
662 c = ip.Completer
662 c = ip.Completer
663
663
664 s, matches = c.complete(None, 'colo')
664 s, matches = c.complete(None, 'colo')
665 nt.assert_in('%colors', matches)
665 nt.assert_in('%colors', matches)
666 s, matches = c.complete(None, 'colo')
666 s, matches = c.complete(None, 'colo')
667 nt.assert_not_in('NoColor', matches)
667 nt.assert_not_in('NoColor', matches)
668 s, matches = c.complete(None, '%colors') # No trailing space
668 s, matches = c.complete(None, '%colors') # No trailing space
669 nt.assert_not_in('NoColor', matches)
669 nt.assert_not_in('NoColor', matches)
670 s, matches = c.complete(None, 'colors ')
670 s, matches = c.complete(None, 'colors ')
671 nt.assert_in('NoColor', matches)
671 nt.assert_in('NoColor', matches)
672 s, matches = c.complete(None, '%colors ')
672 s, matches = c.complete(None, '%colors ')
673 nt.assert_in('NoColor', matches)
673 nt.assert_in('NoColor', matches)
674 s, matches = c.complete(None, 'colors NoCo')
674 s, matches = c.complete(None, 'colors NoCo')
675 nt.assert_list_equal(['NoColor'], matches)
675 nt.assert_list_equal(['NoColor'], matches)
676 s, matches = c.complete(None, '%colors NoCo')
676 s, matches = c.complete(None, '%colors NoCo')
677 nt.assert_list_equal(['NoColor'], matches)
677 nt.assert_list_equal(['NoColor'], matches)
678
678
679
679
680 def test_match_dict_keys():
680 def test_match_dict_keys():
681 """
681 """
682 Test that match_dict_keys works on a couple of use case does return what
682 Test that match_dict_keys works on a couple of use case does return what
683 expected, and does not crash
683 expected, and does not crash
684 """
684 """
685 delims = ' \t\n`!@#$^&*()=+[{]}\\|;:\'",<>?'
685 delims = ' \t\n`!@#$^&*()=+[{]}\\|;:\'",<>?'
686
686
687
687
688 keys = ['foo', b'far']
688 keys = ['foo', b'far']
689 assert match_dict_keys(keys, "b'", delims=delims) == ("'", 2 ,['far'])
689 assert match_dict_keys(keys, "b'", delims=delims) == ("'", 2 ,['far'])
690 assert match_dict_keys(keys, "b'f", delims=delims) == ("'", 2 ,['far'])
690 assert match_dict_keys(keys, "b'f", delims=delims) == ("'", 2 ,['far'])
691 assert match_dict_keys(keys, 'b"', delims=delims) == ('"', 2 ,['far'])
691 assert match_dict_keys(keys, 'b"', delims=delims) == ('"', 2 ,['far'])
692 assert match_dict_keys(keys, 'b"f', delims=delims) == ('"', 2 ,['far'])
692 assert match_dict_keys(keys, 'b"f', delims=delims) == ('"', 2 ,['far'])
693
693
694 assert match_dict_keys(keys, "'", delims=delims) == ("'", 1 ,['foo'])
694 assert match_dict_keys(keys, "'", delims=delims) == ("'", 1 ,['foo'])
695 assert match_dict_keys(keys, "'f", delims=delims) == ("'", 1 ,['foo'])
695 assert match_dict_keys(keys, "'f", delims=delims) == ("'", 1 ,['foo'])
696 assert match_dict_keys(keys, '"', delims=delims) == ('"', 1 ,['foo'])
696 assert match_dict_keys(keys, '"', delims=delims) == ('"', 1 ,['foo'])
697 assert match_dict_keys(keys, '"f', delims=delims) == ('"', 1 ,['foo'])
697 assert match_dict_keys(keys, '"f', delims=delims) == ('"', 1 ,['foo'])
698
698
699 match_dict_keys
699 match_dict_keys
700
700
701
701
702 def test_dict_key_completion_string():
702 def test_dict_key_completion_string():
703 """Test dictionary key completion for string keys"""
703 """Test dictionary key completion for string keys"""
704 ip = get_ipython()
704 ip = get_ipython()
705 complete = ip.Completer.complete
705 complete = ip.Completer.complete
706
706
707 ip.user_ns['d'] = {'abc': None}
707 ip.user_ns['d'] = {'abc': None}
708
708
709 # check completion at different stages
709 # check completion at different stages
710 _, matches = complete(line_buffer="d[")
710 _, matches = complete(line_buffer="d[")
711 nt.assert_in("'abc'", matches)
711 nt.assert_in("'abc'", matches)
712 nt.assert_not_in("'abc']", matches)
712 nt.assert_not_in("'abc']", matches)
713
713
714 _, matches = complete(line_buffer="d['")
714 _, matches = complete(line_buffer="d['")
715 nt.assert_in("abc", matches)
715 nt.assert_in("abc", matches)
716 nt.assert_not_in("abc']", matches)
716 nt.assert_not_in("abc']", matches)
717
717
718 _, matches = complete(line_buffer="d['a")
718 _, matches = complete(line_buffer="d['a")
719 nt.assert_in("abc", matches)
719 nt.assert_in("abc", matches)
720 nt.assert_not_in("abc']", matches)
720 nt.assert_not_in("abc']", matches)
721
721
722 # check use of different quoting
722 # check use of different quoting
723 _, matches = complete(line_buffer="d[\"")
723 _, matches = complete(line_buffer="d[\"")
724 nt.assert_in("abc", matches)
724 nt.assert_in("abc", matches)
725 nt.assert_not_in('abc\"]', matches)
725 nt.assert_not_in('abc\"]', matches)
726
726
727 _, matches = complete(line_buffer="d[\"a")
727 _, matches = complete(line_buffer="d[\"a")
728 nt.assert_in("abc", matches)
728 nt.assert_in("abc", matches)
729 nt.assert_not_in('abc\"]', matches)
729 nt.assert_not_in('abc\"]', matches)
730
730
731 # check sensitivity to following context
731 # check sensitivity to following context
732 _, matches = complete(line_buffer="d[]", cursor_pos=2)
732 _, matches = complete(line_buffer="d[]", cursor_pos=2)
733 nt.assert_in("'abc'", matches)
733 nt.assert_in("'abc'", matches)
734
734
735 _, matches = complete(line_buffer="d['']", cursor_pos=3)
735 _, matches = complete(line_buffer="d['']", cursor_pos=3)
736 nt.assert_in("abc", matches)
736 nt.assert_in("abc", matches)
737 nt.assert_not_in("abc'", matches)
737 nt.assert_not_in("abc'", matches)
738 nt.assert_not_in("abc']", matches)
738 nt.assert_not_in("abc']", matches)
739
739
740 # check multiple solutions are correctly returned and that noise is not
740 # check multiple solutions are correctly returned and that noise is not
741 ip.user_ns['d'] = {'abc': None, 'abd': None, 'bad': None, object(): None,
741 ip.user_ns['d'] = {'abc': None, 'abd': None, 'bad': None, object(): None,
742 5: None}
742 5: None}
743
743
744 _, matches = complete(line_buffer="d['a")
744 _, matches = complete(line_buffer="d['a")
745 nt.assert_in("abc", matches)
745 nt.assert_in("abc", matches)
746 nt.assert_in("abd", matches)
746 nt.assert_in("abd", matches)
747 nt.assert_not_in("bad", matches)
747 nt.assert_not_in("bad", matches)
748 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
748 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
749
749
750 # check escaping and whitespace
750 # check escaping and whitespace
751 ip.user_ns['d'] = {'a\nb': None, 'a\'b': None, 'a"b': None, 'a word': None}
751 ip.user_ns['d'] = {'a\nb': None, 'a\'b': None, 'a"b': None, 'a word': None}
752 _, matches = complete(line_buffer="d['a")
752 _, matches = complete(line_buffer="d['a")
753 nt.assert_in("a\\nb", matches)
753 nt.assert_in("a\\nb", matches)
754 nt.assert_in("a\\'b", matches)
754 nt.assert_in("a\\'b", matches)
755 nt.assert_in("a\"b", matches)
755 nt.assert_in("a\"b", matches)
756 nt.assert_in("a word", matches)
756 nt.assert_in("a word", matches)
757 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
757 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
758
758
759 # - can complete on non-initial word of the string
759 # - can complete on non-initial word of the string
760 _, matches = complete(line_buffer="d['a w")
760 _, matches = complete(line_buffer="d['a w")
761 nt.assert_in("word", matches)
761 nt.assert_in("word", matches)
762
762
763 # - understands quote escaping
763 # - understands quote escaping
764 _, matches = complete(line_buffer="d['a\\'")
764 _, matches = complete(line_buffer="d['a\\'")
765 nt.assert_in("b", matches)
765 nt.assert_in("b", matches)
766
766
767 # - default quoting should work like repr
767 # - default quoting should work like repr
768 _, matches = complete(line_buffer="d[")
768 _, matches = complete(line_buffer="d[")
769 nt.assert_in("\"a'b\"", matches)
769 nt.assert_in("\"a'b\"", matches)
770
770
771 # - when opening quote with ", possible to match with unescaped apostrophe
771 # - when opening quote with ", possible to match with unescaped apostrophe
772 _, matches = complete(line_buffer="d[\"a'")
772 _, matches = complete(line_buffer="d[\"a'")
773 nt.assert_in("b", matches)
773 nt.assert_in("b", matches)
774
774
775 # need to not split at delims that readline won't split at
775 # need to not split at delims that readline won't split at
776 if '-' not in ip.Completer.splitter.delims:
776 if '-' not in ip.Completer.splitter.delims:
777 ip.user_ns['d'] = {'before-after': None}
777 ip.user_ns['d'] = {'before-after': None}
778 _, matches = complete(line_buffer="d['before-af")
778 _, matches = complete(line_buffer="d['before-af")
779 nt.assert_in('before-after', matches)
779 nt.assert_in('before-after', matches)
780
780
781 def test_dict_key_completion_contexts():
781 def test_dict_key_completion_contexts():
782 """Test expression contexts in which dict key completion occurs"""
782 """Test expression contexts in which dict key completion occurs"""
783 ip = get_ipython()
783 ip = get_ipython()
784 complete = ip.Completer.complete
784 complete = ip.Completer.complete
785 d = {'abc': None}
785 d = {'abc': None}
786 ip.user_ns['d'] = d
786 ip.user_ns['d'] = d
787
787
788 class C:
788 class C:
789 data = d
789 data = d
790 ip.user_ns['C'] = C
790 ip.user_ns['C'] = C
791 ip.user_ns['get'] = lambda: d
791 ip.user_ns['get'] = lambda: d
792
792
793 def assert_no_completion(**kwargs):
793 def assert_no_completion(**kwargs):
794 _, matches = complete(**kwargs)
794 _, matches = complete(**kwargs)
795 nt.assert_not_in('abc', matches)
795 nt.assert_not_in('abc', matches)
796 nt.assert_not_in('abc\'', matches)
796 nt.assert_not_in('abc\'', matches)
797 nt.assert_not_in('abc\']', matches)
797 nt.assert_not_in('abc\']', matches)
798 nt.assert_not_in('\'abc\'', matches)
798 nt.assert_not_in('\'abc\'', matches)
799 nt.assert_not_in('\'abc\']', matches)
799 nt.assert_not_in('\'abc\']', matches)
800
800
801 def assert_completion(**kwargs):
801 def assert_completion(**kwargs):
802 _, matches = complete(**kwargs)
802 _, matches = complete(**kwargs)
803 nt.assert_in("'abc'", matches)
803 nt.assert_in("'abc'", matches)
804 nt.assert_not_in("'abc']", matches)
804 nt.assert_not_in("'abc']", matches)
805
805
806 # no completion after string closed, even if reopened
806 # no completion after string closed, even if reopened
807 assert_no_completion(line_buffer="d['a'")
807 assert_no_completion(line_buffer="d['a'")
808 assert_no_completion(line_buffer="d[\"a\"")
808 assert_no_completion(line_buffer="d[\"a\"")
809 assert_no_completion(line_buffer="d['a' + ")
809 assert_no_completion(line_buffer="d['a' + ")
810 assert_no_completion(line_buffer="d['a' + '")
810 assert_no_completion(line_buffer="d['a' + '")
811
811
812 # completion in non-trivial expressions
812 # completion in non-trivial expressions
813 assert_completion(line_buffer="+ d[")
813 assert_completion(line_buffer="+ d[")
814 assert_completion(line_buffer="(d[")
814 assert_completion(line_buffer="(d[")
815 assert_completion(line_buffer="C.data[")
815 assert_completion(line_buffer="C.data[")
816
816
817 # greedy flag
817 # greedy flag
818 def assert_completion(**kwargs):
818 def assert_completion(**kwargs):
819 _, matches = complete(**kwargs)
819 _, matches = complete(**kwargs)
820 nt.assert_in("get()['abc']", matches)
820 nt.assert_in("get()['abc']", matches)
821
821
822 assert_no_completion(line_buffer="get()[")
822 assert_no_completion(line_buffer="get()[")
823 with greedy_completion():
823 with greedy_completion():
824 assert_completion(line_buffer="get()[")
824 assert_completion(line_buffer="get()[")
825 assert_completion(line_buffer="get()['")
825 assert_completion(line_buffer="get()['")
826 assert_completion(line_buffer="get()['a")
826 assert_completion(line_buffer="get()['a")
827 assert_completion(line_buffer="get()['ab")
827 assert_completion(line_buffer="get()['ab")
828 assert_completion(line_buffer="get()['abc")
828 assert_completion(line_buffer="get()['abc")
829
829
830
830
831
831
832 def test_dict_key_completion_bytes():
832 def test_dict_key_completion_bytes():
833 """Test handling of bytes in dict key completion"""
833 """Test handling of bytes in dict key completion"""
834 ip = get_ipython()
834 ip = get_ipython()
835 complete = ip.Completer.complete
835 complete = ip.Completer.complete
836
836
837 ip.user_ns['d'] = {'abc': None, b'abd': None}
837 ip.user_ns['d'] = {'abc': None, b'abd': None}
838
838
839 _, matches = complete(line_buffer="d[")
839 _, matches = complete(line_buffer="d[")
840 nt.assert_in("'abc'", matches)
840 nt.assert_in("'abc'", matches)
841 nt.assert_in("b'abd'", matches)
841 nt.assert_in("b'abd'", matches)
842
842
843 if False: # not currently implemented
843 if False: # not currently implemented
844 _, matches = complete(line_buffer="d[b")
844 _, matches = complete(line_buffer="d[b")
845 nt.assert_in("b'abd'", matches)
845 nt.assert_in("b'abd'", matches)
846 nt.assert_not_in("b'abc'", matches)
846 nt.assert_not_in("b'abc'", matches)
847
847
848 _, matches = complete(line_buffer="d[b'")
848 _, matches = complete(line_buffer="d[b'")
849 nt.assert_in("abd", matches)
849 nt.assert_in("abd", matches)
850 nt.assert_not_in("abc", matches)
850 nt.assert_not_in("abc", matches)
851
851
852 _, matches = complete(line_buffer="d[B'")
852 _, matches = complete(line_buffer="d[B'")
853 nt.assert_in("abd", matches)
853 nt.assert_in("abd", matches)
854 nt.assert_not_in("abc", matches)
854 nt.assert_not_in("abc", matches)
855
855
856 _, matches = complete(line_buffer="d['")
856 _, matches = complete(line_buffer="d['")
857 nt.assert_in("abc", matches)
857 nt.assert_in("abc", matches)
858 nt.assert_not_in("abd", matches)
858 nt.assert_not_in("abd", matches)
859
859
860
860
861 def test_dict_key_completion_unicode_py3():
861 def test_dict_key_completion_unicode_py3():
862 """Test handling of unicode in dict key completion"""
862 """Test handling of unicode in dict key completion"""
863 ip = get_ipython()
863 ip = get_ipython()
864 complete = ip.Completer.complete
864 complete = ip.Completer.complete
865
865
866 ip.user_ns['d'] = {u'a\u05d0': None}
866 ip.user_ns['d'] = {u'a\u05d0': None}
867
867
868 # query using escape
868 # query using escape
869 if sys.platform != 'win32':
869 if sys.platform != 'win32':
870 # Known failure on Windows
870 # Known failure on Windows
871 _, matches = complete(line_buffer="d['a\\u05d0")
871 _, matches = complete(line_buffer="d['a\\u05d0")
872 nt.assert_in("u05d0", matches) # tokenized after \\
872 nt.assert_in("u05d0", matches) # tokenized after \\
873
873
874 # query using character
874 # query using character
875 _, matches = complete(line_buffer="d['a\u05d0")
875 _, matches = complete(line_buffer="d['a\u05d0")
876 nt.assert_in(u"a\u05d0", matches)
876 nt.assert_in(u"a\u05d0", matches)
877
877
878 with greedy_completion():
878 with greedy_completion():
879 # query using escape
879 # query using escape
880 _, matches = complete(line_buffer="d['a\\u05d0")
880 _, matches = complete(line_buffer="d['a\\u05d0")
881 nt.assert_in("d['a\\u05d0']", matches) # tokenized after \\
881 nt.assert_in("d['a\\u05d0']", matches) # tokenized after \\
882
882
883 # query using character
883 # query using character
884 _, matches = complete(line_buffer="d['a\u05d0")
884 _, matches = complete(line_buffer="d['a\u05d0")
885 nt.assert_in(u"d['a\u05d0']", matches)
885 nt.assert_in(u"d['a\u05d0']", matches)
886
886
887
887
888
888
889 @dec.skip_without('numpy')
889 @dec.skip_without('numpy')
890 def test_struct_array_key_completion():
890 def test_struct_array_key_completion():
891 """Test dict key completion applies to numpy struct arrays"""
891 """Test dict key completion applies to numpy struct arrays"""
892 import numpy
892 import numpy
893 ip = get_ipython()
893 ip = get_ipython()
894 complete = ip.Completer.complete
894 complete = ip.Completer.complete
895 ip.user_ns['d'] = numpy.array([], dtype=[('hello', 'f'), ('world', 'f')])
895 ip.user_ns['d'] = numpy.array([], dtype=[('hello', 'f'), ('world', 'f')])
896 _, matches = complete(line_buffer="d['")
896 _, matches = complete(line_buffer="d['")
897 nt.assert_in("hello", matches)
897 nt.assert_in("hello", matches)
898 nt.assert_in("world", matches)
898 nt.assert_in("world", matches)
899 # complete on the numpy struct itself
899 # complete on the numpy struct itself
900 dt = numpy.dtype([('my_head', [('my_dt', '>u4'), ('my_df', '>u4')]),
900 dt = numpy.dtype([('my_head', [('my_dt', '>u4'), ('my_df', '>u4')]),
901 ('my_data', '>f4', 5)])
901 ('my_data', '>f4', 5)])
902 x = numpy.zeros(2, dtype=dt)
902 x = numpy.zeros(2, dtype=dt)
903 ip.user_ns['d'] = x[1]
903 ip.user_ns['d'] = x[1]
904 _, matches = complete(line_buffer="d['")
904 _, matches = complete(line_buffer="d['")
905 nt.assert_in("my_head", matches)
905 nt.assert_in("my_head", matches)
906 nt.assert_in("my_data", matches)
906 nt.assert_in("my_data", matches)
907 # complete on a nested level
907 # complete on a nested level
908 with greedy_completion():
908 with greedy_completion():
909 ip.user_ns['d'] = numpy.zeros(2, dtype=dt)
909 ip.user_ns['d'] = numpy.zeros(2, dtype=dt)
910 _, matches = complete(line_buffer="d[1]['my_head']['")
910 _, matches = complete(line_buffer="d[1]['my_head']['")
911 nt.assert_true(any(["my_dt" in m for m in matches]))
911 nt.assert_true(any(["my_dt" in m for m in matches]))
912 nt.assert_true(any(["my_df" in m for m in matches]))
912 nt.assert_true(any(["my_df" in m for m in matches]))
913
913
914
914
915 @dec.skip_without('pandas')
915 @dec.skip_without('pandas')
916 def test_dataframe_key_completion():
916 def test_dataframe_key_completion():
917 """Test dict key completion applies to pandas DataFrames"""
917 """Test dict key completion applies to pandas DataFrames"""
918 import pandas
918 import pandas
919 ip = get_ipython()
919 ip = get_ipython()
920 complete = ip.Completer.complete
920 complete = ip.Completer.complete
921 ip.user_ns['d'] = pandas.DataFrame({'hello': [1], 'world': [2]})
921 ip.user_ns['d'] = pandas.DataFrame({'hello': [1], 'world': [2]})
922 _, matches = complete(line_buffer="d['")
922 _, matches = complete(line_buffer="d['")
923 nt.assert_in("hello", matches)
923 nt.assert_in("hello", matches)
924 nt.assert_in("world", matches)
924 nt.assert_in("world", matches)
925
925
926
926
927 def test_dict_key_completion_invalids():
927 def test_dict_key_completion_invalids():
928 """Smoke test cases dict key completion can't handle"""
928 """Smoke test cases dict key completion can't handle"""
929 ip = get_ipython()
929 ip = get_ipython()
930 complete = ip.Completer.complete
930 complete = ip.Completer.complete
931
931
932 ip.user_ns['no_getitem'] = None
932 ip.user_ns['no_getitem'] = None
933 ip.user_ns['no_keys'] = []
933 ip.user_ns['no_keys'] = []
934 ip.user_ns['cant_call_keys'] = dict
934 ip.user_ns['cant_call_keys'] = dict
935 ip.user_ns['empty'] = {}
935 ip.user_ns['empty'] = {}
936 ip.user_ns['d'] = {'abc': 5}
936 ip.user_ns['d'] = {'abc': 5}
937
937
938 _, matches = complete(line_buffer="no_getitem['")
938 _, matches = complete(line_buffer="no_getitem['")
939 _, matches = complete(line_buffer="no_keys['")
939 _, matches = complete(line_buffer="no_keys['")
940 _, matches = complete(line_buffer="cant_call_keys['")
940 _, matches = complete(line_buffer="cant_call_keys['")
941 _, matches = complete(line_buffer="empty['")
941 _, matches = complete(line_buffer="empty['")
942 _, matches = complete(line_buffer="name_error['")
942 _, matches = complete(line_buffer="name_error['")
943 _, matches = complete(line_buffer="d['\\") # incomplete escape
943 _, matches = complete(line_buffer="d['\\") # incomplete escape
944
944
945 class KeyCompletable(object):
945 class KeyCompletable(object):
946 def __init__(self, things=()):
946 def __init__(self, things=()):
947 self.things = things
947 self.things = things
948
948
949 def _ipython_key_completions_(self):
949 def _ipython_key_completions_(self):
950 return list(self.things)
950 return list(self.things)
951
951
952 def test_object_key_completion():
952 def test_object_key_completion():
953 ip = get_ipython()
953 ip = get_ipython()
954 ip.user_ns['key_completable'] = KeyCompletable(['qwerty', 'qwick'])
954 ip.user_ns['key_completable'] = KeyCompletable(['qwerty', 'qwick'])
955
955
956 _, matches = ip.Completer.complete(line_buffer="key_completable['qw")
956 _, matches = ip.Completer.complete(line_buffer="key_completable['qw")
957 nt.assert_in('qwerty', matches)
957 nt.assert_in('qwerty', matches)
958 nt.assert_in('qwick', matches)
958 nt.assert_in('qwick', matches)
959
959
960
960
961 class NamedInstanceMetaclass(type):
961 class NamedInstanceMetaclass(type):
962 def __getitem__(cls, item):
962 def __getitem__(cls, item):
963 return cls.get_instance(item)
963 return cls.get_instance(item)
964
964
965 class NamedInstanceClass(object, metaclass=NamedInstanceMetaclass):
965 class NamedInstanceClass(object, metaclass=NamedInstanceMetaclass):
966 def __init__(self, name):
966 def __init__(self, name):
967 if not hasattr(self.__class__, 'instances'):
967 if not hasattr(self.__class__, 'instances'):
968 self.__class__.instances = {}
968 self.__class__.instances = {}
969 self.__class__.instances[name] = self
969 self.__class__.instances[name] = self
970
970
971 @classmethod
971 @classmethod
972 def _ipython_key_completions_(cls):
972 def _ipython_key_completions_(cls):
973 return cls.instances.keys()
973 return cls.instances.keys()
974
974
975 @classmethod
975 @classmethod
976 def get_instance(cls, name):
976 def get_instance(cls, name):
977 return cls.instances[name]
977 return cls.instances[name]
978
978
979 def test_class_key_completion():
979 def test_class_key_completion():
980 ip = get_ipython()
980 ip = get_ipython()
981 NamedInstanceClass('qwerty')
981 NamedInstanceClass('qwerty')
982 NamedInstanceClass('qwick')
982 NamedInstanceClass('qwick')
983 ip.user_ns['named_instance_class'] = NamedInstanceClass
983 ip.user_ns['named_instance_class'] = NamedInstanceClass
984
984
985 _, matches = ip.Completer.complete(line_buffer="named_instance_class['qw")
985 _, matches = ip.Completer.complete(line_buffer="named_instance_class['qw")
986 nt.assert_in('qwerty', matches)
986 nt.assert_in('qwerty', matches)
987 nt.assert_in('qwick', matches)
987 nt.assert_in('qwick', matches)
988
988
989 def test_tryimport():
989 def test_tryimport():
990 """
990 """
991 Test that try-import don't crash on trailing dot, and import modules before
991 Test that try-import don't crash on trailing dot, and import modules before
992 """
992 """
993 from IPython.core.completerlib import try_import
993 from IPython.core.completerlib import try_import
994 assert(try_import("IPython."))
994 assert(try_import("IPython."))
995
995
996
996
997 def test_aimport_module_completer():
997 def test_aimport_module_completer():
998 ip = get_ipython()
998 ip = get_ipython()
999 _, matches = ip.complete('i', '%aimport i')
999 _, matches = ip.complete('i', '%aimport i')
1000 nt.assert_in('io', matches)
1000 nt.assert_in('io', matches)
1001 nt.assert_not_in('int', matches)
1001 nt.assert_not_in('int', matches)
1002
1002
1003 def test_nested_import_module_completer():
1003 def test_nested_import_module_completer():
1004 ip = get_ipython()
1004 ip = get_ipython()
1005 _, matches = ip.complete(None, 'import IPython.co', 17)
1005 _, matches = ip.complete(None, 'import IPython.co', 17)
1006 nt.assert_in('IPython.core', matches)
1006 nt.assert_in('IPython.core', matches)
1007 nt.assert_not_in('import IPython.core', matches)
1007 nt.assert_not_in('import IPython.core', matches)
1008 nt.assert_not_in('IPython.display', matches)
1008 nt.assert_not_in('IPython.display', matches)
1009
1009
1010 def test_import_module_completer():
1010 def test_import_module_completer():
1011 ip = get_ipython()
1011 ip = get_ipython()
1012 _, matches = ip.complete('i', 'import i')
1012 _, matches = ip.complete('i', 'import i')
1013 nt.assert_in('io', matches)
1013 nt.assert_in('io', matches)
1014 nt.assert_not_in('int', matches)
1014 nt.assert_not_in('int', matches)
1015
1015
1016 def test_from_module_completer():
1016 def test_from_module_completer():
1017 ip = get_ipython()
1017 ip = get_ipython()
1018 _, matches = ip.complete('B', 'from io import B', 16)
1018 _, matches = ip.complete('B', 'from io import B', 16)
1019 nt.assert_in('BytesIO', matches)
1019 nt.assert_in('BytesIO', matches)
1020 nt.assert_not_in('BaseException', matches)
1020 nt.assert_not_in('BaseException', matches)
1021
1021
1022 def test_snake_case_completion():
1022 def test_snake_case_completion():
1023 ip = get_ipython()
1023 ip = get_ipython()
1024 ip.Completer.use_jedi = False
1024 ip.Completer.use_jedi = False
1025 ip.user_ns['some_three'] = 3
1025 ip.user_ns['some_three'] = 3
1026 ip.user_ns['some_four'] = 4
1026 ip.user_ns['some_four'] = 4
1027 _, matches = ip.complete("s_", "print(s_f")
1027 _, matches = ip.complete("s_", "print(s_f")
1028 nt.assert_in('some_three', matches)
1028 nt.assert_in('some_three', matches)
1029 nt.assert_in('some_four', matches)
1029 nt.assert_in('some_four', matches)
1030
1031 def test_mix_terms():
1032 ip = get_ipython()
1033 from textwrap import dedent
1034 ip.Completer.use_jedi = False
1035 ip.ex(dedent("""
1036 class Test:
1037 def meth(self, meth_arg1):
1038 print("meth")
1039
1040 def meth_1(self, meth1_arg1, meth1_arg2):
1041 print("meth1")
1042
1043 def meth_2(self, meth2_arg1, meth2_arg2):
1044 print("meth2")
1045 test = Test()
1046 """))
1047 _, matches = ip.complete(None, "test.meth(")
1048 nt.assert_in('meth_arg1=', matches)
1049 nt.assert_not_in('meth2_arg1=', matches)
General Comments 0
You need to be logged in to leave comments. Login now