File size: 641 Bytes
b67c03f
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from src.worker_payload import UNIVERSAL_MODEL_CARD_WORKER_SCRIPT


def test_hardware_requested_at_space_creation_first():
    assert "space_hardware=hardware" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
    assert "create_space_with_hardware_strategy" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
    assert "create_repo_space_hardware_first" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT


def test_post_create_hardware_retry_stops_on_auth_errors():
    assert "is_auth_or_billing_like_error" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
    assert "401" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
    assert "manual_action_required" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT