def test_nr(self):
nr0, nr1 = 5, 9
ellipse_array = np.random.randint(10, 100, size=(2, 3, 5))
marker_list0 = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, nr=nr0)
marker_list1 = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, nr=nr1)
assert len(marker_list0) == nr0
assert len(marker_list1) == nr1
def test_color(self):
color0, color1 = "blue", "green"
ellipse_array = np.random.randint(10, 100, size=(2, 3, 5))
marker_list0 = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, color=color0)
marker_list1 = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, color=color1)
for marker0, marker1 in zip(marker_list0, marker_list1):
assert marker0.marker_properties["color"] == color0
assert marker1.marker_properties["color"] == color1
def test_linestyle_linewidth(self):
linewidth0, linewidth1 = 12, 32
linestyle0, linestyle1 = "solid", "dashed"
ellipse_array = np.random.randint(10, 100, size=(2, 3, 5))
marker_list0 = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, linestyle=linestyle0, linewidth=linewidth0)
marker_list1 = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, linestyle=linestyle1, linewidth=linewidth1)
for marker0, marker1 in zip(marker_list0, marker_list1):
assert marker0.marker_properties["linestyle"] == linestyle0
assert marker1.marker_properties["linestyle"] == linestyle1
assert marker0.marker_properties["linewidth"] == linewidth0
assert marker1.marker_properties["linewidth"] == linewidth1
def test_correct_values(self):
xc_array = np.random.randint(120, 130, size=(2, 3))
yc_array = np.random.randint(130, 140, size=(2, 3))
sx_array = np.random.randint(70, 80, size=(2, 3))
sy_array = np.random.randint(89, 99, size=(2, 3))
ro_array = np.zeros((2, 3))
ellipse_array = np.empty((2, 3), dtype=np.object)
for iy, ix in np.ndindex(ellipse_array.shape):
xc, yc = xc_array[iy, ix], yc_array[iy, ix]
sx, sy = sx_array[iy, ix], sy_array[iy, ix]
ro = ro_array[iy, ix]
ellipse_array[iy, ix] = (yc, xc, sy, sx, ro)
nr = 4
marker_list = ret._get_ellipse_marker_list_from_ellipse_array(
ellipse_array, nr=nr
)
assert len(marker_list) == nr
for marker in marker_list:
assert marker.data["x1"][()].shape == xc_array.shape
m0, m1, m2, m3 = marker_list
assert_allclose(m0.data["x1"][()], xc_array)
assert_allclose(m0.data["y1"][()], yc_array + sy_array)
assert_allclose(m0.data["x2"][()], xc_array + sx_array)
assert_allclose(m0.data["y2"][()], yc_array)
assert_allclose(m1.data["x1"][()], xc_array + sx_array)
assert_allclose(m1.data["y1"][()], yc_array)
assert_allclose(m1.data["x2"][()], xc_array)
assert_allclose(m1.data["y2"][()], yc_array - sy_array)
assert_allclose(m2.data["x1"][()], xc_array)
assert_allclose(m2.data["y1"][()], yc_array - sy_array)
assert_allclose(m2.data["x2"][()], xc_array - sx_array)
assert_allclose(m2.data["y2"][()], yc_array)
assert_allclose(m3.data["x1"][()], xc_array - sx_array)
assert_allclose(m3.data["y1"][()], yc_array)
assert_allclose(m3.data["x2"][()], xc_array)
assert_allclose(m3.data["y2"][()], yc_array + sy_array)
