PEP 8 conformance for test script