Import tests use TestHelper

Removes duplicate code, increases speed and makes some files clake8 clean.