Add a python unittest