More unit tests

[robertmartin8]

Although I have written some unit tests, right now they really only check the datasets and some of the helper functions.

I would much appreciate any help in testing this project's core functionality.