def test_simple(self):
sorted_cluster_dict = {}
sorted_cluster_dict["centre"] = randint(10, size=(3, 4, 2, 2))
sorted_cluster_dict["rest"] = randint(50, 60, size=(3, 4, 3, 2))
sorted_cluster_dict["none"] = randint(90, size=(3, 4, 1, 2))
marker_list = ct._sorted_cluster_dict_to_marker_list(sorted_cluster_dict)
assert len(marker_list) == 2 + 3 + 1
def test_size(self):
marker_size = 30
sorted_cluster_dict = {}
sorted_cluster_dict["centre"] = randint(10, size=(3, 4, 2, 2))
sorted_cluster_dict["rest"] = randint(50, 60, size=(3, 4, 3, 2))
sorted_cluster_dict["none"] = randint(90, size=(3, 4, 1, 2))
marker_list = ct._sorted_cluster_dict_to_marker_list(
sorted_cluster_dict, size=marker_size)
for marker in marker_list:
assert marker.get_data_position("size") == marker_size
def test_color(self):
marker_color = "orange"
sorted_cluster_dict = {}
sorted_cluster_dict["centre"] = randint(10, size=(3, 4, 2, 2))
sorted_cluster_dict["rest"] = randint(50, 60, size=(3, 4, 3, 2))
sorted_cluster_dict["none"] = randint(90, size=(3, 4, 1, 2))
marker_list = ct._sorted_cluster_dict_to_marker_list(
sorted_cluster_dict,
color_rest=marker_color,
color_centre=marker_color,
color_none=marker_color,
)
for marker in marker_list:
assert marker.marker_properties["color"] == marker_color
