Apply PEP8 to tests.py