Update py_test.py