edit test.py code, ref #27