Refactor ai-test-suite (python script).