add python test code