tests.yml

name: tests

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

jobs:
  build-and-test:
    name: "Python ${{ matrix.python-version }} on ${{ matrix.os }}"
    runs-on: "${{ matrix.os }}"

    strategy:
      matrix:
        python-version: ["3.9", "3.10", "3.11"]
        os: [ubuntu-latest]

    steps:
      - uses: "actions/checkout@v3"
      - uses: "actions/setup-python@v4"
        with:
            python-version: "${{ matrix.python-version }}"
            cache: 'pip'
      - name: Install dependencies
        run: |
          set -xe
          pip install --upgrade pip setuptools wheel
          pip install -r requirements.txt
          pip install -r requirements_test.txt
        shell: bash
      - name: Build
        run: |
          set -xe
          python -VV
          python setup.py install
        shell: bash
      - name: Run tests
        timeout-minutes: 60
        run: |
          set -xe
          python -VV
          python -c "import jax; print('jax', jax.__version__)"
          python -c "import jaxlib; print('jaxlib', jaxlib.__version__)"
          pytest tests
        shell: bash


  build-and-test-docs:
    name: "Build documentation"
    runs-on: ubuntu-latest
    steps:
      - name: Cancel previous
        uses: styfle/cancel-workflow-action@0.11.0
        with:
          access_token: ${{ github.token }}
        if: ${{github.ref != 'refs/heads/main'}}
      - uses: "actions/checkout@v3"
      - name: Set up Python 3.11
        uses: "actions/setup-python@v4"
        with:
            python-version: 3.11
            cache: 'pip'
      - name: Install dependencies
        run: |
          set -xe
          pip install --upgrade pip setuptools wheel
          pip install -r docs/requirements.txt
      - name: Build documentation
        run: |
          set -xe
          python -VV
          cd docs && make clean && make html