52 lines
884 B
Rust
52 lines
884 B
Rust
|
#![cfg(all(feature = "macros", not(PyPy)))]
|
||
|
|
|
||
|
|
use pyo3::prelude::*;
|
||
|
|
|
||
|
|
#[pyclass(subclass)]
|
||
|
|
struct BaseClass {
|
||
|
|
val1: usize,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[pymethods]
|
||
|
|
impl BaseClass {
|
||
|
|
#[new]
|
||
|
|
fn new() -> Self {
|
||
|
|
BaseClass { val1: 10 }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn method(&self) -> usize {
|
||
|
|
self.val1
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[pyclass(extends=BaseClass)]
|
||
|
|
struct SubClass {}
|
||
|
|
|
||
|
|
#[pymethods]
|
||
|
|
impl SubClass {
|
||
|
|
#[new]
|
||
|
|
fn new() -> (Self, BaseClass) {
|
||
|
|
(SubClass {}, BaseClass::new())
|
||
|
|
}
|
||
|
|
|
||
|
|
fn method(self_: &PyCell<Self>) -> PyResult<&PyAny> {
|
||
|
|
let super_ = self_.py_super()?;
|
||
|
|
super_.call_method("method", (), None)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_call_super_method() {
|
||
|
|
Python::with_gil(|py| {
|
||
|
|
let cls = py.get_type::<SubClass>();
|
||
|
|
pyo3::py_run!(
|
||
|
|
py,
|
||
|
|
cls,
|
||
|
|
r#"
|
||
|
|
obj = cls()
|
||
|
|
assert obj.method() == 10
|
||
|
|
"#
|
||
|
|
)
|
||
|
|
});
|
||
|
|
}
|