pep8 clean up in tests