run_tests.yml

name: Run tests

on: [pull_request, push]

jobs:
  run-tests:
    strategy:
      matrix:
        python-version: [ "3.9", "3.10", "3.11" ]
        os: [ ubuntu-latest ]
      fail-fast: false
    runs-on: ${{ matrix.os }}
    steps:
      - name: Checkout code
        uses: actions/checkout@v2

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

      - name: Checks with pre-commit
        uses: pre-commit/action@v2.0.3

      - name: Test with pytest
        run: |
          python -m pip install jaxlib==4.11
          python -m pip install .[dev]
          python -m pytest --runslow --durations=0