add smoke test for BatchStatement.__len__

mambocab · mambocab · commit 6598eabe722e · 2017-01-20T10:32:55.000-05:00
diff --git a/tests/unit/test_query.py b/tests/unit/test_query.py
@@ -66,3 +66,10 @@ def test_add_all(self):
         bound_statements = [t[1] for t in batch._statements_and_parameters]
         str_parameters = [str(i) for i in range(10)]
         self.assertEqual(bound_statements, str_parameters)
+
+    def test_len(self):
+        for n in 0, 10, 100:
+            batch = BatchStatement()
+            batch.add_all(statements=['%s'] * n,
+                          parameters=[(i,) for i in range(n)])
+            self.assertEqual(len(batch), n)
