1
0
forked from VimPlug/jedi

Python 2 compatibility

This commit is contained in:
Dave Halter
2019-08-06 00:36:50 +02:00
parent 4b8505b78d
commit 98a550e352
2 changed files with 7 additions and 4 deletions

View File

@@ -26,8 +26,11 @@ class _ModuleAttributeName(AbstractNameDefinition):
def infer(self):
if self._string_value is not None:
s = self._string_value
if self.parent_context.evaluator.environment.version_info.major == 2:
s = bytes(s)
return ContextSet([
create_simple_object(self.parent_context.evaluator, self._string_value)
create_simple_object(self.parent_context.evaluator, s)
])
return compiled.get_string_context_set(self.parent_context.evaluator)

View File

@@ -166,8 +166,8 @@ def test_keyword_completion(Script, code, has_keywords):
(None, '"test', None, [s]),
(None, '"test', 4, ['t' + s]),
('example.py', '"test%scomp' % s, None, ['letion' + s]),
('example.py', 'r"comp"', None, ...),
('example.py', 'r"tes"', None, ...),
('example.py', 'r"comp"', None, "A LOT"),
('example.py', 'r"tes"', None, "A LOT"),
('example.py', 'r"tes"', 5, ['t' + s]),
('example.py', 'r" tes"', 6, []),
('test%sexample.py' % s, 'r"tes"', 5, ['t' + s]),
@@ -203,7 +203,7 @@ def test_file_path_completions(Script, file, code, column, expected):
if isinstance(column, tuple):
line, column = column
comps = Script(code, path=file, line=line, column=column).completions()
if expected == ...:
if expected == "A LOT":
assert len(comps) > 100 # This is basically global completions.
else:
assert [c.complete for c in comps] == expected