PEP8 unittest changes