Revert "change test.py"

This reverts commit 6f82034b784411b5470b71ba715c77f35b6cb6c3.
Revert "Revert "change test.py""

This reverts commit d4c1ebd51ae82616c977b123f555a69becf0c01e.