Fix PEP8 violations in tests