Skip to content

Commit 5bed105

Browse files
committed
Fix job submit test
1 parent 6c74ea0 commit 5bed105

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/test-job.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,13 @@ def test_job_submit():
1515
"cpus_per_task": 3,
1616
}
1717
test_job_id = pyslurm.job().submit_batch_job(test_job)
18-
test_job_search = pyslurm.job().find(name="name", val="pyslurm_test_job")
18+
test_job_search = pyslurm.job().find(name="name", val=test_job["job_name"])
19+
test_job_info = pyslurm.job().find_id(test_job_id)
20+
1921
assert_true(test_job_id in test_job_search)
20-
assert_equals(test_job_search["cpus_per_task"], 3)
21-
assert_equals(test_job_search["num_tasks"], 2)
22+
assert_equals(len(test_job_info), 1)
23+
assert_equals(test_job_info[0]["cpus_per_task"], test_job["cpus_per_task"])
24+
assert_equals(test_job_info[0]["num_tasks"], test_job["ntasks"])
2225

2326

2427
def test_job_get():

0 commit comments

Comments
 (0)