def test_get_radius_vs_angle(self):
test_data = mdtd.MakeTestData(
101, 121, blur=False, downscale=False, default=False
)
x, y, r0, r1 = 51, 55, 21, 35
test_data.add_ring(x, y, r0)
test_data.add_ring(x, y, r1)
s = test_data.signal
s.axes_manager.signal_axes[0].offset = -x
s.axes_manager.signal_axes[1].offset = -y
s_ar0 = ra.get_radius_vs_angle(
s, radial_signal_span=(15, 25), show_progressbar=False
)
s_ar1 = ra.get_radius_vs_angle(
s, radial_signal_span=(28, 40), show_progressbar=False
)
np.testing.assert_allclose(s_ar0.data, r0, 3)
np.testing.assert_allclose(s_ar1.data, r1, 3)
def test_find_parameters(self):
axis1, axis2 = self.axis1, self.axis2
s = self.s
s_ra = ra.get_radius_vs_angle(s, (30.0, 80.0), angleN=20)
x, y = ra._get_xy_points_from_radius_angle_plot(s_ra)
ellipse_parameters = ra._fit_ellipse_to_xy_points(x, y)
xC, yC, semi_len0, semi_len1, rot, eccen = ra._get_ellipse_parameters(
ellipse_parameters)
assert approx(xC, abs=0.1) == 0.0
assert approx(yC, abs=0.1) == 0.0
assert approx(semi_len0, abs=1.0) == axis2
assert approx(semi_len1, abs=1.0) == axis1
def test_get_signal_with_markers(self):
s = self.s
s_ra = ra.get_radius_vs_angle(s, (30.0, 80.0), angleN=20)
x, y = ra._get_xy_points_from_radius_angle_plot(s_ra)
ellipse_parameters = ra._fit_ellipse_to_xy_points(x, y)
ra._get_marker_list(ellipse_parameters, x_list=x, y_list=y)
