如何用unittest修补不同返回值的链式函数？

在Python中，unittest 是一个内建的库，用于编写和运行测试。链式函数通常指的是一系列方法调用，每个方法都返回对象本身，以便可以继续链式调用。

假设我们有一个链式函数，它根据不同的输入返回不同的值，我们可以使用 unittest 来测试这些不同的返回值。下面是一个简单的例子，展示如何为这样的链式函数编写测试用例。

示例链式函数

首先，我们定义一个简单的链式函数类：

class ChainFunction:
    def __init__(self, value):
        self.value = value

    def add(self, amount):
        self.value += amount
        return self

    def subtract(self, amount):
        self.value -= amount
        return self

    def multiply(self, amount):
        self.value *= amount
        return self

    def result(self):
        return self.value

使用unittest编写测试用例

接下来，我们使用 unittest 来编写测试用例，检查链式函数在不同操作下的返回值：

import unittest

class TestChainFunction(unittest.TestCase):
    def test_addition(self):
        chain = ChainFunction(0)
        result = chain.add(5).result()
        self.assertEqual(result, 5)

    def test_subtraction(self):
        chain = ChainFunction(10)
        result = chain.subtract(3).result()
        self.assertEqual(result, 7)

    def test_multiplication(self):
        chain = ChainFunction(2)
        result = chain.multiply(4).result()
        self.assertEqual(result, 8)

    def test_combined_operations(self):
        chain = ChainFunction(1)
        result = chain.add(2).subtract(1).multiply(3).result()
        self.assertEqual(result, 6)

if __name__ == '__main__':
    unittest.main()