Update tests.py

Updating format of tests.py.