Add some tests from the documentation
Merge pull request #11 from andyatkinson/add_tests

Add some tests from the documentation