def test_two_maps_masked(self):
maps2 = Maps(self.array2,
Ni=self.Ni,
Nj=self.Nj,
Nk=self.Nk,
mask=self.mask)
self.assertTrue(
np.array_equal(maps2.sum(axis=0, keepdims=True),
self.expected_masked2))
self.assertTrue(np.array_equal(maps2.sum(), 20.))
def test_one_map_masked(self):
maps = Maps(self.array,
Ni=self.Ni,
Nj=self.Nj,
Nk=self.Nk,
mask=self.mask)
self.assertTrue(
np.array_equal(maps.sum(axis=0, keepdims=True),
self.expected_masked1))
self.assertTrue(np.array_equal(maps.sum(), 8.))