Add more tests

[Jvanrhijn]

Test coverage is currently pitiful. Each public unit of code should have at least one tests. Edge cases can be added in as the code is developed.