From fadc7d18e695e8217e26e5191ff11e36dac44fde Mon Sep 17 00:00:00 2001 From: mart-r Date: Tue, 13 Aug 2024 14:14:39 +0100 Subject: [PATCH] CU-8695d4www: Fix allowing deprecation during test time --- tests/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/__init__.py b/tests/__init__.py index b34de257..cd9d53b9 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -14,11 +14,11 @@ def __init__(self, func: Callable, msg: str, def deprecation_exception_raiser(message: str, depr_version: Tuple[int, int, int], - removal_version: Tuple[int, int, int]): + removal_version: Tuple[int, int, int], + allow_usage: bool = False): def decorator(func: Callable) -> Callable: def wrapper(*args, **kwargs): - if ('allow_usage' in kwargs and kwargs['allow_usage'] or - len(args) >= 4 and args[3]): + if allow_usage: return func(*args, **kwargs) raise DeprecatedMethodCallException(func, message, depr_version, removal_version) return wrapper