PEP8 clean-up for tests.