test.py was a bit out of date. Including current copy