Moved doctest to unittest