fffiloni commited on
Commit
34dde93
·
verified ·
1 Parent(s): 1b7c496

Upload test_hardware_strategy_payload.py

Browse files
tests/test_hardware_strategy_payload.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from src.worker_payload import UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
2
+
3
+
4
+ def test_hardware_requested_at_space_creation_first():
5
+ assert "space_hardware=hardware" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
6
+ assert "create_space_with_hardware_strategy" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
7
+ assert "create_repo_space_hardware_first" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
8
+
9
+
10
+ def test_post_create_hardware_retry_stops_on_auth_errors():
11
+ assert "is_auth_or_billing_like_error" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
12
+ assert "401" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT
13
+ assert "manual_action_required" in UNIVERSAL_MODEL_CARD_WORKER_SCRIPT