make sure we always reset backends, even if we don't set them explicitly

scikit-hep · Sep 19, 2018 · f6a4228 · f6a4228
1 parent de46ec7
commit f6a4228
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -2,6 +2,11 @@
 import pyhf
 import tensorflow as tf
 
+@pytest.fixture(scope='function', autouse=True)
+def reset_backend():
+    yield reset_backend
+    pyhf.set_backend(pyhf.default_backend)
+
 @pytest.fixture(scope='function', params=[
                              (pyhf.tensor.numpy_backend(),),
                              (pyhf.tensor.tensorflow_backend(session=tf.Session()),),
@@ -50,5 +55,3 @@ def backend(request):
         pyhf.tensorlib.session = tf.Session()
 
     yield request.param
-
-    pyhf.set_backend(pyhf.default_backend)