Add doctests to the tests