def test_one_map_bis_template(self):
"""Test creating one map from template."""
maps = Maps.random(self.size1_bis, template=template)
arr = maps.to_array()
self.assertEqual(np.sum(arr), self.size1)
self.assertEqual(maps.n_m, 1)
self.assertEqual(maps.n_v, Ni*Nj*Nk)
def simulate_maps(coords, sigma, size, random_state=None):
"""
Create maps the same noised activation point
Args:
coords (tuple): Tuple of size 3 storing x y z coordinates.
sigma (float): Standard deviation of the gaussian kernel.
N (int): Number of maps to generate.
Returns:
(Maps): Instance of Maps object containing generated maps.
"""
x, y, z = coords
Ni, Nj, Nk = template.shape
p = Maps.zeros(template=template)
p.set_coord(0, x, y, z, 1)
p.smooth(sigma=sigma, inplace=True)
# plotting.plot_glass_brain(p.to_img())
# plt.show()
# rand_maps = Maps.zeros(Ni*Nj*Nk, template=template)
# rand_maps.randomize(size=10*np.ones(N).astype(int), p=p, inplace=True)
# return rand_maps
return Maps.random(size=size,
p=p,
random_state=random_state,
template=template)
def test_two_maps_array_manual(self):
"""Test creating several maps from Ni Nj Nk."""
maps = Maps.random(self.size3, Ni=Ni, Nj=Nj, Nk=Nk)
arr = maps.to_array()
self.assertEqual(np.sum(arr), np.sum(self.size3))
self.assertEqual(maps.n_m, 2)
self.assertEqual(maps.n_v, Ni*Nj*Nk)
def test_one_map_manual(self):
"""Test creating one map from Ni Nj Nk."""
maps = Maps.random(self.size1, Ni=Ni, Nj=Nj, Nk=Nk)
arr = maps.to_array()
self.assertEqual(np.sum(arr), self.size1)
self.assertEqual(maps.n_m, 1)
self.assertEqual(maps.n_v, Ni*Nj*Nk)
def test_two_maps_array_template(self):
"""Test creating several maps from template."""
maps = Maps.random(self.size3, template=template)
arr = maps.to_array()
self.assertEqual(np.sum(arr), np.sum(self.size3))
self.assertEqual(maps.n_m, 2)
self.assertEqual(maps.n_v, Ni*Nj*Nk)
def test_two_maps_template(self):
"""Test creating several maps from template."""
maps = Maps.random(self.size2, template=template)
arr = maps.to_array()
n_peaks, n_maps = self.size2
self.assertEqual(np.sum(arr), n_peaks)
self.assertEqual(maps.n_m, n_maps)
self.assertEqual(maps.n_v, Ni*Nj*Nk)
def random_maps(draw, min_maps=1):
Ni = draw(strats.integers(min_value=1, max_value=max_box_width))
Nj = draw(strats.integers(min_value=1, max_value=max_box_width))
Nk = draw(strats.integers(min_value=1, max_value=max_box_width))
n_maps = draw(strats.integers(min_value=min_maps, max_value=max_maps))
n_peaks = draw(strats.integers(min_value=0, max_value=max_peaks))
return Maps.random(Ni, Nj, Nk, n_peaks, n_maps)
def test_no_kwargs(self):
"""Test without keywords."""
with self.assertRaises(TypeError):
Maps.random(size=None)
def test_random_seed(self):
"""Test if obtain same results with same seeds."""
maps1 = Maps.random(self.size3, Ni=Ni, Nj=Nj, Nk=Nk, random_state=0)
maps2 = Maps.random(self.size3, Ni=Ni, Nj=Nj, Nk=Nk, random_state=0)
self.assertTrue(np.array_equal(maps1.to_array(), maps2.to_array()))