Revert "Update test.py"

This reverts commit f68ff42d0ce1b2d64e545aa2b92b69248ffd9422.