Add python unit test, and fix setup as needed.