Use doctests instead of separate tests