def test_RecipientTracker_countRecipients_nonRepeatDeliveries_north(self):
pos = 5 - 9j
move = 1 - 0j
rt = santa.RecipientTracker()
for i in range(1, 11):
rt.registerDelivery(pos)
self.assertEqual(i, rt.countRecipients(),
"start: {} iteration: {}".format(pos, i))
pos += move
def test_RecipientTracker_countRecipients_repeatDeliveries_cyclic(self):
poss = [4 + 1j, 3 - 3j, 4 - 6j, 10 - 3j]
rt = santa.RecipientTracker()
for i in range(5):
for pos in poss:
rt.registerDelivery(pos)
self.assertEqual(len(poss), rt.countRecipients(),
"iteration: {}".format(i))
def test_RecipientTracker_countRecipients_repeatDeliveries_grid(self):
size = 5
poss = [
complex(a, b) for a in range(-size, size + 1)
for b in range(-size, size + 1)
]
rt = santa.RecipientTracker()
for i in range(5):
for pos in poss:
rt.registerDelivery(pos)
self.assertEqual(len(poss), rt.countRecipients(),
"iteration: {}".format(i))
def test_RecipientTracker_init(self):
rt = santa.RecipientTracker()
self.assertEqual(0, rt.countRecipients())
