def test_mask_gaps_base_class(self):
det = Detector("Maxipix")
data = np.ones((1, 1))
mask = np.zeros((1, 1))
output = det._mask_gaps(data, mask)
self.assertTrue(np.all(np.isclose(output[0], data)))
self.assertTrue(np.all(np.isclose(output[1], mask)))
def test_mask_gaps_base_class_wrong_shape(self):
det = Detector("Maxipix")
data = np.ones((1, 1))
mask = np.zeros((1, 2))
with self.assertRaises(ValueError):
det._mask_gaps(data, mask)
def test_mask_gaps_base_class_wrong_type(self):
det = Detector("Maxipix")
data = 1
mask = np.zeros((1, 1))
with self.assertRaises(TypeError):
det._mask_gaps(data, mask)