Update tests.py
Merge pull request #2 from RicterZ/patch-2

Update tests.py
fixed merge problem