ci.yml

name: Tests

on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]

concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true

env:
  FORCE_COLOR: 3

jobs:
  test:
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: ["3.9", "3.10", "3.11", "3.12"]
        mpi-dist: ["openmpi", "mpich"]
        mpipy-version: ["~=3.0", "~=4.0"]
    steps:

    - uses: actions/checkout@v4

    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
      with:
        python-version: ${{ matrix.python-version }}

    - name: Install OpenMPI
      if: matrix.mpi-dist == 'openmpi'
      run: |
        sudo apt-get install -y -q openmpi-bin libopenmpi-dev
        mpiexec --version

    - name: Install MPICH
      if: matrix.mpi-dist == 'mpich'
      run: sudo apt-get install -y -q mpich libmpich-dev

    - name: Install dependencies
      run: |
        python -m pip install -U pip
        python -m pip install -e .[test]
        python -m pip install mpi4py${{ matrix.mpipy-version }}

    - name: Test MPI
      run: |
        mpiexec -n 2 python $PWD/tests/test_mpi.py
        mpiexec -n 2 python $PWD/tests/test_mpi_with_dill.py

    - name: Test package
      run: >-
        python -m pytest -ra --cov --cov-report=xml --cov-report=term --durations=20

    - name: Upload coverage report
      uses: codecov/codecov-action@v5.1.2