blob: c575c2ff63dc0625ec4b14739b0a42abdc69f46b [file] [log] [blame]
use super::*;
use crate::report::ComparisonData;
use std::path::Path;
pub(crate) fn t_test(
path: &Path,
title: &str,
comparison: &ComparisonData,
size: Option<(u32, u32)>,
) {
let t = comparison.t_value;
let (xs, ys) = kde::sweep(&comparison.t_distribution, KDE_POINTS, None);
let x_range = plotters::data::fitting_range(xs.iter());
let mut y_range = plotters::data::fitting_range(ys.iter());
y_range.start = 0.0;
y_range.end *= 1.1;
let root_area = SVGBackend::new(&path, size.unwrap_or(SIZE)).into_drawing_area();
let mut chart = ChartBuilder::on(&root_area)
.margin((5).percent())
.caption(format!("{}: Welch t test", title), (DEFAULT_FONT, 20))
.set_label_area_size(LabelAreaPosition::Left, (5).percent_width().min(60))
.set_label_area_size(LabelAreaPosition::Bottom, (5).percent_height().min(40))
.build_cartesian_2d(x_range, y_range.clone())
.unwrap();
chart
.configure_mesh()
.disable_mesh()
.y_desc("Density")
.x_desc("t score")
.draw()
.unwrap();
chart
.draw_series(AreaSeries::new(
xs.iter().zip(ys.iter()).map(|(x, y)| (*x, *y)),
0.0,
DARK_BLUE.mix(0.25),
))
.unwrap()
.label("t distribution")
.legend(|(x, y)| {
Rectangle::new([(x, y - 5), (x + 20, y + 5)], DARK_BLUE.mix(0.25).filled())
});
chart
.draw_series(std::iter::once(PathElement::new(
vec![(t, 0.0), (t, y_range.end)],
DARK_BLUE.filled().stroke_width(2),
)))
.unwrap()
.label("t statistic")
.legend(|(x, y)| PathElement::new(vec![(x, y), (x + 20, y)], DARK_BLUE));
chart.configure_series_labels().draw().unwrap();
}