ENH: Add test dataset

WIP. Clearly needs to be fixed/updated, but it shows a gist of what I did.