def generate_image_name(image):
"""Generate a name for an image"""
label = image.get_label().lstrip('_')
ax_name = generate_ax_name(image.axes)
if label:
return "{} - {}".format(ax_name, label)
else:
return "{}".format(ax_name)
def set_ax_title(self, ax, new_title):
"""Set axes' title and update its entry in the axes selector"""
ax.set_title(new_title)
self.rename_selected_axes(generate_ax_name(ax))
def test_generate_ax_name_returns_correct_name(self):
ax0_name = generate_ax_name(self.fig.get_axes()[0])
self.assertEqual("My Axes: (0, 0)", ax0_name)
ax1_name = generate_ax_name(self.fig.get_axes()[1])
self.assertEqual("(1, 0)", ax1_name)