Added python tests