PEP8 whitespace cleanup in test.py