PEP8
Simplify test code