def test_invited_within_zero(self): result = invited(self.customers, self.santa_clara, 0) self.assertEqual(len(result), 0)
def test_invited_within_500(self): result = invited(self.customers, self.santa_clara, 500) self.assertEqual(len(result), 3) result = invited(self.customers, self.dublin, 500) self.assertEqual(len(result), 1)