tests for show_contrast

nipy/modalities/fmri/tests/test_dmtx.py

@@ -71,6 +71,23 @@ def test_show_dmtx():
     ax = DM.show()
     assert (ax is not None)
 
+    # test the colormap
+    ax = DM.show(cmap=matplotlib.pyplot.cm.gray)
+    assert (ax is not None)
+
+@dec.skipif(not have_mpl)
+def test_show_constrast():
+    # test that the show code indeed (formally) runs
+    frametimes = np.linspace(0, 127 * 1.,128)
+    DM = make_dmtx(frametimes, drift_model='polynomial', drift_order=3)
+    contrast = np.random.standard_normal((3, DM.matrix.shape[1]))
+    ax = DM.show_contrast(contrast)
+    assert (ax is not None)
+
+    # test the colormap
+    ax = DM.show_contrast(contrast, cmap=matplotlib.pyplot.cm.gray)
+    assert (ax is not None)
+
 
 def test_dmtx0():
     # Test design matrix creation when no paradigm is provided