Further Test Coverage

[atymic]

There's currently no test coverage, so it would be a good to add some tests :)
I plan to do this soon, but if anyone would like to help I'd love a PR!

[atymic]

I've added some tests to cover the generator, but it could still do with more. Also needs test for the command as well.