Update test.py
Merge pull request #4 from trumf/patch-3

Update test.py