Replace doctests with unittest

For now we test the same as the old doctests