Revert "Update test.py"
Merge pull request #3 from vidyar/revert-2-testPR

Revert "Update test.py"