Skip to content

Commit d137aa6

Browse files
author
Ramon Adolfo Arambula
committed
tests: added new dir + pytestable file for each snippet from docs
1 parent d46d09e commit d137aa6

File tree

5 files changed

+67
-0
lines changed

5 files changed

+67
-0
lines changed
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from psij import Job, JobSpec, JobExecutor
2+
from executor_test_params import ExecutorTestParams
3+
4+
5+
def test_getting_started_job_arguements(execparams: ExecutorTestParams) -> None:
6+
ex = JobExecutor.get_instance(execparams.executor)
7+
job = Job(JobSpec(executable='/bin/date', arguments=['-utc', '--debug']))
8+
ex.submit(job)
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from psij import Job, JobSpec, JobExecutor
2+
from executor_test_params import ExecutorTestParams
3+
4+
5+
def test_getting_started_multiple_jobs(execparams: ExecutorTestParams) -> None:
6+
ex = JobExecutor.get_instance(execparams.executor)
7+
for _ in range(10):
8+
job = Job(JobSpec(executable="/bin/date"))
9+
ex.submit(job)
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
from psij import Job, JobSpec, JobExecutor, ResourceSpecV1, JobAttributes
2+
from executor_test_params import ExecutorTestParams
3+
4+
5+
def test_getting_started_scheduling_info(execparams: ExecutorTestParams) -> None:
6+
executor = JobExecutor.get_instance(execparams.executor)
7+
8+
job = Job(
9+
JobSpec(
10+
executable="/bin/date",
11+
resources=ResourceSpecV1(node_count=1),
12+
attributes=JobAttributes(
13+
queue_name="<QUEUE_NAME>", project_name="<ALLOCATION>"
14+
),
15+
)
16+
)
17+
18+
executor.submit(job)
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from psij import Job, JobSpec, JobExecutor
2+
from executor_test_params import ExecutorTestParams
3+
4+
5+
def test_getting_started_single_job(execparams: ExecutorTestParams) -> None:
6+
ex = JobExecutor.get_instance(execparams.executor)
7+
job = Job(JobSpec(executable='/bin/date', arguments=['-utc', '--debug']))
8+
ex.submit(job)
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
import time
2+
from psij import Job, JobSpec, JobExecutor
3+
from executor_test_params import ExecutorTestParams
4+
5+
count = 10
6+
7+
8+
def test_getting_started_scheduling_info(execparams: ExecutorTestParams) -> None:
9+
def callback(job, status):
10+
global count
11+
12+
if status.final:
13+
print(f"Job {job} completed with status {status}")
14+
count -= 1
15+
16+
ex = JobExecutor.get_instance(execparams.executor)
17+
ex.set_job_status_callback(callback)
18+
19+
for _ in range(count):
20+
job = Job(JobSpec(executable="/bin/date"))
21+
ex.submit(job)
22+
23+
while count > 0:
24+
time.sleep(0.01)

0 commit comments

Comments
 (0)