Skip to content

Commit

Permalink
[Smoke Tests] Fix tests for job queue (skypilot-org#1265)
Browse files Browse the repository at this point in the history
Fix tests for job queue
  • Loading branch information
Michaelvll authored and ewzeng committed Oct 24, 2022
1 parent c091556 commit 09b7720
Showing 1 changed file with 18 additions and 11 deletions.
29 changes: 18 additions & 11 deletions tests/test_smoke.py
Original file line number Diff line number Diff line change
Expand Up @@ -285,11 +285,15 @@ def test_job_queue():
'job_queue',
[
f'sky launch -y -c {name} examples/job_queue/cluster.yaml',
f'sky exec {name} -d examples/job_queue/job.yaml',
f'sky exec {name} -d examples/job_queue/job.yaml',
f'sky exec {name} -d examples/job_queue/job.yaml',
f'sky logs {name} 2',
f'sky queue {name}',
f'sky exec {name} -n {name}-1 -d examples/job_queue/job.yaml',
f'sky exec {name} -n {name}-2 -d examples/job_queue/job.yaml',
f'sky exec {name} -n {name}-3 -d examples/job_queue/job.yaml',
f'sky queue {name} | grep {name}-1 | grep RUNNING',
f'sky queue {name} | grep {name}-2 | grep RUNNING',
f'sky queue {name} | grep {name}-3 | grep PENDING',
f'sky cancel {name} 2',
'sleep 5',
f'sky queue {name} | grep {name}-3 | grep RUNNING',
],
f'sky down -y {name}',
)
Expand All @@ -302,12 +306,15 @@ def test_n_node_job_queue():
'job_queue_multinode',
[
f'sky launch -y -c {name} examples/job_queue/cluster_multinode.yaml',
f'sky exec {name} -d examples/job_queue/job_multinode.yaml',
f'sky exec {name} -d examples/job_queue/job_multinode.yaml',
f'sky exec {name} -d examples/job_queue/job_multinode.yaml',
f'sky exec {name} -n {name}-1 -d examples/job_queue/job_multinode.yaml',
f'sky exec {name} -n {name}-2 -d examples/job_queue/job_multinode.yaml',
f'sky exec {name} -n {name}-3 -d examples/job_queue/job_multinode.yaml',
f'sky queue {name} | grep {name}-1 | grep RUNNING',
f'sky queue {name} | grep {name}-2 | grep RUNNING',
f'sky queue {name} | grep {name}-3 | grep PENDING',
f'sky cancel {name} 1',
f'sky logs {name} 2',
f'sky queue {name}',
'sleep 5',
f'sky queue {name} | grep {name}-3 | grep RUNNING',
],
f'sky down -y {name}',
)
Expand All @@ -319,7 +326,7 @@ def test_large_job_queue():
test = Test(
'large_job_queue',
[
f'sky launch -y -c {name} --cloud gcp ""',
f'sky launch -y -c {name} --cloud gcp',
f'for i in `seq 1 75`; do sky exec {name} -d "echo $i; sleep 100000000"; done',
f'sky cancel {name} 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16',
'sleep 20',
Expand Down

0 comments on commit 09b7720

Please sign in to comment.