Slight refactor

1. Makes it obvious that the unittests all use the same data by defining a setUp function in the super class.
2. Removing redundant import from __init__.py

Relates to #1076