Added tests for env.py as well