写一个测试杯子的案例

好的，下面是一个测试杯子的案例：

假设我们有一个杯子类 Cup，它有以下属性和方法：

属性：
- volume: 杯子的容量，单位为毫升
- current_volume: 杯子当前的水量，单位为毫升

方法：
- add_water(volume): 往杯子中添加指定容量的水
- drink_water(volume): 从杯子中喝掉指定容量的水

现在我们需要对 Cup 类进行测试，以确保它的行为符合预期。下面是一个测试案例：

import unittest
from cup import Cup

class CupTest(unittest.TestCase):

    def test_add_water(self):
        cup = Cup(volume=500, current_volume=0)
        cup.add_water(250)
        self.assertEqual(cup.current_volume, 250)
        cup.add_water(300)
        self.assertEqual(cup.current_volume, 500)

    def test_drink_water(self):
        cup = Cup(volume=500, current_volume=500)
        cup.drink_water(250)
        self.assertEqual(cup.current_volume, 250)
        cup.drink_water(300)
        self.assertEqual(cup.current_volume, 0)

在这个测试案例中，我们编写了两个测试方法 test_add_water 和 test_drink_water。分别测试了添加水和喝水的行为。我们首先创建了一个 Cup 对象，并调用了 add_water 和 drink_water 方法。然后使用 assertEqual 方法来确保杯子中的水量符合预期。

通过这个测试案例，我们可以确保 Cup 类的行为符合预期，从而提高代码的质量和可靠性。