Coding Agent、补丁与测试反馈

New-Item -ItemType Directory -Force l9-coding
cd l9-coding

mkdir -p l9-coding
cd l9-coding

from pathlib import Path
import subprocess
import sys
import tempfile

EXPRESSION = "price * count"

def validate_patch(expression):
    if expression not in {"price * count", "price + count"}:
        raise ValueError("补丁表达式未批准")
    # 使用系统配置的临时位置创建独立工作区，退出时清理文件。
    with tempfile.TemporaryDirectory() as folder:
        work = Path(folder)
        target = work / "shop.py"
        # 先落盘错误基线，再只替换指定的金额公式行，生成真实候选文件。
        baseline = "def total(price, count):\n    return price + count\n"
        target.write_text(baseline, encoding="utf-8")
        source = target.read_text(encoding="utf-8")
        source = source.replace("    return price + count\n",
                                "    return " + expression + "\n", 1)
        target.write_text(source, encoding="utf-8")
        # 测试是预先给定的业务要求：单价10元、3件共30元，0件为0。
        checks = ("import sys\nfrom pathlib import Path\n"
                  "sys.path.insert(0, str(Path(__file__).parent))\n"
                  "from shop import total\n"
                  "assert total(10, 3) == 30\n"
                  "assert total(10, 0) == 0\n"
                  "print('两条业务断言通过')\n")
        (work / "check.py").write_text(checks, encoding="utf-8")
        result = subprocess.run([sys.executable, "-X", "utf8", "check.py"],
                                cwd=work, capture_output=True, text=True,
                                encoding="utf-8", timeout=5)
        return result.returncode, result.stdout.strip(), result.stderr.strip()

if __name__ == "__main__":
    code, feedback, errors = validate_patch(EXPRESSION)
    print("候选补丁：return " + EXPRESSION)
    print("测试退出码：" + str(code))
    print("处理：" + ("允许进入人工审查" if code == 0 else "退回补丁"))
    if feedback:
        print(feedback)
    if errors:
        # stderr保留完整回溯；正文演示抽出失败的业务断言和错误类别。
        failed = next((line.strip() for line in errors.splitlines() if "assert " in line), "")
        if failed:
            print("失败断言：" + failed)
        print("错误：" + errors.splitlines()[-1])

import unittest
import patch_lab as lab


class MechanismTests(unittest.TestCase):
    # 正常、边界和失败输入都由这里提供，直接运行即可。

    def test_normal(self):
        self.assertEqual(lab.validate_patch("price * count"), (0, "两条业务断言通过", ""))

    def test_boundary_zero_in_child(self):
        code, feedback, errors = lab.validate_patch("price * count")
        self.assertEqual(code, 0)
        self.assertIn("两条", feedback)

    def test_failure_patch(self):
        code, output, errors = lab.validate_patch("price + count")
        self.assertNotEqual(code, 0)
        self.assertIn("assert total(10, 3) == 30", errors)
        self.assertIn("AssertionError", errors)
        with self.assertRaises(ValueError):
            lab.validate_patch("__import__('os')")


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

# 在刚才创建的练习目录检查文件，再运行程序。
ls
python -X utf8 patch_lab.py
python -X utf8 -m unittest -v test_patch_lab.py

候选补丁：return price * count
测试退出码：0
处理：允许进入人工审查
两条业务断言通过

python -X utf8 patch_lab.py