def test_invite_more_people(self): party = PizzaParty().invite(7).order(2) self.assertEqual(2, party.leftovers()) self.assertEqual(2, party.pieces_per_person()) party.invite(9) self.assertEqual(0, party.leftovers()) self.assertEqual(1, party.pieces_per_person())
def test_order_more_pizza(self): party = PizzaParty().invite(8).order(2) self.assertEqual(0, party.leftovers()) self.assertEqual(2, party.pieces_per_person()) party.order(2) self.assertEqual(0, party.leftovers()) self.assertEqual(4, party.pieces_per_person())
def run(read, prt): people = int(read('How many people? ')) pizza_order = read('How many pizzas? ') order_parts = str.split(pizza_order) pizzas = int(order_parts[0]) size = None if len(order_parts) == 2: size = order_parts[1] prt('%s people with %s pizzas' % (people, pizzas)) party = PizzaParty().invite(people).order(pizzas, size) prt('Each person gets %s pieces of pizza.' % (party.pieces_per_person())) prt('There are %s leftover pieces.' % (party.leftovers()))
def test_order_large_pizza(self): party = PizzaParty().invite(30).order(4, 'large') self.assertEqual(2, party.leftovers()) self.assertEqual(1, party.pieces_per_person())
def test_order_medium_pizza(self): party = PizzaParty().invite(5).order(2, 'medium') self.assertEqual(2, party.leftovers()) self.assertEqual(2, party.pieces_per_person())
def test_order_small_pizza(self): party = PizzaParty().invite(3).order(1, 'small') self.assertEqual(1, party.leftovers()) self.assertEqual(1, party.pieces_per_person())
def test_some_leftovers(self): party = PizzaParty().invite(7).order(2) self.assertEqual(2, party.leftovers()) self.assertEqual(2, party.pieces_per_person())
def test_per_person(self): party = PizzaParty().invite(8).order(2) self.assertEqual(2, party.pieces_per_person())