def test_plot_range(self, mpl_cleanup):
self.s.plot()
self.range.set_mpl_ax(self.s._plot.signal_plot.ax)
assert self.range.ax == self.s._plot.signal_plot.ax
assert self.range.color == 'red' # default color
assert self.range.position == (0.0, )
assert self.range.size == (1.2, )
assert self.range.span.rect.get_alpha() == 0.5
w = widgets.RangeWidget(self.s.axes_manager, color='blue')
w.set_mpl_ax(self.s._plot.signal_plot.ax)
w.set_ibounds(left=4, width=3)
assert w.color == 'blue'
color_rgba = matplotlib.colors.to_rgba('blue', alpha=0.5)
assert w.span.rect.get_fc() == color_rgba
assert w.span.rect.get_ec() == color_rgba
nt.assert_allclose(w.position[0], 4.8)
nt.assert_allclose(w.size[0], 3.6)
w2 = widgets.RangeWidget(self.s.axes_manager)
w2.set_mpl_ax(self.s._plot.signal_plot.ax)
assert w2.ax == self.s._plot.signal_plot.ax
w2.set_bounds(left=24.0, width=12.0)
w2.color = 'green'
assert w2.color == 'green'
w2.alpha = 0.25
assert w2.alpha == 0.25
return self.s._plot.signal_plot.figure
def test_plot_range(self):
s = self.s
range_widget = self.range_widget
s.plot()
range_widget.set_mpl_ax(s._plot.signal_plot.ax)
assert range_widget.ax == s._plot.signal_plot.ax
assert range_widget.color == 'r' # default color
assert range_widget.position == (0.0, )
assert range_widget.size == (1.2, )
assert range_widget.span.artists[0].get_alpha() == 0.25
w = widgets.RangeWidget(s.axes_manager, color='blue')
w.set_mpl_ax(s._plot.signal_plot.ax)
w.set_ibounds(left=4, width=3)
assert w.color == 'blue'
color_rgba = matplotlib.colors.to_rgba('blue', alpha=0.25)
assert w.span.artists[0].get_fc() == color_rgba
assert w.span.artists[0].get_ec() == color_rgba
np.testing.assert_allclose(w.position[0], 4.8)
np.testing.assert_allclose(w.size[0], 3.6)
w2 = widgets.RangeWidget(s.axes_manager)
w2.set_mpl_ax(s._plot.signal_plot.ax)
assert w2.ax == s._plot.signal_plot.ax
w2.set_bounds(left=24.0, width=12.0)
assert w2.position[0] == 24.0
assert w2.size[0] == 12.0
w2.color = 'green'
assert w2.color == 'green'
w2.alpha = 0.25
assert w2.alpha == 0.25
return s._plot.signal_plot.figure
def test_plot_range_Signal2D(self):
im = Signal2D(np.arange(10 * 10).reshape((10, 10)))
im.axes_manager[0].scale = 0.1
im.axes_manager[1].scale = 5
im.plot()
range_h = widgets.RangeWidget(im.axes_manager, direction='horizontal')
range_h.set_mpl_ax(im._plot.signal_plot.ax)
range_v = widgets.RangeWidget(im.axes_manager, direction='vertical',
color='blue')
range_v.axes = (im.axes_manager[1],)
range_v.set_mpl_ax(im._plot.signal_plot.ax)
assert range_v.position == (0.0, )
assert range_v.size == (5.0, )
range_v.set_bounds(left=20.0, width=15.0)
assert range_v.position == (20.0, )
assert range_v.size == (15.0, )
return im._plot.signal_plot.figure
def setup_method(self, method):
self.s = Signal1D(np.arange(50))
self.s.axes_manager[0].scale = 1.2
self.range = widgets.RangeWidget(self.s.axes_manager)