Doc tests.

Add python doc tests.