PEP8 cleanup In Tests