luxolus/pyo3
2
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

add benchmark for mapping conversion from dicts

Browse source
This commit is contained in:
David Hewitt 2023-02-15 08:07:07 +00:00
parent 40709db801
commit 59630beb51
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
benches/bench_dict.rs
View file

@ -1,7 +1,7 @@
use criterion::{criterion_group, criterion_main, Bencher, Criterion};
use pyo3::prelude::*;
use pyo3::types::IntoPyDict;
use pyo3::{prelude::*, types::PyMapping};
use std::collections::{BTreeMap, HashMap};
fn iter_dict(b: &mut Bencher<'_>) {
@ -63,6 +63,19 @@ fn extract_hashbrown_map(b: &mut Bencher<'_>) {
});
}
fn mapping_from_dict(b: &mut Bencher<'_>) {
Python::with_gil(|py| {
const LEN: usize = 100_000;
let dict = (0..LEN as u64)
.map(|i| (i, i * 2))
.into_py_dict(py)
.to_object(py);
b.iter(|| {
let _: &PyMapping = dict.extract(py).unwrap();
});
});
}
fn criterion_benchmark(c: &mut Criterion) {
c.bench_function("iter_dict", iter_dict);
c.bench_function("dict_new", dict_new);
@ -72,6 +85,8 @@ fn criterion_benchmark(c: &mut Criterion) {
#[cfg(feature = "hashbrown")]
c.bench_function("extract_hashbrown_map", extract_hashbrown_map);
c.bench_function("mapping_from_dict", mapping_from_dict);
}
criterion_group!(benches, criterion_benchmark);