Add more tests

ummae · May 3, 2015 · c5761f6 · c5761f6
1 parent 209ce27
commit c5761f6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -199,9 +199,17 @@ def test_multiple_counters():
     for col, delta in zip(columns, [1, 3, 5]):
         table.counters_inc(row, [(col, delta)])
         assert_equal(delta, table.counter_get(row, col))
-        table.counter_dec(row, [(col, delta)])
+        table.counter_dec(row, [(col, -delta)])
         assert_equal(0, table.counter_get(row, col))
 
+    table.counters_inc(row, zip(columns, [1, 3, 5]))
+    table.counters_inc(row, zip(columns, [1, 3, 5]))
+    assert_equal([2, 6, 10], [table.counter_get(row, c) for c in columns])
+
+    table.counters_inc(row, zip(columns, [-1, -3, -5]))
+    table.counters_inc(row, zip(columns, [-1, -3, -5]))
+    assert_equal([0, 0, 0], [table.counter_get(row, c) for c in columns])
+
 
 def test_batch():
     with assert_raises(TypeError):