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.n_peaks(), self.expected_masked2))
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.n_peaks(), self.expected_masked))