Add python unit tests.