def test_count_add_capacity(self):
""" Count addition with consumed_capacity """
count = Count(4, 2, Capacity.create_read({'CapacityUnits': 3}))
count2 = Count(5, 3, Capacity.create_read({'CapacityUnits': 2}))
ret = count + count2
self.assertEqual(ret, 9)
self.assertEqual(ret.scanned_count, 5)
self.assertEqual(ret.consumed_capacity.read, 5)
def test_count_add_capacity(self):
"""Count addition with consumed_capacity"""
count = Count(4, 2, Capacity(3, 0))
count2 = Count(5, 3, Capacity(2, 0))
ret = count + count2
self.assertEqual(ret, 9)
self.assertEqual(ret.scanned_count, 5)
self.assertEqual(ret.consumed_capacity.read, 5)
def test_count_add_none_capacity(self):
""" Count addition with one None consumed_capacity """
cap = Capacity.create_read({'CapacityUnits': 3})
count = Count(4, 2)
count2 = Count(5, 3, cap)
ret = count + count2
self.assertEqual(ret, 9)
self.assertEqual(ret.scanned_count, 5)
self.assertEqual(ret.consumed_capacity, cap)
def test_count_division(self):
""" Count division """
count = Count(4, 2)
self.assertEqual(count / 2, 2)
def test_count_multiplication(self):
""" Count multiplication """
count = Count(4, 2)
self.assertEqual(2 * count, 8)
def test_count_subtraction(self):
""" Count subtraction """
count = Count(4, 2)
self.assertEqual(count - 2, 2)
def test_count_addition(self):
""" Count addition """
count = Count(4, 2)
self.assertEqual(count + 5, 9)
def test_count_repr(self):
""" Count repr """
count = Count(0, 0)
self.assertEqual(repr(count), "Count(0)")
