move some tests into specific directories

2014-01-29 20:43:42 +01:00
parent 36de8c427f
commit 78114b12e9
16 changed files with 11 additions and 9 deletions
--- a/test/api/test_call_signatures.py
+++ b/test/api/test_call_signatures.py
@@ -0,0 +1,137 @@
+import textwrap
+
+from ..helpers import TestCase
+from jedi import Script
+
+
+class TestCallSignatures(TestCase):
+    def _run(self, source, expected_name, expected_index=0, line=None, column=None):
+        signatures = Script(source, line, column).call_signatures()
+
+        assert len(signatures) <= 1
+
+        if not signatures:
+            assert expected_name is None
+        else:
+            assert signatures[0].call_name == expected_name
+            assert signatures[0].index == expected_index
+
+    def _run_simple(self, source, name, index=0, column=None, line=1):
+        self._run(source, name, index, line, column)
+
+    def test_simple(self):
+        run = self._run_simple
+
+        # simple
+        s1 = "abs(a, str("
+        run(s1, 'abs', 0, 4)
+        run(s1, 'abs', 1, 6)
+        run(s1, 'abs', 1, 7)
+        run(s1, 'abs', 1, 8)
+        run(s1, 'str', 0, 11)
+
+        s2 = "abs(), "
+        run(s2, 'abs', 0, 4)
+        run(s2, None, column=5)
+        run(s2, None)
+
+        s3 = "abs()."
+        run(s3, None, column=5)
+        run(s3, None)
+
+        # more complicated
+        s4 = 'abs(zip(), , set,'
+        run(s4, None, column=3)
+        run(s4, 'abs', 0, 4)
+        run(s4, 'zip', 0, 8)
+        run(s4, 'abs', 0, 9)
+        #run(s4, 'abs', 1, 10)
+
+        s5 = "abs(1,\nif 2:\n def a():"
+        run(s5, 'abs', 0, 4)
+        run(s5, 'abs', 1, 6)
+
+        s6 = "str().center("
+        run(s6, 'center', 0)
+        run(s6, 'str', 0, 4)
+
+        s7 = "str().upper().center("
+        s8 = "str(int[zip("
+        run(s7, 'center', 0)
+        run(s8, 'zip', 0)
+        run(s8, 'str', 0, 8)
+
+        run("import time; abc = time; abc.sleep(", 'sleep', 0)
+
+        # jedi #57
+        s = "def func(alpha, beta): pass\n" \
+            "func(alpha='101',"
+        run(s, 'func', 0, column=13, line=2)
+
+    def test_flows(self):
+        # jedi-vim #9
+        self._run_simple("with open(", 'open', 0)
+
+        # jedi-vim #11
+        self._run_simple("for sorted(", 'sorted', 0)
+        self._run_simple("for s in sorted(", 'sorted', 0)
+
+    def test_complex(self):
+        s = """
+                def abc(a,b):
+                    pass
+
+                def a(self):
+                    abc(
+
+                if 1:
+                    pass
+            """
+        self._run(s, 'abc', 0, line=6, column=24)
+        s = """
+                import re
+                def huhu(it):
+                    re.compile(
+                    return it * 2
+            """
+        self._run(s, 'compile', 0, line=4, column=31)
+
+        # jedi-vim #70
+        s = """def foo("""
+        assert Script(s).call_signatures() == []
+
+        # jedi-vim #116
+        s = """import functools; test = getattr(functools, 'partial'); test("""
+        self._run(s, 'partial', 0)
+
+    def test_call_signature_on_module(self):
+        """github issue #240"""
+        s = 'import datetime; datetime('
+        # just don't throw an exception (if numpy doesn't exist, just ignore it)
+        assert Script(s).call_signatures() == []
+
+    def test_call_signatures_empty_parentheses_pre_space(self):
+        s = textwrap.dedent("""\
+        def f(a, b):
+            pass
+        f( )""")
+        self._run(s, 'f', 0, line=3, column=3)
+
+    def test_multiple_signatures(self):
+        s = textwrap.dedent("""\
+        if x:
+            def f(a, b):
+                pass
+        else:
+            def f(a, b):
+                pass
+        f(""")
+        assert len(Script(s).call_signatures()) == 2
+
+    def test_call_signatures_whitespace(self):
+        s = textwrap.dedent("""\
+        abs( 
+        def x():
+            pass
+        """)
+        self._run(s, 'abs', 0, line=1, column=5)