def _test_dynamic_dummy_nodes(remote_command_executor, max_queue_size):
logging.info("Testing dummy nodes are automatically reconfigured based on actual compute nodes")
_assert_dummy_nodes(remote_command_executor, max_queue_size)
slurm_commands = SlurmCommands(remote_command_executor)
result = slurm_commands.submit_command("sleep 1", nodes=1)
job_id = slurm_commands.assert_job_submitted(result.stdout)
slurm_commands.wait_job_completed(job_id)
_assert_dummy_nodes(remote_command_executor, max_queue_size - 1)
def _test_dynamic_dummy_nodes(remote_command_executor, max_queue_size):
logging.info("Testing dummy nodes are automatically reconfigured based on actual compute nodes")
_assert_dummy_nodes(remote_command_executor, max_queue_size)
slurm_commands = SlurmCommands(remote_command_executor)
result = slurm_commands.submit_command("sleep 1", nodes=1)
job_id = slurm_commands.assert_job_submitted(result.stdout)
slurm_commands.wait_job_completed(job_id)
_assert_dummy_nodes(remote_command_executor, max_queue_size - 1)
def _test_dynamic_dummy_nodes(remote_command_executor,
region,
asg_name,
max_queue_size,
slots=4,
gpus=0):
logging.info(
"Testing dummy nodes are automatically reconfigured based on actual compute nodes"
)
slurm_commands = SlurmCommands(remote_command_executor)
# Assert initial conditions
_assert_asg_has_no_node(region, asg_name)
_assert_no_nodes_in_scheduler(slurm_commands)
_assert_dummy_nodes(remote_command_executor, max_queue_size, slots, gpus)
result = slurm_commands.submit_command("sleep 1", nodes=1)
job_id = slurm_commands.assert_job_submitted(result.stdout)
slurm_commands.wait_job_completed(job_id)
_assert_dummy_nodes(remote_command_executor, max_queue_size - 1, slots,
gpus)