PEP fixes for the test_utils file