Group tests by cluster during distributed tests

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Apr 15, 2024 · f9f0a43 · f9f0a43
1 parent c1c45c4
commit f9f0a43
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,9 +41,10 @@ pytest-xdist = "^3.5.0"
 
 
 [tool.pytest.ini_options]
-addopts = "--doctest-modules"
+addopts = "--doctest-modules --dist loadgroup"
 markers = "--enable-internet: Allow some tests to run using real connections to the cluster."
 
+
 [tool.poetry-dynamic-versioning]
 enable = true
 vcs = "git"

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -257,19 +257,19 @@ def job_name(request: pytest.FixtureRequest) -> str | None:
     return getattr(request, "param", JOB_NAME)
 
 
-@pytest_asyncio.fixture
+@pytest_asyncio.fixture(scope="session")
 async def launches_job_fixture(login_node_v2: RemoteV2, job_name: str):
     jobs_before = await get_queued_milatools_job_ids(login_node_v2, job_name=job_name)
     if jobs_before:
-        logger.info(f"Jobs in squeue before test: {jobs_before}")
+        logger.debug(f"Jobs in squeue before tests: {jobs_before}")
     try:
         yield
     finally:
         jobs_after = await get_queued_milatools_job_ids(
             login_node_v2, job_name=job_name
         )
         if jobs_before:
-            logger.info(f"Jobs after test: {jobs_before}")
+            logger.debug(f"Jobs after tests: {jobs_before}")
 
         new_jobs = jobs_after - jobs_before
         if new_jobs: