add the python test