| ############################################################################ |
| # Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. |
| ############################################################################ |
| """Test sbatch --parsable output format.""" |
| |
| import re |
| |
| import pytest |
| |
| import atf |
| |
| |
| @pytest.fixture(scope="module", autouse=True) |
| def setup(): |
| atf.require_slurm_running() |
| |
| |
| def test_sbatch_parsable(): |
| """sbatch --parsable writes only the job id (or jobid;cluster) to stdout.""" |
| result = atf.run_command( |
| "sbatch --parsable -t1 --job-name=test_sbatch_parsable -o /dev/null --wrap 'true'", |
| fatal=True, |
| ) |
| |
| match = re.match(r"^(\d+)(?:;.+)?\s*$", result["stdout"]) |
| assert match, ( |
| f"expected bare jobid (or jobid;cluster) on stdout, " |
| f"got: {result['stdout']!r}" |
| ) |
| atf.properties["submitted-jobs"].append(int(match.group(1))) |
