VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-10 07:41:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add test for annotations to test_parser_tree

Browse Source
This commit is contained in:
Claude
2015-12-13 23:45:37 +01:00
parent 7e8112d607
commit c61f39cb2b
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/test_parser/test_parser_tree.py
View File

@@ -12,10 +12,11 @@ from jedi.parser import tree as pt
class TestsFunctionAndLambdaParsing(object): class TestsFunctionAndLambdaParsing(object):
FIXTURES = [ FIXTURES = [
('def my_function(x, y, z):\n return x + y * z\n', { ('def my_function(x, y, z) -> str:\n return x + y * z\n', {
'name': 'my_function', 'name': 'my_function',
'call_sig': 'my_function(x, y, z)', 'call_sig': 'my_function(x, y, z)',
'params': ['x', 'y', 'z'], 'params': ['x', 'y', 'z'],
'annotation': "str",
}), }),
('lambda x, y, z: x + y * z\n', { ('lambda x, y, z: x + y * z\n', {
'name': '<lambda>', 'name': '<lambda>',
@@ -55,7 +56,11 @@ class TestsFunctionAndLambdaParsing(object):
assert not node.yields assert not node.yields
def test_annotation(self, node, expected): def test_annotation(self, node, expected):
assert node.annotation() is expected.get('annotation', None) expected_annotation = expected.get('annotation', None)
if expected_annotation is None:
assert node.annotation() is None
else:
assert node.annotation().value == expected_annotation
def test_get_call_signature(self, node, expected): def test_get_call_signature(self, node, expected):
assert node.get_call_signature() == expected['call_sig'] assert node.get_call_signature() == expected['call_sig']