VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Refactor tests for function_definition

Browse Source
This commit is contained in:
Takafumi Arakaki
2013-03-10 01:52:15 +01:00
parent 0e99c33c67
commit 3542b7f80a
1 changed files with 33 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
test/regression.py
View File

@@ -192,10 +192,15 @@ class TestRegression(TestBase):
assert self.complete("from datetime import")[0].word == 'import' assert self.complete("from datetime import")[0].word == 'import'
assert self.complete("from datetime import ") assert self.complete("from datetime import ")
def assert_call_def(self, call_def, name, index):
self.assertEqual(
{'call_name': getattr(call_def, 'call_name', None),
'index': getattr(call_def, 'index', None)},
{'call_name': name, 'index': index},
)
def test_function_definition(self): def test_function_definition(self):
def check(call_def, name, index): check = self.assert_call_def
return call_def and call_def.call_name == name \
and call_def.index == index
# simple # simple
s = "abs(a, str(" s = "abs(a, str("
@@ -208,13 +213,13 @@ class TestRegression(TestBase):
s7 = "str().upper().center(" s7 = "str().upper().center("
s8 = "str(int[zip(" s8 = "str(int[zip("
assert check(self.function_definition(s, (1, 4)), 'abs', 0) check(self.function_definition(s, (1, 4)), 'abs', 0)
assert check(self.function_definition(s, (1, 6)), 'abs', 1) check(self.function_definition(s, (1, 6)), 'abs', 1)
assert check(self.function_definition(s, (1, 7)), 'abs', 1) check(self.function_definition(s, (1, 7)), 'abs', 1)
assert check(self.function_definition(s, (1, 8)), 'abs', 1) check(self.function_definition(s, (1, 8)), 'abs', 1)
assert check(self.function_definition(s, (1, 11)), 'str', 0) check(self.function_definition(s, (1, 11)), 'str', 0)
assert check(self.function_definition(s2, (1, 4)), 'abs', 0) check(self.function_definition(s2, (1, 4)), 'abs', 0)
assert self.function_definition(s2, (1, 5)) is None assert self.function_definition(s2, (1, 5)) is None
assert self.function_definition(s2) is None assert self.function_definition(s2) is None
@@ -222,43 +227,41 @@ class TestRegression(TestBase):
assert self.function_definition(s3) is None assert self.function_definition(s3) is None
assert self.function_definition(s4, (1, 3)) is None assert self.function_definition(s4, (1, 3)) is None
assert check(self.function_definition(s4, (1, 4)), 'abs', 0) check(self.function_definition(s4, (1, 4)), 'abs', 0)
assert check(self.function_definition(s4, (1, 8)), 'zip', 0) check(self.function_definition(s4, (1, 8)), 'zip', 0)
assert check(self.function_definition(s4, (1, 9)), 'abs', 0) check(self.function_definition(s4, (1, 9)), 'abs', 0)
#assert check(self.function_definition(s4, (1, 10)), 'abs', 1) #check(self.function_definition(s4, (1, 10)), 'abs', 1)
assert check(self.function_definition(s5, (1, 4)), 'abs', 0) check(self.function_definition(s5, (1, 4)), 'abs', 0)
assert check(self.function_definition(s5, (1, 6)), 'abs', 1) check(self.function_definition(s5, (1, 6)), 'abs', 1)
assert check(self.function_definition(s6), 'center', 0) check(self.function_definition(s6), 'center', 0)
assert check(self.function_definition(s6, (1, 4)), 'str', 0) check(self.function_definition(s6, (1, 4)), 'str', 0)
assert check(self.function_definition(s7), 'center', 0) check(self.function_definition(s7), 'center', 0)
assert check(self.function_definition(s8), 'zip', 0) check(self.function_definition(s8), 'zip', 0)
assert check(self.function_definition(s8, (1, 8)), 'str', 0) check(self.function_definition(s8, (1, 8)), 'str', 0)
s = "import time; abc = time; abc.sleep(" s = "import time; abc = time; abc.sleep("
assert check(self.function_definition(s), 'sleep', 0) check(self.function_definition(s), 'sleep', 0)
# jedi-vim #9 # jedi-vim #9
s = "with open(" s = "with open("
assert check(self.function_definition(s), 'open', 0) check(self.function_definition(s), 'open', 0)
# jedi-vim #11 # jedi-vim #11
s1 = "for sorted(" s1 = "for sorted("
assert check(self.function_definition(s1), 'sorted', 0) check(self.function_definition(s1), 'sorted', 0)
s2 = "for s in sorted(" s2 = "for s in sorted("
assert check(self.function_definition(s2), 'sorted', 0) check(self.function_definition(s2), 'sorted', 0)
# jedi #57 # jedi #57
s = "def func(alpha, beta): pass\n" \ s = "def func(alpha, beta): pass\n" \
"func(alpha='101'," "func(alpha='101',"
assert check(self.function_definition(s, (2, 13)), 'func', 0) check(self.function_definition(s, (2, 13)), 'func', 0)
def test_function_definition_complex(self): def test_function_definition_complex(self):
def check(call_def, name, index): check = self.assert_call_def
return call_def and call_def.call_name == name \
and call_def.index == index
s = """ s = """
def abc(a,b): def abc(a,b):
@@ -270,14 +273,14 @@ class TestRegression(TestBase):
if 1: if 1:
pass pass
""" """
assert check(self.function_definition(s, (6, 24)), 'abc', 0) check(self.function_definition(s, (6, 24)), 'abc', 0)
s = """ s = """
import re import re
def huhu(it): def huhu(it):
re.compile( re.compile(
return it * 2 return it * 2
""" """
assert check(self.function_definition(s, (4, 31)), 'compile', 0) check(self.function_definition(s, (4, 31)), 'compile', 0)
# jedi-vim #70 # jedi-vim #70
s = """def foo(""" s = """def foo("""
assert self.function_definition(s) is None assert self.function_definition(s) is None