def test_triple_quaff(self): customer = Customer() customer.buy_beer("triple") customer.quaff() self.assertEqual(customer.glass.content, 56)
def test_half_quaff(self): customer = Customer() customer.buy_beer("half") customer.quaff() self.assertEqual(customer.glass.content, 6)