Added a python test