Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
File size: 1,184 Bytes
ff8c636 | 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 | from agent.core.hf_access import jobs_access_from_whoami
def test_personal_pro_prefers_username_namespace():
access = jobs_access_from_whoami({
"name": "alice",
"plan": "pro",
"orgs": [],
})
assert access.plan == "pro"
assert access.eligible_namespaces == ["alice"]
assert access.default_namespace == "alice"
def test_free_user_with_paid_org_uses_org_namespace():
access = jobs_access_from_whoami({
"name": "alice",
"plan": "free",
"orgs": [
{"name": "team-a", "plan": "team"},
{"name": "oss-friends", "plan": "free"},
],
})
assert access.plan == "org"
assert access.personal_can_run_jobs is False
assert access.eligible_namespaces == ["team-a"]
assert access.default_namespace is None
def test_free_user_without_paid_org_cannot_run_jobs():
access = jobs_access_from_whoami({
"name": "alice",
"plan": "free",
"orgs": [{"name": "community", "plan": "free"}],
})
assert access.plan == "free"
assert access.can_run_jobs is False
assert access.eligible_namespaces == []
assert access.default_namespace is None
|