Updated tests.py