Split up tests.py