import unittest
from src.drink import Drink

class TestDrink(unittest.TestCase):
    def setUp(self):
        self.drink = Drink("Rum", 5.50, 4)

    def test_drink_has_name(self):
        self.assertEqual("Rum", self.drink.name)

    def test_drink_has_price(self):
        self.assertEqual(5.50, self.drink.price)
    
    def test_drink_has_alcohol_units(self):
        self.assertEqual(4, self.drink.alcohol_units)