reset disable flag again after test

brain-score · Jun 21, 2019 · d798431 · d798431
1 parent 033fd10
commit d798431
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test___init__.py b/tests/test___init__.py
@@ -197,6 +197,7 @@ def __repr__(self):
             assert c.f(1) == 1
 
     def test_disable_store(self):
+        previous_disable_value = os.getenv('RESULTCACHING_DISABLE', '0')
         with tempfile.TemporaryDirectory() as storage_dir:
             os.environ['RESULTCACHING_HOME'] = storage_dir
             os.environ['RESULTCACHING_DISABLE'] = '1'
@@ -215,6 +216,8 @@ def func(x):
             assert func(1) == 1
             assert function_calls == 2
 
+        os.environ['RESULTCACHING_DISABLE'] = previous_disable_value
+
 
 class TestCache:
     def test(self):