More unit test coverage.

I've noticed other people are starting to add their own unit tests which may potentially clash with tests I've been adding. So I'm going to open PR's more regularly to ensure that people are working with the latest version of the tests.

Cheers!