pyo3/tests/test_module.rs

446 lines
11 KiB
Rust
Raw Normal View History

2021-12-03 00:03:32 +00:00
#![cfg(feature = "macros")]
use pyo3::prelude::*;
2018-11-12 21:28:45 +00:00
2021-06-24 14:18:48 +00:00
use pyo3::py_run;
use pyo3::types::{IntoPyDict, PyDict, PyTuple};
2018-09-03 18:50:18 +00:00
mod common;
#[pyclass]
struct AnonClass {}
#[pyclass]
struct ValueClass {
value: usize,
}
#[pymethods]
impl ValueClass {
#[new]
fn new(value: usize) -> ValueClass {
ValueClass { value }
}
}
#[pyclass(module = "module")]
struct LocatedClass {}
2018-04-30 20:38:48 +00:00
#[pyfunction]
2018-07-30 20:52:22 +00:00
/// Doubles the given value
fn double(x: usize) -> usize {
x * 2
}
2018-04-30 20:38:48 +00:00
/// This module is implemented in Rust.
2018-11-12 21:28:45 +00:00
#[pymodule]
2022-03-23 07:07:28 +00:00
fn module_with_functions(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
2021-05-16 09:41:59 +00:00
#[pyfn(m)]
#[pyo3(name = "no_parameters")]
fn function_with_name() -> usize {
2021-02-11 21:37:38 +00:00
42
2018-04-30 20:38:48 +00:00
}
#[pyfn(m)]
#[pyo3(pass_module)]
fn with_module(module: &PyModule) -> PyResult<&str> {
module.name()
}
2021-05-16 09:41:59 +00:00
#[pyfn(m)]
fn double_value(v: &ValueClass) -> usize {
v.value * 2
}
m.add_class::<AnonClass>().unwrap();
m.add_class::<ValueClass>().unwrap();
m.add_class::<LocatedClass>().unwrap();
2018-04-30 20:38:48 +00:00
2018-05-05 13:50:04 +00:00
m.add("foo", "bar").unwrap();
2018-04-30 20:38:48 +00:00
m.add_function(wrap_pyfunction!(double, m)?).unwrap();
m.add("also_double", wrap_pyfunction!(double, m)?).unwrap();
2018-04-30 20:38:48 +00:00
Ok(())
}
#[test]
fn test_module_with_functions() {
2019-02-01 14:00:41 +00:00
use pyo3::wrap_pymodule;
2018-04-30 20:38:48 +00:00
let gil = Python::acquire_gil();
let py = gil.python();
let d = [(
2018-11-12 21:25:45 +00:00
"module_with_functions",
2019-02-01 13:01:18 +00:00
wrap_pymodule!(module_with_functions)(py),
2019-03-24 16:19:15 +00:00
)]
.into_py_dict(py);
2021-03-14 07:34:05 +00:00
py_assert!(
py,
*d,
"module_with_functions.__doc__ == 'This module is implemented in Rust.'"
);
py_assert!(py, *d, "module_with_functions.no_parameters() == 42");
py_assert!(py, *d, "module_with_functions.foo == 'bar'");
py_assert!(py, *d, "module_with_functions.AnonClass != None");
py_assert!(py, *d, "module_with_functions.LocatedClass != None");
py_assert!(
py,
*d,
"module_with_functions.LocatedClass.__module__ == 'module'"
);
py_assert!(py, *d, "module_with_functions.double(3) == 6");
py_assert!(
py,
*d,
"module_with_functions.double.__doc__ == 'Doubles the given value'"
);
py_assert!(py, *d, "module_with_functions.also_double(3) == 6");
py_assert!(
py,
*d,
"module_with_functions.also_double.__doc__ == 'Doubles the given value'"
);
py_assert!(
py,
*d,
"module_with_functions.double_value(module_with_functions.ValueClass(1)) == 2"
);
py_assert!(
py,
*d,
"module_with_functions.with_module() == 'module_with_functions'"
);
2018-04-30 20:38:48 +00:00
}
#[pymodule]
#[pyo3(name = "other_name")]
2022-03-23 07:07:28 +00:00
fn some_name(_: Python<'_>, m: &PyModule) -> PyResult<()> {
2021-02-11 21:37:38 +00:00
m.add("other_name", "other_name")?;
Ok(())
}
#[test]
fn test_module_renaming() {
2019-02-23 17:01:22 +00:00
use pyo3::wrap_pymodule;
let gil = Python::acquire_gil();
let py = gil.python();
2021-12-28 01:25:01 +00:00
let d = [("different_name", wrap_pymodule!(some_name)(py))].into_py_dict(py);
2021-03-14 07:34:05 +00:00
py_run!(py, *d, "assert different_name.__name__ == 'other_name'");
}
#[test]
fn test_module_from_code() {
let gil = Python::acquire_gil();
let py = gil.python();
let adder_mod = PyModule::from_code(
py,
"def add(a,b):\n\treturn a+b",
"adder_mod.py",
"adder_mod",
)
.expect("Module code should be loaded");
let add_func = adder_mod
2021-03-13 17:36:17 +00:00
.getattr("add")
.expect("Add function should be in the module")
.to_object(py);
let ret_value: i32 = add_func
.call1(py, (1, 2))
.expect("A value should be returned")
.extract(py)
.expect("The value should be able to be converted to an i32");
assert_eq!(ret_value, 3);
}
2018-09-03 18:50:18 +00:00
#[pyfunction]
fn r#move() -> usize {
42
}
2018-11-12 21:28:45 +00:00
#[pymodule]
2022-03-23 07:07:28 +00:00
fn raw_ident_module(_py: Python<'_>, module: &PyModule) -> PyResult<()> {
module.add_function(wrap_pyfunction!(r#move, module)?)
2018-09-03 18:50:18 +00:00
}
#[test]
fn test_raw_idents() {
2019-02-01 14:00:41 +00:00
use pyo3::wrap_pymodule;
2018-09-03 18:50:18 +00:00
let gil = Python::acquire_gil();
let py = gil.python();
2019-02-01 13:01:18 +00:00
let module = wrap_pymodule!(raw_ident_module)(py);
2018-09-03 18:50:18 +00:00
py_assert!(py, module, "module.move() == 42");
}
2018-11-12 21:25:45 +00:00
#[pyfunction]
#[pyo3(name = "foobar")]
fn custom_named_fn() -> usize {
42
}
#[pymodule]
2022-03-23 07:07:28 +00:00
fn foobar_module(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
m.add_function(wrap_pyfunction!(custom_named_fn, m)?)?;
2020-03-16 06:34:02 +00:00
m.dict().set_item("yay", "me")?;
Ok(())
}
#[test]
fn test_custom_names() {
let gil = Python::acquire_gil();
let py = gil.python();
2020-03-16 06:34:02 +00:00
let module = pyo3::wrap_pymodule!(foobar_module)(py);
py_assert!(py, module, "not hasattr(module, 'custom_named_fn')");
py_assert!(py, module, "module.foobar() == 42");
}
2020-03-16 06:34:02 +00:00
#[test]
fn test_module_dict() {
let gil = Python::acquire_gil();
let py = gil.python();
let module = pyo3::wrap_pymodule!(foobar_module)(py);
py_assert!(py, module, "module.yay == 'me'");
}
2021-03-20 03:16:52 +00:00
#[test]
fn test_module_dunder_all() {
let gil = Python::acquire_gil();
let py = gil.python();
let module = pyo3::wrap_pymodule!(foobar_module)(py);
py_assert!(py, module, "module.__all__ == ['foobar']");
}
2018-11-12 21:25:45 +00:00
#[pyfunction]
fn subfunction() -> String {
"Subfunction".to_string()
}
fn submodule(module: &PyModule) -> PyResult<()> {
module.add_function(wrap_pyfunction!(subfunction, module)?)?;
Ok(())
}
2018-11-12 21:28:45 +00:00
#[pymodule]
2022-03-23 07:07:28 +00:00
fn submodule_with_init_fn(_py: Python<'_>, module: &PyModule) -> PyResult<()> {
module.add_function(wrap_pyfunction!(subfunction, module)?)?;
2018-11-12 21:25:45 +00:00
Ok(())
}
2018-11-12 21:28:45 +00:00
#[pyfunction]
2018-11-12 21:25:45 +00:00
fn superfunction() -> String {
"Superfunction".to_string()
}
2018-11-12 21:28:45 +00:00
#[pymodule]
2022-03-23 07:07:28 +00:00
fn supermodule(py: Python<'_>, module: &PyModule) -> PyResult<()> {
module.add_function(wrap_pyfunction!(superfunction, module)?)?;
let module_to_add = PyModule::new(py, "submodule")?;
submodule(module_to_add)?;
module.add_submodule(module_to_add)?;
let module_to_add = PyModule::new(py, "submodule_with_init_fn")?;
submodule_with_init_fn(py, module_to_add)?;
module.add_submodule(module_to_add)?;
2018-11-12 21:25:45 +00:00
Ok(())
}
#[test]
fn test_module_nesting() {
2019-02-01 14:00:41 +00:00
use pyo3::wrap_pymodule;
let gil = Python::acquire_gil();
2018-11-12 21:25:45 +00:00
let py = gil.python();
2019-02-01 13:01:18 +00:00
let supermodule = wrap_pymodule!(supermodule)(py);
2018-11-12 21:25:45 +00:00
py_assert!(
py,
supermodule,
"supermodule.superfunction() == 'Superfunction'"
);
py_assert!(
py,
supermodule,
"supermodule.submodule.subfunction() == 'Subfunction'"
);
py_assert!(
py,
supermodule,
"supermodule.submodule_with_init_fn.subfunction() == 'Subfunction'"
);
2018-11-12 21:25:45 +00:00
}
// Test that argument parsing specification works for pyfunctions
#[pyfunction(a = 5, vararg = "*")]
2022-03-23 07:07:28 +00:00
fn ext_vararg_fn(py: Python<'_>, a: i32, vararg: &PyTuple) -> PyObject {
[a.to_object(py), vararg.into()].to_object(py)
}
#[pymodule]
2022-03-23 07:07:28 +00:00
fn vararg_module(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
2021-05-16 09:41:59 +00:00
#[pyfn(m, a = 5, vararg = "*")]
2022-03-23 07:07:28 +00:00
fn int_vararg_fn(py: Python<'_>, a: i32, vararg: &PyTuple) -> PyObject {
ext_vararg_fn(py, a, vararg)
}
2021-05-16 09:41:59 +00:00
m.add_function(wrap_pyfunction!(ext_vararg_fn, m)?).unwrap();
Ok(())
}
#[test]
fn test_vararg_module() {
let gil = Python::acquire_gil();
let py = gil.python();
let m = pyo3::wrap_pymodule!(vararg_module)(py);
py_assert!(py, m, "m.ext_vararg_fn() == [5, ()]");
py_assert!(py, m, "m.ext_vararg_fn(1, 2) == [1, (2,)]");
py_assert!(py, m, "m.int_vararg_fn() == [5, ()]");
py_assert!(py, m, "m.int_vararg_fn(1, 2) == [1, (2,)]");
}
2020-08-30 15:22:41 +00:00
#[test]
fn test_module_with_constant() {
// Regression test for #1102
#[pymodule]
2022-03-23 07:07:28 +00:00
fn module_with_constant(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
2020-08-30 15:22:41 +00:00
const ANON: AnonClass = AnonClass {};
m.add("ANON", ANON)?;
m.add_class::<AnonClass>()?;
Ok(())
}
Python::with_gil(|py| {
let m = pyo3::wrap_pymodule!(module_with_constant)(py);
py_assert!(py, m, "isinstance(m.ANON, m.AnonClass)");
});
}
#[pyfunction]
#[pyo3(pass_module)]
fn pyfunction_with_module(module: &PyModule) -> PyResult<&str> {
module.name()
}
#[pyfunction]
#[pyo3(pass_module)]
fn pyfunction_with_module_and_py<'a>(
module: &'a PyModule,
_python: Python<'a>,
) -> PyResult<&'a str> {
module.name()
}
#[pyfunction]
#[pyo3(pass_module)]
fn pyfunction_with_module_and_arg(module: &PyModule, string: String) -> PyResult<(&str, String)> {
module.name().map(|s| (s, string))
}
#[pyfunction(string = "\"foo\"")]
#[pyo3(pass_module)]
fn pyfunction_with_module_and_default_arg<'a>(
module: &'a PyModule,
string: &str,
) -> PyResult<(&'a str, String)> {
module.name().map(|s| (s, string.into()))
}
#[pyfunction(args = "*", kwargs = "**")]
#[pyo3(pass_module)]
fn pyfunction_with_module_and_args_kwargs<'a>(
module: &'a PyModule,
args: &PyTuple,
kwargs: Option<&PyDict>,
) -> PyResult<(&'a str, usize, Option<usize>)> {
module
.name()
.map(|s| (s, args.len(), kwargs.map(|d| d.len())))
}
#[pyfunction]
#[pyo3(pass_module)]
fn pyfunction_with_pass_module_in_attribute(module: &PyModule) -> PyResult<&str> {
module.name()
}
#[pymodule]
2022-03-23 07:07:28 +00:00
fn module_with_functions_with_module(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
2021-05-16 09:41:59 +00:00
m.add_function(wrap_pyfunction!(pyfunction_with_module, m)?)?;
m.add_function(wrap_pyfunction!(pyfunction_with_module_and_py, m)?)?;
m.add_function(wrap_pyfunction!(pyfunction_with_module_and_arg, m)?)?;
m.add_function(wrap_pyfunction!(pyfunction_with_module_and_default_arg, m)?)?;
m.add_function(wrap_pyfunction!(pyfunction_with_module_and_args_kwargs, m)?)?;
m.add_function(wrap_pyfunction!(
pyfunction_with_pass_module_in_attribute,
m
)?)?;
Ok(())
}
#[test]
fn test_module_functions_with_module() {
let gil = Python::acquire_gil();
let py = gil.python();
let m = pyo3::wrap_pymodule!(module_with_functions_with_module)(py);
py_assert!(
py,
m,
"m.pyfunction_with_module() == 'module_with_functions_with_module'"
);
py_assert!(
py,
m,
"m.pyfunction_with_module_and_py() == 'module_with_functions_with_module'"
);
py_assert!(
py,
m,
"m.pyfunction_with_module_and_default_arg() \
== ('module_with_functions_with_module', 'foo')"
);
py_assert!(
py,
m,
"m.pyfunction_with_module_and_args_kwargs(1, x=1, y=2) \
== ('module_with_functions_with_module', 1, 2)"
);
py_assert!(
py,
m,
"m.pyfunction_with_pass_module_in_attribute() == 'module_with_functions_with_module'"
);
}
#[test]
fn test_module_doc_hidden() {
#[doc(hidden)]
2022-02-28 08:03:54 +00:00
#[allow(clippy::unnecessary_wraps)]
#[pymodule]
2022-03-23 07:07:28 +00:00
fn my_module(_py: Python<'_>, _m: &PyModule) -> PyResult<()> {
Ok(())
}
Python::with_gil(|py| {
let m = pyo3::wrap_pymodule!(my_module)(py);
py_assert!(py, m, "m.__doc__ == ''");
})
}