Move the test environment initialization to tests.py