More pep8 violations

* tests/test_codegen.py: hadn't run flake8 here before, fixing