48 lines
1.2 KiB
Rust
48 lines
1.2 KiB
Rust
#[macro_use]
|
|
extern crate criterion;
|
|
|
|
use criterion::Criterion;
|
|
|
|
use clap::{App, Arg};
|
|
use starship::modules;
|
|
use std::path::Path;
|
|
|
|
fn char_segment(c: &mut Criterion) {
|
|
let args = App::new("starship")
|
|
.arg(Arg::with_name("status_code"))
|
|
.get_matches_from(vec!["starship", "0"]);
|
|
|
|
let path = Path::new("~");
|
|
|
|
c.bench_function("char segment", move |b| {
|
|
b.iter(|| modules::handle("char", &path, &args))
|
|
});
|
|
}
|
|
|
|
fn dir_segment(c: &mut Criterion) {
|
|
let args = App::new("starship")
|
|
.arg(Arg::with_name("status_code"))
|
|
.get_matches_from(vec!["starship", "0"]);
|
|
|
|
let path = Path::new("~");
|
|
|
|
c.bench_function("dir segment", move |b| {
|
|
b.iter(|| modules::handle("dir", &path, &args))
|
|
});
|
|
}
|
|
|
|
fn line_break_segment(c: &mut Criterion) {
|
|
let args = App::new("starship")
|
|
.arg(Arg::with_name("status_code"))
|
|
.get_matches_from(vec!["starship", "0"]);
|
|
|
|
let path = Path::new("~");
|
|
|
|
c.bench_function("line break segment", move |b| {
|
|
b.iter(|| modules::handle("line_break", &path, &args))
|
|
});
|
|
}
|
|
|
|
criterion_group!(benches, dir_segment, char_segment, line_break_segment);
|
|
criterion_main!(benches);
|