Reset environment variables after test

mila-iqia · Jan 15, 2015 · 7d0fd4b · 7d0fd4b
1 parent a6169bf
commit 7d0fd4b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_config_parser.py b/tests/test_config_parser.py
@@ -6,14 +6,14 @@
 
 
 def test_config_parser():
+    _environ = dict(os.environ)
     os.environ['BLOCKS_CONFIG'] = os.path.join(os.getcwd(),
                                                '.test_blocksrc')
-    assert not os.path.exists(os.environ['BLOCKS_CONFIG'])
     with open(os.environ['BLOCKS_CONFIG'], 'w') as f:
         f.write('data_path: yaml_path')
+    if 'BLOCKS_DATA_PATH' in os.environ:
+        del os.environ['BLOCKS_DATA_PATH']
     try:
-        if 'BLOCKS_DATA_PATH' in os.environ:
-            del os.environ['BLOCKS_DATA_PATH']
         config = Configuration()
         config.add_config('data_path', str, env_var='BLOCKS_DATA_PATH')
         config.add_config('config_with_default', int, default='1',
@@ -31,3 +31,5 @@ def test_config_parser():
                       'config_without_default')
     finally:
         os.remove(os.environ['BLOCKS_CONFIG'])
+        os.environ.clear()
+        os.environ.update(_environ)