test_parsing -> test_parser

test/test_parser/test_parser.py

@@ -0,0 +1,110 @@
+from jedi._compatibility import u
+from jedi.parser import Parser
+from jedi.parser.user_context import UserContextParser
+from jedi.parser import representation as pr
+from textwrap import dedent
+
+
+def test_user_statement_on_import():
+    """github #285"""
+    s = u("from datetime import (\n"
+          "    time)")
+
+    for pos in [(2, 1), (2, 4)]:
+        p = UserContextParser(s, None, pos, None).user_stmt()
+        assert isinstance(p, pr.Import)
+        assert p.defunct is False
+        assert [str(n) for n in p.get_defined_names()] == ['time']
+
+
+class TestCallAndName():
+    def get_call(self, source):
+        stmt = Parser(u(source), no_docstr=True).module.statements[0]
+        return stmt.expression_list()[0]
+
+    def test_name_and_call_positions(self):
+        call = self.get_call('name\nsomething_else')
+        assert str(call.name) == 'name'
+        assert call.name.start_pos == call.start_pos == (1, 0)
+        assert call.name.end_pos == call.end_pos == (1, 4)
+
+        call = self.get_call('1.0\n')
+        assert call.value == 1.0
+        assert call.start_pos == (1, 0)
+        assert call.end_pos == (1, 3)
+
+    def test_call_type(self):
+        call = self.get_call('hello')
+        assert isinstance(call, pr.Call)
+        assert type(call.name) == pr.Name
+
+    def test_literal_type(self):
+        literal = self.get_call('1.0')
+        assert isinstance(literal, pr.Literal)
+        assert type(literal.value) == float
+
+        literal = self.get_call('1')
+        assert isinstance(literal, pr.Literal)
+        assert type(literal.value) == int
+
+        literal = self.get_call('"hello"')
+        assert isinstance(literal, pr.Literal)
+        assert literal.value == 'hello'
+
+
+class TestSubscopes():
+    def get_sub(self, source):
+        return Parser(u(source)).module.subscopes[0]
+
+    def test_subscope_names(self):
+        name = self.get_sub('class Foo: pass').name
+        assert name.start_pos == (1, len('class '))
+        assert name.end_pos == (1, len('class Foo'))
+        assert str(name) == 'Foo'
+
+        name = self.get_sub('def foo(): pass').name
+        assert name.start_pos == (1, len('def '))
+        assert name.end_pos == (1, len('def foo'))
+        assert str(name) == 'foo'
+
+
+class TestImports():
+    def get_import(self, source):
+        return Parser(source).module.imports[0]
+
+    def test_import_names(self):
+        imp = self.get_import(u('import math\n'))
+        names = imp.get_defined_names()
+        assert len(names) == 1
+        assert str(names[0]) == 'math'
+        assert names[0].start_pos == (1, len('import '))
+        assert names[0].end_pos == (1, len('import math'))
+
+        assert imp.start_pos == (1, 0)
+        assert imp.end_pos == (1, len('import math'))
+
+
+def test_module():
+    module = Parser(u('asdf'), 'example.py', no_docstr=True).module
+    name = module.name
+    assert str(name) == 'example'
+    assert name.start_pos == (0, 0)
+    assert name.end_pos == (0, 0)
+
+    module = Parser(u('asdf'), no_docstr=True).module
+    name = module.name
+    assert str(name) == ''
+    assert name.start_pos == (0, 0)
+    assert name.end_pos == (0, 0)
+
+
+def test_end_pos():
+    s = u(dedent('''
+                 x = ['a', 'b', 'c']
+                 def func():
+                     y = None
+                 '''))
+    parser = Parser(s)
+    scope = parser.module.subscopes[0]
+    assert scope.start_pos == (3, 0)
+    assert scope.end_pos == (5, 0)