Fix pep8 in test cases