tests: Add python script tests