TEST: Update NBS import test

Author: effigies

nipype/interfaces/cmtk/tests/test_nbs.py

@@ -4,6 +4,7 @@
 import numpy as np
 import networkx as nx
 import pytest
+from nibabel.testing import clear_and_catch_warnings
 
 have_cv = True
 try:
@@ -34,14 +35,17 @@ def test_importerror(creating_graphs, tmpdir):
     group1 = graphlist[:3]
     group2 = graphlist[3:]
 
-    nbs = NetworkBasedStatistic()
+    with clear_and_catch_warnings() as w:
+        nbs = NetworkBasedStatistic()
+        assert len(w) == 1
+        assert str(w.message).startswith('Unable to import cviewer')
     nbs.inputs.in_group1 = group1
     nbs.inputs.in_group2 = group2
     nbs.inputs.edge_key = "weight"
 
     with pytest.raises(ImportError) as e:
         nbs.run()
-    assert "cviewer library is not available" == str(e.value)
+    assert "No module named 'cviewer'" == str(e.value)
 
 
 @pytest.mark.skipif(not have_cv, reason="cviewer has to be available")