tests

Explicitly wait until computations finish in the verbose test. Should be NFC since currently `jax.jit` is sync on CPU. #1531

	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/[email protected]
	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

Provide feedback