def test_array_animator_wcs_2d_colorbar_buttons(wcs_4d):
data = np.arange(120).reshape((5, 4, 3, 2))
bf = [lambda x: x] * 10
bl = ['h'] * 10
a = ArrayAnimatorWCS(data, wcs_4d, [0, 0, 'y', 'x'],
colorbar=True, button_func=bf, button_labels=bl)
a.update_plot(1, a.im, a.sliders[0]._slider)
return a.fig
def test_array_animator_wcs_2d_clip_interval_change(wcs_4d):
data = np.arange(120).reshape((5, 4, 3, 2))
pclims = [5, 95]
a = ArrayAnimatorWCS(data, wcs_4d, [0, 0, 'x', 'y'],
clip_interval=pclims * u.percent)
lims0 = a._get_2d_plot_limits()
a.update_plot(1, a.im, a.sliders[0]._slider)
lims1 = a._get_2d_plot_limits()
assert np.all(lims0 != lims1)
assert np.all(lims0 == np.percentile(data[..., 0, 0], pclims))
assert np.all(lims1 == np.percentile(data[..., 1, 0], pclims))
def test_array_animator_wcs_2d_extra_sliders(wcs_4d):
def vmin_slider(val, im, slider):
im.set_clim(vmin=val)
def vmax_slider(val, im, slider):
im.set_clim(vmax=val)
data = np.arange(120).reshape((5, 4, 3, 2))
a = ArrayAnimatorWCS(data, wcs_4d, [0, 0, 'y', 'x'], colorbar=True,
slider_functions=[vmin_slider, vmax_slider],
slider_ranges=[[0, 100], [0, 100]])
a.update_plot(1, a.im, a.sliders[0]._slider)
return a.fig
def test_array_animator_wcs_2d_transpose_update_plot(wcs_4d):
data = np.arange(120).reshape((5, 4, 3, 2))
a = ArrayAnimatorWCS(data, wcs_4d, [0, 0, 'y', 'x'], colorbar=True)
a.update_plot(1, a.im, a.sliders[0]._slider)
return a.fig