Fixing pep/formatting issues in tests

Fix pep errors. Cosmetics only. Thanks to @procrastinatio for the heads up.