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)