PEP8 fixes for some tests

Makes it a bit better, but not 100% compliant, just like unittest..