Reorganize tests into package

Related to #16