tests.yml

name: Tests

on:
  push:
    branches:
      - main
  pull_request:

jobs:
  tests:
    name: ${{ matrix.os }}-${{ matrix.jax-version }}
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest, macos-latest]
        jax-version: ["jax[cpu]"]
        include:
          - os: ubuntu-latest
            jax-version: "'jax[cpu]==0.4.20' 'numpy<2.0'"

    steps:
      - uses: actions/checkout@v4
        with:
          submodules: true
          fetch-depth: 0

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: "3.9"

      - name: Install fftw on ubuntu
        if: ${{ matrix.os == 'ubuntu-latest' }}
        run: sudo apt-get install -y libfftw3-dev

      - name: Install fftw on macos
        if: ${{ matrix.os == 'macos-latest' }}
        run: brew install fftw

      - name: Build
        run: |
          python -m pip install -U pip
          python -m pip install -U ${{ matrix.jax-version }}
          python -m pip install -v .[test]

      - name: Run tests
        run: python -m pytest -v tests