def test_extra_resources(HelloTest, testsys_system): @test_util.custom_prefix('unittests/resources/checks') class MyTest(HelloTest): def __init__(self): super().__init__() self.name = type(self).__name__ self.executable = os.path.join('.', self.name) self.local = True @run_after('setup') def set_resources(self): test.extra_resources = { 'gpu': { 'num_gpus_per_node': 2 }, 'datawarp': { 'capacity': '100GB', 'stagein_src': test.stagedir } } test.job.options += ['--foo'] test = MyTest() partition = test_util.partition_by_name('gpu') environ = partition.environment('builtin') _run(test, partition, environ) expected_job_options = { '--gres=gpu:2', '#DW jobdw capacity=100GB', f'#DW stage_in source={test.stagedir}', '--foo' } assert expected_job_options == set(test.job.options)
def local_exec_ctx(generic_system): partition = test_util.partition_by_name('default') environ = test_util.environment_by_name('builtin', partition) yield partition, environ
def dummy_gpu_exec_ctx(testsys_system): partition = test_util.partition_by_name('gpu') environ = test_util.environment_by_name('builtin', partition) yield partition, environ