summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing/assertions.py
diff options
context:
space:
mode:
authorDiana Clarke <diana.joan.clarke@gmail.com>2012-11-19 20:34:27 -0500
committerDiana Clarke <diana.joan.clarke@gmail.com>2012-11-19 20:34:27 -0500
commitcc42604ba453d9affa7e3feda985dcdb25cd8f57 (patch)
treed173f495624eaafac0c8c4dcb63d7b2f0f27fd3c /lib/sqlalchemy/testing/assertions.py
parentffb2107703ac4f9463625ddf46adbf52c4029a98 (diff)
downloadsqlalchemy-cc42604ba453d9affa7e3feda985dcdb25cd8f57.tar.gz
just a pep8 pass of lib/sqlalchemy/testing/
Diffstat (limited to 'lib/sqlalchemy/testing/assertions.py')
-rw-r--r--lib/sqlalchemy/testing/assertions.py27
1 files changed, 21 insertions, 6 deletions
diff --git a/lib/sqlalchemy/testing/assertions.py b/lib/sqlalchemy/testing/assertions.py
index e74d13a97..ebd10b130 100644
--- a/lib/sqlalchemy/testing/assertions.py
+++ b/lib/sqlalchemy/testing/assertions.py
@@ -16,6 +16,7 @@ import itertools
from .util import fail
import contextlib
+
def emits_warning(*messages):
"""Mark a test as emitting a warning.
@@ -50,6 +51,7 @@ def emits_warning(*messages):
resetwarnings()
return decorate
+
def emits_warning_on(db, *warnings):
"""Mark a test as emitting a warning on a specific dialect.
@@ -115,7 +117,6 @@ def uses_deprecated(*messages):
return decorate
-
def global_cleanup_assertions():
"""Check things that have to be finalized at the end of a test suite.
@@ -129,28 +130,32 @@ def global_cleanup_assertions():
assert not pool._refs, str(pool._refs)
-
def eq_(a, b, msg=None):
"""Assert a == b, with repr messaging on failure."""
assert a == b, msg or "%r != %r" % (a, b)
+
def ne_(a, b, msg=None):
"""Assert a != b, with repr messaging on failure."""
assert a != b, msg or "%r == %r" % (a, b)
+
def is_(a, b, msg=None):
"""Assert a is b, with repr messaging on failure."""
assert a is b, msg or "%r is not %r" % (a, b)
+
def is_not_(a, b, msg=None):
"""Assert a is not b, with repr messaging on failure."""
assert a is not b, msg or "%r is %r" % (a, b)
+
def startswith_(a, fragment, msg=None):
"""Assert a.startswith(fragment), with repr messaging on failure."""
assert a.startswith(fragment), msg or "%r does not start with %r" % (
a, fragment)
+
def assert_raises(except_cls, callable_, *args, **kw):
try:
callable_(*args, **kw)
@@ -161,6 +166,7 @@ def assert_raises(except_cls, callable_, *args, **kw):
# assert outside the block so it works for AssertionError too !
assert success, "Callable did not raise an exception"
+
def assert_raises_message(except_cls, msg, callable_, *args, **kwargs):
try:
callable_(*args, **kwargs)
@@ -214,7 +220,9 @@ class AssertsCompiledSQL(object):
p = c.construct_params(params)
eq_(tuple([p[x] for x in c.positiontup]), checkpositional)
+
class ComparesTables(object):
+
def assert_tables_equal(self, table, reflected_table, strict_types=False):
assert len(table.c) == len(reflected_table.c)
for c, reflected_c in zip(table.c, reflected_table.c):
@@ -224,15 +232,19 @@ class ComparesTables(object):
eq_(c.nullable, reflected_c.nullable)
if strict_types:
+ msg = "Type '%s' doesn't correspond to type '%s'"
assert type(reflected_c.type) is type(c.type), \
- "Type '%s' doesn't correspond to type '%s'" % (reflected_c.type, c.type)
+ msg % (reflected_c.type, c.type)
else:
self.assert_types_base(reflected_c, c)
if isinstance(c.type, sqltypes.String):
eq_(c.type.length, reflected_c.type.length)
- eq_(set([f.column.name for f in c.foreign_keys]), set([f.column.name for f in reflected_c.foreign_keys]))
+ eq_(
+ set([f.column.name for f in c.foreign_keys]),
+ set([f.column.name for f in reflected_c.foreign_keys])
+ )
if c.server_default:
assert isinstance(reflected_c.server_default,
schema.FetchedValue)
@@ -246,6 +258,7 @@ class ComparesTables(object):
"On column %r, type '%s' doesn't correspond to type '%s'" % \
(c1.name, c1.type, c2.type)
+
class AssertsExecutionResults(object):
def assert_result(self, result, class_, *objects):
result = list(result)
@@ -296,6 +309,7 @@ class AssertsExecutionResults(object):
len(found), len(expected)))
NOVALUE = object()
+
def _compare_item(obj, spec):
for key, value in spec.iteritems():
if isinstance(value, tuple):
@@ -347,7 +361,8 @@ class AssertsExecutionResults(object):
self.assert_sql_execution(db, callable_, *newrules)
def assert_sql_count(self, db, callable_, count):
- self.assert_sql_execution(db, callable_, assertsql.CountStatements(count))
+ self.assert_sql_execution(
+ db, callable_, assertsql.CountStatements(count))
@contextlib.contextmanager
def assert_execution(self, *rules):
@@ -359,4 +374,4 @@ class AssertsExecutionResults(object):
assertsql.asserter.clear_rules()
def assert_statement_count(self, count):
- return self.assert_execution(assertsql.CountStatements(count)) \ No newline at end of file
+ return self.assert_execution(assertsql.CountStatements(count))