"""Regression: pipeline model paths like ckpts/foo must not be treated as Hub org/model."""

from __future__ import annotations

import re
import unittest
from pathlib import Path


_BASE_PATH = Path(__file__).resolve().parents[1] / "trellis/pipelines/base.py"


class PipelineHubIdTests(unittest.TestCase):
    def test_base_py_documents_ckpts_exclusion(self) -> None:
        src = _BASE_PATH.read_text(encoding="utf-8")
        self.assertIn("_NOT_HUB_PATH_PREFIXES", src)
        self.assertIn('"ckpts/"', src)
        self.assertIn("models.from_pretrained(os.path.join(root, vp))", src)

    def test_regex_ckpts_two_segment_not_hub_semantics(self) -> None:
        """Mirror _is_hub_model_id: two-segment ckpts/... is not a Hub id."""
        not_hub = ("ckpts/", "weights/", "configs/", "checkpoints/")

        def is_hub(t: str) -> bool:
            t = t.strip().replace("\\", "/")
            if not re.fullmatch(r"[\w.-]+/[\w.-]+", t):
                return False
            tl = t.lower()
            return not any(tl.startswith(p) for p in not_hub)

        self.assertFalse(is_hub("ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16"))
        self.assertTrue(is_hub("luh0502/NeAR"))


if __name__ == "__main__":
    unittest.main()