Update test.py

Better documentation