some updates on python test script