PEP8 fixes for test code

PEP8 fixes for test code, needed as a result of renaming the base package.