#14 - finished refactored a half-done doctest to py.test