Cleanup pep8 violations in test_correctmap.py