pyo3/tests/test_static_slots.rs

73 lines
1.4 KiB
Rust
Raw Normal View History

#![cfg(feature = "macros")]
use pyo3::exceptions::PyIndexError;
use pyo3::prelude::*;
use pyo3::types::IntoPyDict;
use pyo3::py_run;
2023-09-24 12:34:53 +00:00
#[path = "../src/tests/common.rs"]
mod common;
#[pyclass]
struct Count5();
#[pymethods]
impl Count5 {
#[new]
fn new() -> Self {
Self()
}
#[staticmethod]
fn __len__() -> usize {
5
}
#[staticmethod]
fn __getitem__(idx: isize) -> PyResult<f64> {
if idx < 0 {
Err(PyIndexError::new_err("Count5 cannot count backwards"))
} else if idx > 4 {
Err(PyIndexError::new_err("Count5 cannot count higher than 5"))
} else {
Ok(idx as f64 + 1.0)
}
}
}
/// Return a dict with `s = Count5()`.
2024-02-10 12:59:55 +00:00
fn test_dict(py: Python<'_>) -> Bound<'_, pyo3::types::PyDict> {
let d = [("Count5", py.get_type_bound::<Count5>())].into_py_dict_bound(py);
// Though we can construct `s` in Rust, let's test `__new__` works.
py_run!(py, *d, "s = Count5()");
d
}
#[test]
fn test_len() {
Python::with_gil(|py| {
let d = test_dict(py);
py_assert!(py, *d, "len(s) == 5");
});
}
#[test]
fn test_getitem() {
Python::with_gil(|py| {
let d = test_dict(py);
py_assert!(py, *d, "s[4] == 5.0");
});
}
#[test]
fn test_list() {
Python::with_gil(|py| {
let d = test_dict(py);
py_assert!(py, *d, "list(s) == [1.0, 2.0, 3.0, 4.0, 5.0]");
});
}