from pathlib import Path


def test_requirements_include_official_magpie_runtime_dependencies() -> None:
    requirements = Path("requirements.txt").read_text(encoding="utf-8")

    assert "nemo_toolkit[tts]@git+https://github.com/NVIDIA/NeMo.git@main" in requirements
    assert "kaldialign" in requirements


def test_modal_app_uses_official_magpie_runtime_dependencies() -> None:
    source = Path("modal_app.py").read_text(encoding="utf-8")

    assert '"nemo_toolkit[tts]@git+https://github.com/NVIDIA/NeMo.git@main"' in source
    assert '"kaldialign"' in source