Follow PEP-8 style in unit tests as much as I can