pep8 fixes for unit tests