Work around limitation in warnings.warn

This commit is contained in:
Steven Silvester
2016-01-30 10:37:30 -06:00
parent e24953cf74
commit 4f54b9ed5a
+3 -2
View File
@@ -7,10 +7,11 @@ import re
__all__ = ['all_warnings', 'expected_warnings', 'warn']
def warn(message, category=None, stacklevel=2):
def warn(message, **kwargs):
"""A version of `warnings.warn` with a default stacklevel of 2.
"""
warnings.warn(message, category=category, stacklevel=stacklevel)
kwargs.setdefault('stacklevel', 2)
warnings.warn(message, **kwargs)
@contextmanager