From c96b1ae1bfa0fb1703b695b039d9ed697c05d233 Mon Sep 17 00:00:00 2001 From: Jaromir Latal Date: Thu, 30 Jul 2020 21:02:08 +0100 Subject: [PATCH] [stdlib][unittest] Fix mock.call(...) types (#4374) Co-authored-by: Jaromir Latal --- stdlib/3/unittest/mock.pyi | 4 ++-- third_party/2and3/mock.pyi | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/stdlib/3/unittest/mock.pyi b/stdlib/3/unittest/mock.pyi index e58f1bc6a..1de1d07a9 100644 --- a/stdlib/3/unittest/mock.pyi +++ b/stdlib/3/unittest/mock.pyi @@ -51,13 +51,13 @@ class _Call(Tuple[Any, ...]): ) -> None: ... def __eq__(self, other: Any) -> bool: ... __ne__: Any - def __call__(self, *args: Any, **kwargs: Any) -> Any: ... + def __call__(self, *args: Any, **kwargs: Any) -> _Call: ... def __getattr__(self, attr: Any) -> Any: ... def count(self, *args: Any, **kwargs: Any) -> Any: ... def index(self, *args: Any, **kwargs: Any) -> Any: ... def call_list(self) -> Any: ... -call: Any +call: _Call class _CallList(List[_Call]): def __contains__(self, value: Any) -> bool: ... diff --git a/third_party/2and3/mock.pyi b/third_party/2and3/mock.pyi index e58f1bc6a..1de1d07a9 100644 --- a/third_party/2and3/mock.pyi +++ b/third_party/2and3/mock.pyi @@ -51,13 +51,13 @@ class _Call(Tuple[Any, ...]): ) -> None: ... def __eq__(self, other: Any) -> bool: ... __ne__: Any - def __call__(self, *args: Any, **kwargs: Any) -> Any: ... + def __call__(self, *args: Any, **kwargs: Any) -> _Call: ... def __getattr__(self, attr: Any) -> Any: ... def count(self, *args: Any, **kwargs: Any) -> Any: ... def index(self, *args: Any, **kwargs: Any) -> Any: ... def call_list(self) -> Any: ... -call: Any +call: _Call class _CallList(List[_Call]): def __contains__(self, value: Any) -> bool: ...