modified tests.py