From c6c8d7e54025e0843e4275d21c0fee9e1bdef5e0 Mon Sep 17 00:00:00 2001 From: Dmitry Shachnev Date: Tue, 11 Sep 2018 19:48:44 +0300 Subject: [PATCH] unittest: make assertIn/assertNotIn accept iterables (#2447) Fixes #2381. --- stdlib/3/unittest/__init__.pyi | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/stdlib/3/unittest/__init__.pyi b/stdlib/3/unittest/__init__.pyi index 3337d00ba..0b9ec1378 100644 --- a/stdlib/3/unittest/__init__.pyi +++ b/stdlib/3/unittest/__init__.pyi @@ -49,9 +49,11 @@ class TestCase: msg: Any = ...) -> None: ... def assertIsNone(self, expr: Any, msg: Any = ...) -> None: ... def assertIsNotNone(self, expr: Any, msg: Any = ...) -> None: ... - def assertIn(self, member: Any, container: Container[Any], + def assertIn(self, member: Any, + container: Union[Iterable[Any], Container[Any]], msg: Any = ...) -> None: ... - def assertNotIn(self, member: Any, container: Container[Any], + def assertNotIn(self, member: Any, + container: Union[Iterable[Any], Container[Any]], msg: Any = ...) -> None: ... def assertIsInstance(self, obj: Any, cls: Union[type, Tuple[type, ...]],