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

Fix typing.NewType signature

Browse Source
This commit is contained in:
Dave Halter
2021-01-01 04:22:52 +01:00
parent 30e702de11
commit 85f45771f1
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
jedi/inference/gradual/typing.py
View File

@@ -76,8 +76,8 @@ class TypingModuleName(NameWrapper):
yield OverloadFunction.create_cached( yield OverloadFunction.create_cached(
inference_state, self.parent_context, self.tree_name) inference_state, self.parent_context, self.tree_name)
elif name == 'NewType': elif name == 'NewType':
yield NewTypeFunction.create_cached( v, = self._wrapped_name.infer()
inference_state, self.parent_context, self.tree_name) yield NewTypeFunction.create_cached(inference_state, v)
elif name == 'cast': elif name == 'cast':
cast_fn, = self._wrapped_name.infer() cast_fn, = self._wrapped_name.infer()
yield CastFunction.create_cached(inference_state, cast_fn) yield CastFunction.create_cached(inference_state, cast_fn)
@@ -86,11 +86,10 @@ class TypingModuleName(NameWrapper):
# added soon. # added soon.
yield TypedDictClass.create_cached( yield TypedDictClass.create_cached(
inference_state, self.parent_context, self.tree_name) inference_state, self.parent_context, self.tree_name)
elif name in ('no_type_check', 'no_type_check_decorator'):
# This is not necessary, as long as we are not doing type checking.
yield from self._wrapped_name.infer()
else: else:
# Everything else shouldn't be relevant for type checking. # Not necessary, as long as we are not doing type checking:
# no_type_check & no_type_check_decorator
# Everything else shouldn't be relevant...
yield from self._wrapped_name.infer() yield from self._wrapped_name.infer()
@@ -398,7 +397,7 @@ class OverloadFunction(BaseTypingValue):
return func_value_set return func_value_set
class NewTypeFunction(BaseTypingValue): class NewTypeFunction(ValueWrapper):
def py__call__(self, arguments): def py__call__(self, arguments):
ordered_args = arguments.unpack() ordered_args = arguments.unpack()
next(ordered_args, (None, None)) next(ordered_args, (None, None))

4
test/test_inference/test_signature.py
View File

@@ -117,6 +117,10 @@ class X:
('from typing import List\nList[int](', None), ('from typing import List\nList[int](', None),
('from typing import Tuple\nTuple(', None), ('from typing import Tuple\nTuple(', None),
('from typing import Tuple\nTuple[int](', None), ('from typing import Tuple\nTuple[int](', None),
('from typing import Optional\nOptional(', None),
('from typing import Optional\nOptional[int](', None),
('from typing import Any\nAny(', None),
('from typing import NewType\nNewType(', 'NewType(name: str, tp: Type[_T]) -> Type[_T]'),
] ]
) )
def test_tree_signature(Script, environment, code, expected): def test_tree_signature(Script, environment, code, expected):