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)