Add unit tests

I recommend py.test. Also document links to data for testing.