Update tests.py

Dude, that's it.