test_utils_nib.py 文件源码

def test_remove_nan():
    data_ts = np.array([[[0, 1, 2, 3],
                         [4, 5, np.nan, 7],
                         [8, 9, 10, np.nan]],

                        [[12, 13, 14, 15],
                         [16, np.nan, 18, 19],
                         [20, 21, 22, 23]]])
    im = nib.Nifti1Image(data_ts, np.eye(4))
    im_no_nan = remove_nan(im)

    data_no = np.array([[[0, 1, 2, 3],
                         [4, 5, 0, 7],
                         [8, 9, 10, 0]],

                        [[12, 13, 14, 15],
                         [16, 0, 18, 19],
                         [20, 21, 22, 23]]])

    assert_array_equal(im_no_nan.get_data(), data_no)