Added some simple python tests