[wrt] Fix issue for test.py
Merge pull request #794 from kangxu/kangxu/wrt/pyauto/update

[wrt] Fix issue for test.py