- Notifications
You must be signed in to change notification settings - Fork 633
/
Copy pathtest_api.py
22 lines (15 loc) · 667 Bytes
/
test_api.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
importsky
fromsky.clouds.cloudimportCloud
deftest_sky_launch(enable_all_clouds):
task=sky.Task()
job_id, handle=sky.get(sky.launch(task, dryrun=True))
assertjob_idisNoneandhandleisNone
deftest_k8s_alias(monkeypatch, enable_all_clouds):
defdryrun_task_with_cloud(cloud: Cloud):
task=sky.Task()
task.set_resources_override({'cloud': cloud})
sky.stream_and_get(sky.launch(task, dryrun=True))
monkeypatch.setattr('sky.provision.kubernetes.utils.get_spot_label',
lambda*_, **__: [None, None])
dryrun_task_with_cloud(sky.K8s())
dryrun_task_with_cloud(sky.Kubernetes())