def test_plot_line2d(self, mpl_cleanup):
self.im.plot()
self.line2d.color = 'green'
self.line2d.position = ([12.0, 60.0], [36.0, 96.0])
self.line2d.set_mpl_ax(self.im._plot.signal_plot.ax)
assert self.line2d.ax == self.im._plot.signal_plot.ax
line2d = widgets.Line2DWidget(self.im.axes_manager)
line2d.snap_position = True
line2d.set_mpl_ax(self.im._plot.signal_plot.ax)
line2d.position = ([40.0, 20.0], [96.0, 36.0])
line2d.linewidth = 4
line2d.size = (15.0, )
assert line2d.size == (15.0, )
line2d_snap_all = widgets.Line2DWidget(self.im.axes_manager)
line2d_snap_all.snap_all = True
line2d_snap_all.set_mpl_ax(self.im._plot.signal_plot.ax)
line2d_snap_all.position = ([50.0, 60.0], [96.0, 54.0])
nt.assert_allclose(line2d_snap_all.position[0], [50.4, 60.0])
nt.assert_allclose(line2d_snap_all.position[1], [96.0, 54.0])
line2d_snap_all.size = (15.0, )
nt.assert_allclose(line2d_snap_all.size[0], 14.4)
nt.assert_allclose(line2d_snap_all.size[0], 14.4)
return self.im._plot.signal_plot.figure
def setup_method(self, method):
# Create test image 100x100 pixels:
self.im = Signal2D(np.arange(10000).reshape([100, 100]))
self.im.axes_manager[0].scale = 1.2
self.im.axes_manager[1].scale = 1.2
self.line2d = widgets.Line2DWidget(self.im.axes_manager)
