pep8 fixes in tests