CI: pytest setup fix

docker/Dockerfile.nvidia

@@ -86,7 +86,7 @@ RUN export NVARCH=$(ls -1 /opt/nvidia/hpc_sdk/Linux_x86_64/ | grep '\.' | head -
     ln -sf /opt/nvidia/hpc_sdk/Linux_x86_64/comm_libs/${CUDA_V}/nccl /opt/nvhpc/comm_libs/nccl
 
 # Starting nvhpc 23.5 and cuda 12.1, hpcx and openmpi are inside the cuda version folder, only the bin is in the comm_libs path
-RUN export CUDA_V=$(nvcc --version | sed -n 's/^.*release \([0-9]\+\.[0-9]\+\).*$/\1/p') && \
+RUN export CUDA_V=$(/opt/nvhpc/cuda/bin/nvcc --version | sed -n 's/^.*release \([0-9]\+\.[0-9]\+\).*$/\1/p') && \
     ls /opt/nvhpc/comm_libs/${CUDA_V}/hpcx/ &&\
     if [ -d /opt/nvhpc/comm_libs/${CUDA_V}/hpcx ]; then \
         rm -rf /opt/nvhpc/comm_libs/hpcx && rm -rf /opt/nvhpc/comm_libs/openmpi4 && \

examples/seismic/elastic/elastic_example.py

@@ -40,7 +40,7 @@ def test_elastic(dtype):
     assert np.isclose(norm(rec2), 0.627606, atol=1e-3, rtol=0)
 
 
-@pytest.mark.parametrize('shape', [(101,), (51, 51), (16, 16, 16)])
+@pytest.mark.parametrize('shape', [(51, 51), (16, 16, 16)])
 def test_elastic_stability(shape):
     spacing = tuple([20]*len(shape))
     _, _, _, [rec1, rec2, v, tau] = run(shape=shape, spacing=spacing, tn=20000.0, nbl=0)

examples/seismic/viscoelastic/viscoelastic_example.py

@@ -42,7 +42,7 @@ def test_viscoelastic(dtype):
     assert np.isclose(norm(rec2), 0.312461, atol=1e-3, rtol=0)
 
 
-@pytest.mark.parametrize('shape', [(101,), (51, 51), (16, 16, 16)])
+@pytest.mark.parametrize('shape', [(51, 51), (16, 16, 16)])
 def test_viscoelastic_stability(shape):
     spacing = tuple([20]*len(shape))
     _, _, _, [rec1, rec2, v, tau] = run(shape=shape, spacing=spacing, tn=20000.0, nbl=0)

pytest.ini

@@ -1,4 +1,6 @@
 [pytest]
+python_files = test_*.py *_example.py example_*.py
+addopts = --durations=20 --maxfail=5
 filterwarnings =
     ignore::pytest.PytestCacheWarning
     ignore::pytest.PytestReturnNotNoneWarning

setup.cfg

@@ -8,10 +8,6 @@ max-line-length = 90
 ignore = F403,E226,E731,E275,W503,F405,E722,E741,W504,W605
 exclude = .github
 
-[tool:pytest]
-python_files = test_*.py *_example.py example_*.py
-addopts = --durations=20 --maxfail=5
-
 [versioneer]
 VCS = git
 style = pep440

tests/test_mpi.py

@@ -221,6 +221,7 @@ def test_custom_topology(self):
         (256, ('*', '*', 2), (16, 8, 2)),
         (256, ('*', 32, 2), (4, 32, 2)),
     ])
+    @pytest.mark.parallel(mode=[2])
     def test_custom_topology_v2(self, comm_size, topology, dist_topology):
         dummy_comm = Bunch(size=comm_size)
         custom_topology = CustomTopology(topology, dummy_comm)