Spaces:
Sleeping
Sleeping
File size: 1,622 Bytes
38c9982 696d083 38c9982 696d083 38c9982 696d083 38c9982 | 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 | from pathlib import Path
from src.executive_assistant.deployment import (
HFSpaceDeployConfig,
parse_hf_usernames,
render_space_readme,
stage_space_bundle,
)
def test_parse_hf_usernames_strips_at_signs() -> None:
usernames = parse_hf_usernames("@alice, bob , ,@carol")
assert usernames == ("alice", "bob", "carol")
def test_render_space_readme_includes_team_epsilon_roster() -> None:
config = HFSpaceDeployConfig(
repo_id="Flickinshots/EmailMaestro",
team_name="Team Epsilon",
hf_usernames=("flickinshots", "ShreyaKhatik", "itsayushdey"),
)
rendered = render_space_readme(config)
assert "Team Epsilon" in rendered
assert "@flickinshots" in rendered
assert "Team lead and primary Space owner" in rendered
assert "sdk: docker" in rendered
assert "OpenEnv Scaler x Meta x PyTorch Hack" in rendered
def test_stage_space_bundle_writes_hf_readme_and_checkpoint(tmp_path: Path) -> None:
config = HFSpaceDeployConfig(
repo_id="placeholder/project-epsilon-executive-assistant",
hf_usernames=("HF_USERNAME_1",),
)
checkpoint_path = stage_space_bundle(config, tmp_path)
assert checkpoint_path is not None
assert (tmp_path / "README.md").exists()
assert (tmp_path / "app.py").exists()
assert (tmp_path / "src" / "executive_assistant" / "env.py").exists()
assert (tmp_path / "artifacts" / "checkpoints" / config.checkpoint_name).exists()
assert not (tmp_path / ".env.app").exists()
assert not (tmp_path / ".env.training").exists()
assert not (tmp_path / ".env.hf.space.example").exists()
|