Add some test setup

Closes #1