예제 #1
0
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]])