Update test.py

current copy. Added by hand due to remote push issues