test.yaml

name: test

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

jobs:
  testing:

    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
         python-version: ["3.8", "3.9", "3.10"]
         backend: ["numpy", "numba"]
         os: ["ubuntu-latest"]

    steps:
    - uses: actions/checkout@v4
    
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5.0.0
      with:
        python-version: ${{ matrix.python-version }}
        
    - name: Install dependencies
      run: |
        python -m pip install pytest
        python -m pip install .

    - name: Install dependencies for backend ${{ matrix.backend }}
      run: |
        if [[ ${{ matrix.backend }} == "jax" ]]; then
          python -m pip install jax jaxlib
        elif [[ ${{ matrix.backend }} == "numba" ]]; then
          python -m pip install numba
        else
          python -m pip install numpy
        fi
    - name: Test with pytest
      run: |
        HS_BACKEND=${{ matrix.backend }} pytest