modified test.py
Merge pull request #2 from prajnamort/master

modified test.py