tests: assert_queries_count behaviour changed

it now returns function result when everything is ok
jqb · Oct 5, 2013 · 77f39e3 · 77f39e3
1 parent a7fbfc9
commit 77f39e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -20,10 +20,10 @@ def assert_queries_count(self, num, func=None, *args, **kwargs):
 
         context = _AssertNumQueriesContext(self, num, conn)
         if func is None:
-            return context
+            return None
 
         with context:
-            func(*args, **kwargs)
+            return func(*args, **kwargs)
 
 
 class TestCase(unittest.TestCase):