Add tests

[wip-abramson]

[wip-abramson]

@lohanspies this isn't a priority, but as we add new functionality we should aim to cover the functionality in the demo folder at minimum. These could then form the basis of a test suite when we get round to it.

[lohanspies]

+1 will need to go through a upskill process to create unit tests etc. ;-)