Add unit test for python version