| from pathlib import Path |
| import textwrap |
|
|
| ROOT = Path(__file__).resolve().parents[1] |
|
|
|
|
| def test_worker_pins_transformers_and_safetensors_to_stable_build_line(): |
| worker = (ROOT / "src" / "worker_payload.py").read_text(encoding="utf-8") |
| assert "transformers>=4.51.0,<5.0.0" in worker |
| assert "Avoid uncontrolled Transformers 5.x" in worker |
|
|
|
|
| def test_worker_detects_safetensors_080_pip_resolver_failure(): |
| worker = (ROOT / "src" / "worker_payload.py").read_text(encoding="utf-8") |
| assert "def known_dependency_guardrail_needed" in worker |
| assert "safetensors>=0.8.0" in worker |
| assert "no matching distribution" in worker |
| assert "def apply_dependency_guardrail_repair" in worker |
| assert "dependency_guardrail_repair" in worker |
|
|
|
|
| def test_dependency_guardrail_runs_before_pi_diagnosis_loop(): |
| worker = (ROOT / "src" / "worker_payload.py").read_text(encoding="utf-8") |
| recovery = worker.split("def recover_after_live_validation_failure", 1)[1].split("def upload_workspace", 1)[0] |
| assert recovery.index("apply_dependency_guardrail_repair") < recovery.index("for iteration in range(1, 5)") |
| assert "Dependency-guardrailed workspace uploaded" in recovery |
| assert "falling back to Pi diagnosis" in recovery |
|
|
