Make tests conform to PEP8