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.max(axis=0), self.expected_masked2))
self.assertTrue(np.array_equal(maps2.max(), 7.))
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.max(axis=0),
self.expected_masked1))
self.assertTrue(np.array_equal(maps.max(), 5.))