Add python tests