Update test.py
Merge pull request #3 from onemoreline/master

Update test.py