update tests.py