File size: 2,030 Bytes
dc72977 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 | 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())]
|