JaxPM/.github/workflows/tests.yml

name: Tests

on:
  push:
    branches:
      - main
  pull_request:
    branches:
      - main

jobs:
  run_tests:

    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: ["3.10" , "3.11" , "3.12"]

    steps:
    - name: Checkout Source
      uses: actions/checkout@v2.3.1

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

    - name: Install dependencies
      run: |
        sudo apt-get install -y libopenmpi-dev
        python -m pip install --upgrade pip
        pip install numpy==2.2.6 setuptools cython wheel mpi4py
        pip install -r requirements-test.txt  --no-build-isolation
        pip install jax
        pip install pytest>=8.0.0
        pip install diffrax
        pip install .

    - name: Run Single Device Tests
      run: |
        cd tests
        pytest -v -m "not distributed"
    - name: Run Distributed tests
      run: |
        pytest -v -m distributed