Add unit test support for Python