Add doctest

[pseudo-rnd-thoughts]

In fixing an issue I realised that some of the docstrings did not work This adds doctest which runs in the tests and checks that if they contain work, it will run