Update test.py

Added a line.