def test_allowed_template_mask_atlas(self):
maps = Maps(df_ex,
template=template,
groupby_col=groupby_col,
mask=gray_mask,
atlas=atlas)
self.assertTrue(maps._has_mask())
self.assertTrue(maps._has_atlas())
def test_allowed_manual_mask(self):
maps = Maps(df_ex,
Ni=Ni,
Nj=Nj,
Nk=Nk,
affine=affine,
groupby_col=groupby_col,
mask=gray_mask)
self.assertTrue(maps._has_mask())
def test_allowed_manual(self):
maps = Maps(df_ex,
Ni=Ni,
Nj=Nj,
Nk=Nk,
affine=affine,
groupby_col=groupby_col)
self.assertEqual(maps.n_maps, 2)
self.assertFalse(maps._has_mask())
self.assertFalse(maps._has_atlas())
def test_mask(self):
maps = Maps(template=template, mask=gray_mask)
self.assertTrue(maps._has_mask())
def test_no_mask(self):
maps = Maps(np.array([[0.]]), Ni=1, Nj=1, Nk=1)
self.assertFalse(maps._has_mask())
def test_allowed_template(self):
maps = Maps(df_ex, template=template, groupby_col=groupby_col)
self.assertEqual(maps.n_maps, 2)
self.assertFalse(maps._has_mask())
self.assertFalse(maps._has_atlas())