PEP8 in tests

PEP8 in tests