]> git.ipfire.org Git - thirdparty/sqlalchemy/sqlalchemy.git/commitdiff
Tweaks for assert_unordered_result
authorJason Kirtland <jek@discorporate.us>
Thu, 1 Nov 2007 23:13:30 +0000 (23:13 +0000)
committerJason Kirtland <jek@discorporate.us>
Thu, 1 Nov 2007 23:13:30 +0000 (23:13 +0000)
test/testlib/testing.py

index 60b0173d8f81844c952865ca0892e24c7f3b1f27..941ddf2497b3553a86b632b6737f76cf5f49b250 100644 (file)
@@ -316,12 +316,18 @@ class AssertMixin(PersistTest):
             self.fail('Unexpected type "%s", expected "%s"' % (
                 type(wrong).__name__, cls.__name__))
 
+        if len(found) != len(expected):
+            self.fail('Unexpected object count "%s", expected "%s"' % (
+                len(found), len(expected)))
+
         NOVALUE = object()
         def _compare_item(obj, spec):
             for key, value in spec.iteritems():
                 if isinstance(value, tuple):
-                    if (not self.assert_unordered_result(
-                          getattr(obj, key), value[0], *value[1])):
+                    try:
+                        self.assert_unordered_result(
+                            getattr(obj, key), value[0], *value[1])
+                    except AssertionError:
                         return False
                 else:
                     if getattr(obj, key, NOVALUE) != value: