pep8 fixes for test