Function test_stream_arrows_2

tests/test_stream.rs:59–95

()

#[test]
fn test_stream_arrows_2() -> Result<(), StrError> {
    // object and options
    let mut stream = Stream::new();
    let mut quiver = Stream::new();

    // data
    let n = 10;
    let x = linspace(1.0, 2.0, n);
    let y = x.clone();
    let u = linspace(1.0, 2.0, n);
    let v = u.clone();
    let (uu, vv) = generate2d(1.0, 2.0, 1.0, 2.0, n, n);

    // draw arrows
    stream.set_streamline_zorder(1).draw_alt(&x, &y, &uu, &vv);
    quiver
        .set_quiver_zorder(2)
        .set_color("red")
        .set_quiver_inv_scale(20.0)
        .draw_arrows_alt(&x, &y, &u, &v);

    // add contour to plot
    let mut plot = Plot::new();
    plot.add(&stream).add(&quiver);

    // save figure
    let path = Path::new(OUT_DIR).join("integ_stream_arrows_2.svg");
    plot.set_equal_axes(true).save(&path)?;

    // check number of lines
    let file = File::open(path).map_err(|_| "cannot open file")?;
    let buffered = BufReader::new(file);
    let lines_iter = buffered.lines();
    let n = lines_iter.count();
    assert!(n > 2000 && n < 3680);
    Ok(())
}