def test_correct_radius(self): s0, s1, r0, r1 = self.s0, self.s1, self.r0, self.r1 s = ra.get_angle_image_comparison(s0, s1, angleN=2) assert s.axes_manager.signal_shape == (100, 100) # Check that radius is correct by getting a line profile s_top = s.isig[0.0, :0.0] s_bot = s.isig[0.0, 0.0:] argmax0 = s_top.data.argmax() argmax1 = s_bot.data.argmax() assert abs(s_top.axes_manager[0].index2value(argmax0)) == r0 assert abs(s_bot.axes_manager[0].index2value(argmax1)) == r1
def test_mask(self): s0, s1 = self.s0, self.s1 s_no_mask = ra.get_angle_image_comparison(s0, s1) s_mask = ra.get_angle_image_comparison(s0, s1, mask_radius=40) assert s_no_mask.data.sum() != 0.0 assert s_mask.data.sum() == 0.0
def test_different_signal_size(self): s0 = mdtd.MakeTestData(100, 100).signal s1 = mdtd.MakeTestData(100, 150).signal with pytest.raises(ValueError): ra.get_angle_image_comparison(s0, s1)
def test_different_angleN(self): s0, s1 = self.s0, self.s1 for i in range(1, 10): ra.get_angle_image_comparison(s0, s1, angleN=i)