edited tests.py for further clarity.