舍入引发单元测试冲突，如何同时通过Task1与Task2的单元测试？

问题本质 #

你碰到的是典型的浮点数精度冲突：Task1需要对计算结果做舍入以匹配预期值，但Task2依赖未舍入的高精度结果进行累计计算，复用同一个函数就会顾此失彼。

三种可行解决办法 #

1. 给函数加可选舍入参数（推荐） #

修改calculate_gains，增加一个可选参数控制是否舍入，默认不开启（适配Task2），Task1调用时手动触发舍入。

示例代码：

def calculate_gains(investment, round_to=None):
    # 你的原始收益计算逻辑
    total_gains = ...  # 比如根据投资额计算出的高精度结果
    
    # 如果指定了舍入位数，就返回舍入后的值；否则返回原始值
    if round_to is not None:
        return round(total_gains, round_to)
    return total_gains

调用方式：

• Task1：calculate_gains(3000000, round_to=2) 或者 round_to=0，直接得到164000.0
• Task2：直接调用calculate_gains(your_investment)，保留原始高精度数值，不影响累计计算结果

2. 拆分计算与格式化逻辑 #

把函数拆成两个：一个负责核心计算（返回高精度结果，供Task2复用），另一个负责Task1的舍入格式化。

示例代码：

# 核心计算函数，只做纯计算，不处理格式
def calculate_gains_raw(investment):
    total_gains = ...  # 原始高精度计算逻辑
    return total_gains

# Task1专用函数，调用核心函数后做舍入
def calculate_gains(investment):
    raw_result = calculate_gains_raw(investment)
    return round(raw_result, 2)

使用场景：

• Task1直接调用calculate_gains(3000000)，得到符合预期的舍入值
• Task2循环调用calculate_gains_raw计算每月收益，累计后得到正确的高精度结果

3. 调整Task1的测试断言逻辑（仅当允许修改测试时用） #

如果不能改函数代码，只能调整测试用例，给浮点数断言加精度容忍：

# 把原来的精确断言改成误差容忍断言
assert abs(calculate_gains(3000000) - 164000.0) < 1e-8

这种方式利用浮点数计算的特性，允许极小的误差范围，既保留原始函数的高精度，又能让Task1测试通过。

方案选择建议 #

• 优先选方案1：代码改动最小，灵活性最高，一个函数适配两个场景
• 追求代码职责清晰选方案2：计算和格式化完全分离，后续扩展更方便
• 只能改测试不能改函数选方案3：快速解决测试冲突，但要确保测试规则允许精度容忍

内容的提问来源于stack exchange，提问作者dmangee