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.))