Adjust tests with a slight abstraction

Moved some test code to the new package, since it will be required for the package-internal test later.

Closes #36 (closed)