updated tests.py.