def test_triple_one_off(self): customer = Customer() customer.buy_beer("triple") customer.down_in_one() self.assertEqual(customer.glass.content, 0)
def test_too_much(self): customer = Customer() customer.buy_beer() customer.down_in_one() self.assertEqual(customer.glass.content, 0) self.assertRaises(Exception, customer.quaff)