Documentation tweaks to tests.py