Adding tests so we can tell that everything is working as it should

Signed-off-by: Andrew Lowther <andrew.lowther@mademedia.co.uk>