pyo3/tests/test_anyhow.rs

53 lines
1.1 KiB
Rust
Raw Normal View History

#![cfg(feature = "anyhow")]
use pyo3::wrap_pyfunction_bound;
#[test]
fn test_anyhow_py_function_ok_result() {
use pyo3::{py_run, pyfunction, Python};
#[pyfunction]
2022-02-28 08:03:54 +00:00
#[allow(clippy::unnecessary_wraps)]
fn produce_ok_result() -> anyhow::Result<String> {
Ok(String::from("OK buddy"))
}
Python::with_gil(|py| {
let func = wrap_pyfunction_bound!(produce_ok_result)(py).unwrap();
py_run!(
py,
func,
r#"
func()
"#
);
});
}
#[test]
fn test_anyhow_py_function_err_result() {
2024-02-05 18:44:00 +00:00
use pyo3::prelude::PyDictMethods;
use pyo3::{pyfunction, types::PyDict, Python};
#[pyfunction]
fn produce_err_result() -> anyhow::Result<String> {
anyhow::bail!("error time")
}
Python::with_gil(|py| {
let func = wrap_pyfunction_bound!(produce_err_result)(py).unwrap();
2024-02-05 18:44:00 +00:00
let locals = PyDict::new_bound(py);
locals.set_item("func", func).unwrap();
2024-02-05 18:44:00 +00:00
py.run_bound(
r#"
func()
"#,
None,
2024-02-05 18:44:00 +00:00
Some(&locals),
)
.unwrap_err();
});
}