diff --git a/pyproject.toml b/pyproject.toml index 518a86e8..dba836ff 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -122,9 +122,11 @@ warn_unreachable = true disallow_any_unimported = true extra_checks = true enable_error_code = [ + "deprecated", "exhaustive-match", "ignore-without-code", "mutable-override", + "possibly-undefined", "redundant-expr", "truthy-bool", "truthy-iterable", diff --git a/tests/test_asyncio.py b/tests/test_asyncio.py index 96560b07..c9d58027 100644 --- a/tests/test_asyncio.py +++ b/tests/test_asyncio.py @@ -85,7 +85,6 @@ async def test_retry(self) -> None: @asynctest async def test_iscoroutinefunction(self) -> None: - assert asyncio.iscoroutinefunction(_retryable_coroutine) assert inspect.iscoroutinefunction(_retryable_coroutine) @asynctest diff --git a/tests/test_tenacity.py b/tests/test_tenacity.py index 8cfd34e5..d5dd8f17 100644 --- a/tests/test_tenacity.py +++ b/tests/test_tenacity.py @@ -692,10 +692,9 @@ def waitfunc(retry_state: RetryCallState) -> float: def returnval() -> int: return 123 - try: + with self.assertRaises(ExtractCallState) as caught: retrying(returnval) - except ExtractCallState as err: - retry_state = err.args[0] + retry_state = caught.exception.args[0] self.assertIs(retry_state.fn, returnval) self.assertEqual(retry_state.args, ()) self.assertEqual(retry_state.kwargs, {}) @@ -706,10 +705,9 @@ def returnval() -> int: def dying() -> None: raise Exception("Broken") - try: + with self.assertRaises(ExtractCallState) as caught: retrying(dying) - except ExtractCallState as err: - retry_state = err.args[0] + retry_state = caught.exception.args[0] self.assertIs(retry_state.fn, dying) self.assertEqual(retry_state.args, ()) self.assertEqual(retry_state.kwargs, {})