Tests

I made some tests while working on #15. They're quite rough but they helped me so I though you may want to keep them for the future. They're up to date with the latest #15 changes.