updated python tests