Include Sphinx builds into CI tests.

README.md

@@ -1,6 +1,8 @@
 # Chex
 
 ![CI status](https://github.com/deepmind/chex/workflows/ci/badge.svg)
+![docs](https://readthedocs.org/projects/chex/badge/?version=latest)
+![pypi](https://img.shields.io/pypi/v/chex)
 
 Chex is a library of utilities for helping to write reliable JAX code.
 

docs/Makefile

@@ -2,7 +2,7 @@
 #
 
 # You can set these variables from the command line.
-SPHINXOPTS    =
+SPHINXOPTS    = "-W"
 SPHINXBUILD   = sphinx-build
 SOURCEDIR     = .
 BUILDDIR      = _build

test.sh

@@ -26,7 +26,6 @@ python --version
 pip install --upgrade pip setuptools wheel
 pip install flake8 pytest-xdist pytype pylint pylint-exit
 pip install -r requirements/requirements.txt
-pip install -r requirements/requirements-test.txt
 
 # Lint with flake8.
 flake8 `find chex -name '*.py' | xargs` --count --select=E9,F63,F7,F82,E225,E251 --show-source --statistics
@@ -48,6 +47,7 @@ pytype `find chex/_src/ -name "*py" | xargs` -k
 
 # Run tests using pytest.
 # Change directory to avoid importing the package from repo root.
+pip install -r requirements/requirements-test.txt
 mkdir _testing && cd _testing
 
 # Main tests.
@@ -57,6 +57,11 @@ pytest -n "$(grep -c ^processor /proc/cpuinfo)" --pyargs chex -k "not fake_set_n
 pytest -n "$(grep -c ^processor /proc/cpuinfo)" --pyargs chex -k "fake_set_n_cpu_devices_test"
 cd ..
 
+# Build Sphinx docs.
+pip install -r requirements/requirements-docs.txt
+cd docs && make html
+cd ..
+
 set +u
 deactivate
 echo "All tests passed. Congrats!"