import unittest
from cost_unit import CostUnit
class TestCostUnit(unittest.TestCase):
    def setUp(self):
        self.cost_unit = CostUnit()
    def test_add_cost(self):
        """测试添加成本功能"""
        self.cost_unit.add_cost(100)
        self.assertEqual(self.cost_unit.total_cost, 100)
    def test_subtract_cost(self):
        """测试减去成本功能"""
        self.cost_unit.add_cost(200)
        self.cost_unit.subtract_cost(50)
        self.assertEqual(self.cost_unit.total_cost, 150)
    def test_reset_cost(self):
        """测试重置成本功能"""
        self.cost_unit.add_cost(300)
        self.cost_unit.reset_cost()
        self.assertEqual(self.cost_unit.total_cost, 0)
if __name__ == '__main__':
    unittest.main()

在这个例子中，我们创建了一个名为TestCostUnit的测试类，该类继承自unittest.TestCase，我们在setUp方法中初始化了CostUnit类的实例，我们编写了三个测试方法：

1、test_add_cost：测试添加成本的功能是否正确。

2、test_subtract_cost：测试减去成本的功能是否正确。

3、test_reset_cost：测试重置成本的功能是否正确。

每个测试方法都使用了assertEqual断言来验证预期结果与实际结果是否相等，如果所有断言都通过，那么测试就会成功，如果有任何一个断言失败，测试就会失败，并显示相应的错误信息。