Add script for testing Python API