更新单元测试以匹配代码更改

对代码进行更改时，更新任何测试以验证新行为并捕获新代码引入的任何 bug 非常重要。 副驾驶聊天 可以帮助你快速更新测试以匹配代码更改，确保测试套件与实现保持同步。

示例方案

假设你有一个 Python 函数， calculate_discount用于确定给定购买金额的折扣。 在原始代码中，对于超过 100 美元的金额，可享受 10% 的折扣。 你正在更改函数的逻辑，以便只有价格超过 150 美元的项目获得 10% 折扣，而价格超过 200 美元的项目现在享有 20% 折扣。

原始代码

在原始代码中，购买价格超过 100 美元可获得 10% 折扣。

def calculate_discount(amount: float) -> float:
    if amount > 100:
        return amount * 0.1  # 10% discount
    return 0.0

更新的代码

在更改的代码中，只有超过 150 美元的金额将获得 10% 折扣，超过 200 美元的金额现在将获得 20% 折扣。

def calculate_discount(amount: float) -> float:
    if amount > 200:
        return amount * 0.2  
    elif amount > 150:
        return amount * 0.1
    return 0.0

现有测试

此示例演示为原始代码编写的现有测试。

import unittest
from discount import calculate_discount

class TestCalculateDiscount(unittest.TestCase):
    def test_discount_above_100(self):
        # $140 should get a $14 discount (10%)
        self.assertAlmostEqual(calculate_discount(140), 14.0)
    
    def test_discount_below_100(self):
        # $80 should get no discount
        self.assertEqual(calculate_discount(80), 0.0)

    def test_discount_exactly_100(self):
        # $100 is not above 100, so no discount
        self.assertEqual(calculate_discount(100), 0.0)

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

示例提示

可以要求 副驾驶聊天 识别和更新受逻辑更改影响的任何测试。

Given the update to the `calculate_discount` function, update the unit tests that may fail or become outdated as a result.

示例响应

副驾驶聊天 会分析你的代码和现有测试，并识别出哪些测试会失败，或在代码更改后变得具有误导性。

例如， 副驾驶聊天 说明：

• test_discount_above_100 现在将失败
• test_discount_below_100 而且 test_discount_exactly_100 会通过，但他们现在 具有误导性， 因为10% 折扣的门槛现在是150美元，而不是100美元。

此外，副驾驶聊天还能识别出任何缺失的测试覆盖，例如金额超过 200 美元时的新增 20% 折扣档位。

副驾驶聊天 建议一组更新的测试。

更新的测试

import unittest
from discount import calculate_discount

class TestCalculateDiscount(unittest.TestCase):
    def test_discount_above_200(self):
        # $250 should get a $50 discount (20%)
        self.assertAlmostEqual(calculate_discount(250), 50.0)

    def test_discount_above_150(self):
        # $175 should get a $17.50 discount (10%)
        self.assertAlmostEqual(calculate_discount(175), 17.5)

    def test_discount_at_or_below_150(self):
        # $150 is not above 150, so no discount
        self.assertEqual(calculate_discount(150), 0.0)
        # $80 should get no discount
        self.assertEqual(calculate_discount(80), 0.0)

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

延伸阅读

• GitHub Copilot 对话助手的提示设计
• 使用 GitHub Copilot 的最佳做法