| import os |
| import subprocess |
| import textwrap |
| from pathlib import Path |
|
|
| import pytest |
| import yaml |
|
|
|
|
| EXPECTED = { |
| "flm": "https://github.com/david3684/flm.git", |
| "langflow": "https://github.com/nealchen2003/LangFlow.git", |
| "duo": "https://github.com/s-sahoo/duo.git", |
| "mdlm": "https://github.com/kuleshov-group/mdlm.git", |
| "candi": "https://github.com/patrickpynadath1/candi-diffusion.git", |
| "rdlm": "https://github.com/harryjo97/RDLM.git", |
| "sdtt": "https://github.com/jdeschena/sdtt.git", |
| "di4c": "https://github.com/sony/di4c.git", |
| } |
|
|
|
|
| @pytest.fixture |
| def source_manifest(): |
| with Path("artifacts/sources.yaml").open(encoding="utf-8") as manifest_file: |
| return yaml.safe_load(manifest_file) |
|
|
|
|
| def test_source_manifest_has_exact_repositories(source_manifest): |
| assert {name: source["url"] for name, source in source_manifest.items()} == EXPECTED |
| assert all(len(source["commit"]) == 40 for source in source_manifest.values()) |
|
|
|
|
| def test_fetch_sources_uses_active_conda_python(tmp_path): |
| conda_prefix = tmp_path / "conda" |
| python_bin = conda_prefix / "bin" / "python" |
| python_log = tmp_path / "python.log" |
| python_bin.parent.mkdir(parents=True) |
| python_bin.write_text( |
| textwrap.dedent( |
| """\ |
| #!/usr/bin/env bash |
| printf '%s\n' "$0" > "$FAKE_PYTHON_LOG" |
| printf '%s\n' "$@" >> "$FAKE_PYTHON_LOG" |
| exit 42 |
| """ |
| ) |
| ) |
| python_bin.chmod(0o755) |
|
|
| environment = { |
| **os.environ, |
| "CONDA_PREFIX": str(conda_prefix), |
| "FAKE_PYTHON_LOG": str(python_log), |
| } |
| environment.pop("DLB_PYTHON", None) |
| environment.pop("PYTHON_BIN", None) |
|
|
| completed = subprocess.run( |
| ["bash", "scripts/fetch_sources.sh"], |
| cwd=Path.cwd(), |
| env=environment, |
| text=True, |
| capture_output=True, |
| ) |
|
|
| assert completed.returncode == 42, completed.stderr |
| assert python_log.read_text().splitlines() == [str(python_bin), "-", str(Path.cwd())] |
|
|
