File size: 663 Bytes
d5257f7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from dashboard.runtime_env import is_hosted_space


def test_is_hosted_space_false_without_hf_environment(monkeypatch):
    monkeypatch.delenv("SYSTEM", raising=False)
    monkeypatch.delenv("SPACE_ID", raising=False)

    assert is_hosted_space() is False


def test_is_hosted_space_true_for_spaces_system(monkeypatch):
    monkeypatch.setenv("SYSTEM", "spaces")
    monkeypatch.delenv("SPACE_ID", raising=False)

    assert is_hosted_space() is True


def test_is_hosted_space_true_when_space_id_is_defined(monkeypatch):
    monkeypatch.delenv("SYSTEM", raising=False)
    monkeypatch.setenv("SPACE_ID", "Viney/amplegest")

    assert is_hosted_space() is True