doctest code snippets in python

use the doctest module for testing in python