| from __future__ import annotations |
|
|
| import json |
| import re |
| from pathlib import Path |
|
|
|
|
| ROOT = Path(__file__).resolve().parents[1] |
| NOTEBOOK = ROOT / "PolyGuard_SFT_GRPO_One_Run_Runner.ipynb" |
|
|
|
|
| def test_root_one_run_runner_notebook_covers_full_pipeline_without_secrets() -> None: |
| assert NOTEBOOK.exists() |
| payload = json.loads(NOTEBOOK.read_text(encoding="utf-8")) |
| assert payload.get("nbformat") == 4 |
|
|
| source = "\n".join( |
| "".join(cell.get("source", [])) |
| for cell in payload.get("cells", []) |
| if isinstance(cell, dict) |
| ) |
| assert "POLYGUARD_ONE_RUN_RUNNER" in source |
|
|
| required_markers = [ |
| "scripts/bootstrap_data.py", |
| "scripts/build_training_corpus.py", |
| "scripts/train_sft_trl.py", |
| "scripts/train_grpo_trl.py", |
| "scripts/deploy_training_space.py", |
| "scripts/pull_training_artifacts.py", |
| "scripts/generate_hf_training_report.py", |
| "scripts/test_inference_postsave.py", |
| "scripts/deploy_space_api.py", |
| ] |
| for marker in required_markers: |
| assert marker in source |
|
|
| assert not re.search(r"hf_[A-Za-z0-9]{20,}", source) |
|
|
|
|
| def test_readme_points_colab_link_at_root_runner_notebook() -> None: |
| readme = (ROOT / "README.md").read_text(encoding="utf-8") |
| assert "PolyGuard_SFT_GRPO_One_Run_Runner.ipynb" in readme |
|
|
