FIX #328: silence cling error in test_evaluate.py

Author: ndawe

root_numpy/tests/__init__.py

@@ -1,29 +1,82 @@
 import os
+import sys
 import warnings
 import ROOT
-import root_numpy as rnp
 from numpy.random import RandomState
+import tempfile
+from contextlib import contextmanager
+import root_numpy as rnp
+from root_numpy.testdata import get_filepath
+import threading
+
+LOCK = threading.RLock()
 
 ROOT.gErrorIgnoreLevel = ROOT.kFatal
 warnings.filterwarnings('ignore', category=DeprecationWarning)
 warnings.filterwarnings('ignore', category=rnp.RootNumpyUnconvertibleWarning)
 RNG = RandomState(42)
 
-from root_numpy.testdata import get_filepath
 
 def load(data):
     if isinstance(data, list):
         return [get_filepath(x) for x in data]
     return get_filepath(data)
 
-import tempfile
-from contextlib import contextmanager
 
 @contextmanager
 def temp():
     tmp_fd, tmp_path = tempfile.mkstemp(suffix='.root')
     tmp_root = ROOT.TFile.Open(tmp_path, 'recreate')
-    yield tmp_root
-    tmp_root.Close()
-    os.close(tmp_fd)
-    os.remove(tmp_path)
+    try:
+        yield tmp_root
+    finally:
+        tmp_root.Close()
+        os.close(tmp_fd)
+        os.remove(tmp_path)
+
+
+@contextmanager
+def silence_sout():
+    LOCK.acquire()
+    sys.__stdout__.flush()
+    origstdout = sys.__stdout__
+    oldstdout_fno = os.dup(sys.__stdout__.fileno())
+    devnull = os.open(os.devnull, os.O_WRONLY)
+    newstdout = os.dup(1)
+    os.dup2(devnull, 1)
+    os.close(devnull)
+    sys.__stdout__ = os.fdopen(newstdout, 'w')
+    try:
+        yield
+    finally:
+        sys.__stdout__ = origstdout
+        sys.__stdout__.flush()
+        os.dup2(oldstdout_fno, 1)
+        LOCK.release()
+
+
+@contextmanager
+def silence_serr():
+    LOCK.acquire()
+    sys.__stderr__.flush()
+    origstderr = sys.__stderr__
+    oldstderr_fno = os.dup(sys.__stderr__.fileno())
+    devnull = os.open(os.devnull, os.O_WRONLY)
+    newstderr = os.dup(2)
+    os.dup2(devnull, 2)
+    os.close(devnull)
+    sys.__stderr__ = os.fdopen(newstderr, 'w')
+    try:
+        yield
+    finally:
+        sys.__stderr__ = origstderr
+        sys.__stderr__.flush()
+        os.dup2(oldstderr_fno, 2)
+        LOCK.release()
+
+
+@contextmanager
+def silence():
+    with silence_sout():
+        with silence_serr():
+            yield

root_numpy/tests/test_evaluate.py

@@ -2,7 +2,7 @@
 import root_numpy as rnp
 from numpy.testing import assert_array_equal
 from nose.tools import assert_raises
-from . import RNG
+from . import RNG, silence_serr
 
 
 def test_evaluate_func():
@@ -39,7 +39,9 @@ def test_evaluate_func():
     assert_raises(ValueError, rnp.evaluate, f3, arr_1d)
     assert_raises(ValueError, rnp.evaluate, f3, arr_2d)
 
-    assert_raises(ValueError, rnp.evaluate, "f", arr_1d)
+    with silence_serr():  # silence cling error
+        assert_raises(ValueError, rnp.evaluate, "f", arr_1d)
+
     assert_raises(ValueError, rnp.evaluate, "x*y", arr_1d)
     assert_raises(ValueError, rnp.evaluate, "x", arr_2d)
     assert_raises(ValueError, rnp.evaluate, "x*y", arr_3d)