Using doctest to define test cases