def test_nifti(): dm = DataMatrix(length=2) dm.n = NiftiColumn dm.n[0] = nib.Nifti2Image(np.array([[[0, 0], [1, 1]], [[-1, -1], [1, 1]]]), None) dm.n[1] = nib.Nifti2Image( np.array([[[1, 1], [0, 0]], [[np.nan, 1], [0, 0]]]), None) m = dm.n.mean.get_data() assert np.all(m == np.array([[[.5, .5], [.5, .5]], [[-1, 0], [.5, .5]]]))
def test_normalize_time(): dm = DataMatrix(length=2) dm.s = SeriesColumn(depth=2) dm.s[0] = 1, 2 dm.s[1] = np.nan, 3 dm.t = SeriesColumn(depth=2) dm.t[0] = 0, 3 dm.t[1] = 1, 2 dm.n = series.normalize_time(dm.s, dm.t) check_series(dm.n, [[1, np.nan, np.nan, 2], [np.nan, np.nan, 3, np.nan]])