def test_drink(self): customer = Customer() customer.buy_beer() customer.drink() self.assertEqual(customer.glass.content, 19)
def test_triple_drink(self): customer = Customer() customer.buy_beer("triple") customer.drink() self.assertEqual(customer.glass.content, 59)