pep8 fixes for tests