Update tests.py

no fucking idea