Add python tests.