Fix pep8 violations in test.py