updated tests.py