use plotpy::{Curve, Legend, Plot, StrError};
use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::Path;
const OUT_DIR: &str = "/tmp/plotpy/integ_tests";
#[test]
fn test_legend_1() -> Result<(), StrError> {
let mut curve1 = Curve::new();
curve1.set_label("my-curve");
let mut curve2 = Curve::new();
curve2.set_label("another-curve");
let x = &[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0];
let y = &[1.0, 1.414, 1.732, 2.0, 2.236, 2.449, 2.645, 2.828, 3.0, 3.162];
let y2 = &[1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 3.5, 3.5, 3.5, 3.5];
curve1.draw(x, y);
curve2.draw(x, y2);
let mut legend = Legend::new();
legend
.set_fontsize(10.0)
.set_handle_len(3.2)
.set_num_col(2)
.set_outside(true)
.set_show_frame(false)
.set_x_coords(&[0.2, 1.05, 0.8, 0.1]);
legend.draw();
let mut plot = Plot::new();
plot.add(&curve1);
plot.add(&curve2);
plot.add(&legend);
let path = Path::new(OUT_DIR).join("integ_legend_1.svg");
plot.save(&path)?;
let file = File::open(path).map_err(|_| "cannot open file")?;
let buffered = BufReader::new(file);
let lines_iter = buffered.lines();
assert!(lines_iter.count() > 780);
Ok(())
}
#[test]
fn test_legend_2() -> Result<(), StrError> {
let mut curve1 = Curve::new();
curve1.set_marker_style("o").set_label("my-curve");
let mut curve2 = Curve::new();
curve2.set_marker_style("*").set_label("another-curve");
let x = &[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0];
let y = &[1.0, 1.414, 1.732, 2.0, 2.236, 2.449, 2.645, 2.828, 3.0, 3.162];
let y2 = &[1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 3.5, 3.5, 3.5, 3.5];
curve1.draw(x, y);
curve2.draw(x, y2);
let mut legend = Legend::new();
legend.set_extra("numpoints=3").set_location("lower right");
legend.draw();
let mut plot = Plot::new();
plot.add(&curve1);
plot.add(&curve2);
plot.add(&legend);
let path = Path::new(OUT_DIR).join("integ_legend_2.svg");
plot.save(&path)?;
let file = File::open(path).map_err(|_| "cannot open file")?;
let buffered = BufReader::new(file);
let lines_iter = buffered.lines();
assert!(lines_iter.count() > 780);
Ok(())
}