updated test.py
Merge pull request #1 from smeeeeee/feature1

updated test.py, input from feature1