def test_normalize_keypress_cb(mock_show):
x = np.linspace(0, 6, 1000)
y = np.cos(x)
obj = DraggableEnvelope(x, y, mode="u")
mck = mock_event(xdata=50,
ydata=50,
button="d",
key="d",
fig=obj.fig,
canvas=obj.fig.canvas,
inaxes=True)
obj.key_press_callback(event=mck)
def test_normalize_keypress_cb4(mock_show):
x = np.linspace(0, 6, 1000)
y = np.cos(x)
obj = DraggableEnvelope(x, y, mode="l")
mck = mock_event(xdata=2,
ydata=0.5,
button="i",
key="i",
fig=obj.fig,
canvas=obj.fig.canvas,
inaxes=True)
obj.key_press_callback(event=mck)
assert 2 in obj.x_env
assert 0.5 in obj.y_env
mock_show.assert_called()
def test_normalize_k_press_cb2(mock_show):
x = np.linspace(0, 6, 1000)
y = np.cos(x)
obj = DraggableEnvelope(x, y, mode="l")
obj.epsilon = 1000000
obj.x_env = np.array([56, 5])
obj.y_env = np.array([60, 0.5])
xy_pixels = obj.ax.transData.transform([5, 0.5])
xpix, ypix = xy_pixels
mck = mock_event(xdata=xpix,
ydata=ypix,
button="d",
key="d",
fig=obj.fig,
canvas=obj.fig.canvas,
inaxes=obj.ax)
obj.key_press_callback(event=mck)
assert 5 not in obj.x_env
assert 0.5 not in obj.y_env
mock_show.assert_called()