def test_temps_env_var_fallback(): ''' Test that temps uses `gettempdir` when given no TEMPS_DIR env var ''' with environ_cm('TEMPS_DIR', ''): print 'test_pure_fallback' print temps._set_temps_dir() print tempfile.gettempdir() assert temps._set_temps_dir() == tempfile.gettempdir()
def test_temps_env_var_override(): ''' Test that temps uses the env var TEMPS_DIR when it is set. ''' d = '/temps_test_tmp_2' with environ_cm('TEMPS_DIR', d): print 'test_temps_env_var' print temps._set_temps_dir() print d assert temps._set_temps_dir() == d