fix pep8 in tests

Change-Id: Ia332171972d1c54d73ff018d0c996c7e4d513125