def test_braindata(): vol = np.random.randn(*volshape) tf = tempfile.TemporaryFile(suffix='.png') mask = db.get_mask(subj, xfmname, "thick") data = dataset.Volume(vol, subj, xfmname, cmap='RdBu_r', vmin=0, vmax=1) # quickflat.make_png(tf, data) mdata = data.masked['thick'] assert len(mdata.data) == mask.sum() assert np.allclose(mdata.volume[:, mask], mdata.data)
def test_findmask(): vol = np.random.rand(10, *volshape) mask = db.get_mask(subj, xfmname, "thin") ds = dataset.Volume(vol[:, mask], subj, xfmname) assert np.allclose(ds.volume[:, mask], vol[:, mask]) return ds