def test_unlist_relist_streamlines():
streamlines = [np.random.rand(10, 3), np.random.rand(20, 3), np.random.rand(5, 3)]
points, offsets = unlist_streamlines(streamlines)
assert_equal(offsets.dtype, np.dtype("i8"))
assert_equal(points.shape, (35, 3))
assert_equal(len(offsets), len(streamlines))
streamlines2 = relist_streamlines(points, offsets)
assert_equal(len(streamlines), len(streamlines2))
for i in range(len(streamlines)):
assert_array_equal(streamlines[i], streamlines2[i])
def test_unlist_relist_streamlines():
streamlines = [np.random.rand(10, 3),
np.random.rand(20, 3),
np.random.rand(5, 3)]
points, offsets = unlist_streamlines(streamlines)
assert_equal(offsets.dtype, np.dtype('i8'))
assert_equal(points.shape, (35, 3))
assert_equal(len(offsets), len(streamlines))
streamlines2 = relist_streamlines(points, offsets)
assert_equal(len(streamlines), len(streamlines2))
for i in range(len(streamlines)):
assert_array_equal(streamlines[i], streamlines2[i])
def test_efficient_bmd():
a = np.array([[1, 1, 1],
[2, 2, 2],
[3, 3, 3]])
streamlines = [a, a + 2, a + 4]
points, offsets = unlist_streamlines(streamlines)
points = points.astype(np.double)
points2 = points.copy()
D = np.zeros((len(offsets), len(offsets)), dtype='f8')
_bundle_minimum_distance_matrix(points, points2,
len(offsets), len(offsets),
a.shape[0], D)
assert_equal(np.sum(np.diag(D)), 0)
points2 += 2
_bundle_minimum_distance_matrix(points, points2,
len(offsets), len(offsets),
a.shape[0], D)
streamlines2 = relist_streamlines(points2, offsets)
D2 = distance_matrix_mdf(streamlines, streamlines2)
assert_array_almost_equal(D, D2)
cols = D2.shape[1]
rows = D2.shape[0]
dist = 0.25 * (np.sum(np.min(D2, axis=0)) / float(cols) +
np.sum(np.min(D2, axis=1)) / float(rows)) ** 2
dist2 = _bundle_minimum_distance(points, points2,
len(offsets), len(offsets),
a.shape[0])
assert_almost_equal(dist, dist2)
def two_cingulum_bundles():
fname = get_data('cb_2')
res = np.load(fname)
cb1 = relist_streamlines(res['points'], res['offsets'])
cb2 = relist_streamlines(res['points2'], res['offsets2'])
return cb1, cb2
def two_cingulum_bundles():
fname = get_data('cb_2')
res = np.load(fname)
cb1 = relist_streamlines(res['points'], res['offsets'])
cb2 = relist_streamlines(res['points2'], res['offsets2'])
return cb1, cb2
def two_cingulum_bundles():
fname = get_data("cb_2")
res = np.load(fname)
cb1 = relist_streamlines(res["points"], res["offsets"])
cb2 = relist_streamlines(res["points2"], res["offsets2"])
return cb1, cb2
