fix test to conform pep8