def test_wait_any_started(): session_name = generate_random_string() js = JobSession(session_name) ja = js.run_bulk_jobs({'remote_command': '/bin/sleep', 'args': ['5']}, 1, 10, 5, 2) jl = ja.job_list jl_ids = list(map(lambda j: j.id, jl)) print('\nWaiting on start of any job with id from %s' % jl_ids) j = js.wait_any_started(jl) assert (j.id in jl_ids) print('Job %s started' % j)
def test_suspend_and_resume(): session_name = generate_random_string() js = JobSession(session_name) d = {'remote_command': '/bin/sleep', 'args': ['5'], 'output_path': '/dev/null', 'join_files': True} ja = js.run_bulk_jobs(d, 1, 10, 3, 2) j_list = ja.job_list js.wait_any_started(j_list) ja.suspend() suspended_jobs = [] for j in j_list: s, ss = j.get_state() if s == JobState.SUSPENDED: suspended_jobs.append(j) assert (len(suspended_jobs) > 0) print('\nSuspend job array: %s' % (ja)) ja.resume() for j in suspended_jobs: s, ss = j.get_state() assert (s != JobState.SUSPENDED) print('Resume job array: %s' % (ja))
# you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. ########################################################################### # ___INFO__MARK_END__ from drmaa2 import JobSession if __name__ == '__main__': js = JobSession('js-01') print('Created job session: %s' % js.name) d = {'remote_command': '/bin/sleep', 'args': ['10']} print('Running job array using dictionary: %s' % d) begin_index = 1 end_index = 10 step = 2 max_parallel = 2 ja = js.run_bulk_jobs(d, begin_index, end_index, step, max_parallel) print('Submitted job array: %s' % ja) print('Waiting for any job to start') jl = ja.job_list j = js.wait_any_started(jl) print('Job %s started' % j)