Add a python test