Add unit tests

[weakphish]

I'm currently working on the other issue I opened awhile back, and I noticed a lack of unit tests. Would it be okay if after I completed the current feature (exporting) to add in unit tests?

[nanohard]

Sure, have a go!