pyo3/benches/bench_pyclass.rs

50 lines
1.2 KiB
Rust
Raw Normal View History

2021-12-09 00:03:50 +00:00
use criterion::{criterion_group, criterion_main, Criterion};
use pyo3::{class::PyObjectProtocol, prelude::*, type_object::LazyStaticType};
/// This is a feature-rich class instance used to benchmark various parts of the pyclass lifecycle.
#[pyclass]
struct MyClass {
#[pyo3(get, set)]
elements: Vec<i32>,
}
#[pymethods]
impl MyClass {
#[new]
fn new(elements: Vec<i32>) -> Self {
Self { elements }
}
fn __call__(&mut self, new_element: i32) -> usize {
self.elements.push(new_element);
self.elements.len()
}
}
#[pyproto]
impl PyObjectProtocol for MyClass {
/// A basic __str__ implementation.
fn __str__(&self) -> &'static str {
"MyClass"
}
}
pub fn first_time_init(b: &mut criterion::Bencher) {
let gil = Python::acquire_gil();
let py = gil.python();
b.iter(|| {
// This is using an undocumented internal PyO3 API to measure pyclass performance; please
// don't use this in your own code!
let ty = LazyStaticType::new();
ty.get_or_init::<MyClass>(py);
});
}
2021-05-26 06:25:00 +00:00
fn criterion_benchmark(c: &mut Criterion) {
c.bench_function("first_time_init", first_time_init);
2021-05-26 06:25:00 +00:00
}
criterion_group!(benches, criterion_benchmark);
2021-12-09 00:03:50 +00:00
2021-05-26 06:25:00 +00:00
criterion_main!(benches);