Writing unit tests with doctest