#[macro_use]
extern crate criterion;

extern crate byte_ngram;

use criterion::Criterion;

use byte_ngram::*;

fn criterion_benchmark(c: &mut Criterion) {
    let data = String::from(
        "Blackmail is such an ugly word. I prefer extortion. The 'x' makes it sound cool.",
    );

    c.bench_function("from_slice", |b| b.iter(|| for _ in from_slice(data.as_bytes()) {}));
    c.bench_function("ByteNgramReader", |b| b.iter(|| for _ in ByteNgramReader::new(data.as_bytes()) {}));
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);