Add test cases

Did some basic test cases so there's less risk of breaking something obvious when making changes. Did not bump package version as there's no need for users to care about this.