File size: 1,814 Bytes
9fc57a0 | 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 | """The entry file must hydrate its own pinned snapshot, including for forks."""
from pathlib import Path
import runpy
import tempfile
from types import SimpleNamespace
import unittest
from unittest.mock import Mock, patch
class HubLoadingTests(unittest.TestCase):
def test_fork_revision_and_cache_are_preserved(self):
source = Path(__file__).resolve().parents[1] / "env.py"
with tempfile.TemporaryDirectory() as temporary:
cache = Path(temporary)
revision = "a" * 40
snapshot = cache / "models--owner--g1-custom" / "snapshots" / revision
snapshot.mkdir(parents=True)
entry = snapshot / "env.py"
entry.write_bytes(source.read_bytes())
factory = SimpleNamespace(SimulatorFactory=Mock(), init_channel=Mock())
with (
patch("huggingface_hub.snapshot_download") as download,
patch.dict("sys.modules", {"sim.simulator_factory": factory}),
patch("sys.path", list(__import__("sys").path)),
):
runpy.run_path(str(entry))
download.assert_called_once_with("owner/g1-custom", revision=revision, cache_dir=str(cache))
def test_incomplete_local_checkout_fails_without_remote_fallback(self):
source = Path(__file__).resolve().parents[1] / "env.py"
with tempfile.TemporaryDirectory() as temporary:
entry = Path(temporary) / "env.py"
entry.write_bytes(source.read_bytes())
with patch("huggingface_hub.snapshot_download") as download:
with self.assertRaisesRegex(FileNotFoundError, "Incomplete simulator checkout"):
runpy.run_path(str(entry))
download.assert_not_called()
if __name__ == "__main__":
unittest.main()
|