Update test.py

changed my comment!