Added some python tests