use egui_kittest::wgpu::{create_render_state, default_wgpu_setup};
use rsplot::{Colormap, ColormapName, Normalization, Plot2D, PlotWidget};
#[test]
fn plot_default_image_colormap_is_gray_linear() {
let rs = create_render_state(default_wgpu_setup());
rsplot::install(&rs);
let plot = PlotWidget::new(&rs, 0);
let cm = plot.default_colormap();
assert_eq!(cm.normalization, Normalization::Linear);
assert_eq!(cm.lut, Colormap::new(ColormapName::Gray, 0.0, 1.0).lut);
assert_eq!(cm.lut[0], [0u8, 0, 0, 255]);
assert_eq!(cm.lut[255], [255u8, 255, 255, 255]);
assert_ne!(cm.lut, Colormap::viridis(0.0, 1.0).lut);
let plot2d = Plot2D::new(&rs, 1);
assert_eq!(
plot2d.default_colormap().lut,
Colormap::new(ColormapName::Gray, 0.0, 1.0).lut
);
}