from __future__ import annotations

import re
from pathlib import Path

import pytest

tomllib = pytest.importorskip("tomllib")


def test_training_dependencies_are_not_required_for_runtime() -> None:
    root = Path(__file__).parents[1]
    project = tomllib.loads((root / "pyproject.toml").read_text())
    runtime = set(project["project"]["dependencies"])
    training = set(project["project"]["optional-dependencies"]["train"])
    extras = project["project"]["optional-dependencies"]

    assert runtime.isdisjoint(training)
    assert extras["gpu"]
    assert extras["metal"]
    assert not set(extras["gpu"]).intersection(training)
    assert not set(extras["metal"]).intersection(training)

    requirements = {
        line.strip()
        for line in (root / "requirements.txt").read_text().splitlines()
        if line.strip() and not line.startswith("#")
    }
    train_requirements = {
        line.strip()
        for line in (root / "requirements-train.txt").read_text().splitlines()
        if line.strip() and not line.startswith("#") and not line.startswith("-r ")
    }
    assert requirements == runtime
    assert train_requirements == training


def test_setup_scripts_install_extras_that_exist() -> None:
    root = Path(__file__).parents[1]
    extras = set(tomllib.loads((root / "pyproject.toml").read_text())["project"]["optional-dependencies"])
    for script in ("setup", "setup.bat"):
        for group in re.findall(r"\.\[([^\]]+)\]", (root / script).read_text()):
            assert set(group.split(",")) <= extras, script