Update tests.py