Add doctest for Python

[Fokko]

I was going through the Python code, and noticed that not all examples are valid. For Python there is Doctest to automatically check the examples in the docstring: https://docs.python.org/3/library/doctest.html This will make sure that our examples are up to date, and that can do simple assertions in the docstring 👍🏻