update python testdriver to use latest api