updated tests.py a bit