add python tests