Revert "Update test.py"

This reverts commit 72314ea859f44b210a8f8048d48e60f2a778a31f.