summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing/warnings.py
diff options
context:
space:
mode:
authorDiana Clarke <diana.joan.clarke@gmail.com>2012-11-19 20:19:39 -0500
committerDiana Clarke <diana.joan.clarke@gmail.com>2012-11-19 20:19:39 -0500
commitffb2107703ac4f9463625ddf46adbf52c4029a98 (patch)
tree64d0306b9f556b4a10a64e74b2f966301017127a /lib/sqlalchemy/testing/warnings.py
parent77d5317947e1c54a25ab6c8eaf7f23bc546bd2d9 (diff)
downloadsqlalchemy-ffb2107703ac4f9463625ddf46adbf52c4029a98.tar.gz
just a pep8 pass of lib/sqlalchemy/testing/
Diffstat (limited to 'lib/sqlalchemy/testing/warnings.py')
-rw-r--r--lib/sqlalchemy/testing/warnings.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/lib/sqlalchemy/testing/warnings.py b/lib/sqlalchemy/testing/warnings.py
index 7afcc63c5..41f3dbfed 100644
--- a/lib/sqlalchemy/testing/warnings.py
+++ b/lib/sqlalchemy/testing/warnings.py
@@ -4,6 +4,7 @@ import warnings
from .. import exc as sa_exc
from .. import util
+
def testing_warn(msg, stacklevel=3):
"""Replaces sqlalchemy.util.warn during tests."""
@@ -14,6 +15,7 @@ def testing_warn(msg, stacklevel=3):
else:
warnings.warn_explicit(msg, filename, lineno)
+
def resetwarnings():
"""Reset warning behavior to testing defaults."""
@@ -24,6 +26,7 @@ def resetwarnings():
warnings.filterwarnings('error', category=sa_exc.SADeprecationWarning)
warnings.filterwarnings('error', category=sa_exc.SAWarning)
+
def assert_warnings(fn, warnings):
"""Assert that each of the given warnings are emitted by fn."""
@@ -31,6 +34,7 @@ def assert_warnings(fn, warnings):
canary = []
orig_warn = util.warn
+
def capture_warnings(*args, **kw):
orig_warn(*args, **kw)
popwarn = warnings.pop(0)