pep8 clean up for tests

pep8 clean up for tests