add python test
modify to support python