Update test.py

Added test
Merge pull request #1 from ams-bjones/patch-1

Update test.py