Update test.py

Feature complete.