ENH: Better tests

Ready for review/merge.

Closes #179.
Yep that output looks good.