Add staging environment option for tests.py