update test.py, better to read