51 lines
1.4 KiB
Rust
51 lines
1.4 KiB
Rust
#![allow(dead_code, unused_variables)]
|
|
|
|
#[macro_use] extern crate pyo3;
|
|
|
|
use pyo3::*;
|
|
|
|
|
|
#[test]
|
|
fn test_basics() {
|
|
let gil = Python::acquire_gil();
|
|
let py = gil.python();
|
|
|
|
let v = PySlice::new(py, 1, 10, 2);
|
|
let indices = v.indices(py, 100).unwrap();
|
|
assert_eq!(1, indices.start);
|
|
assert_eq!(10, indices.stop);
|
|
assert_eq!(2, indices.step);
|
|
assert_eq!(5, indices.slicelength);
|
|
}
|
|
|
|
|
|
py_class!(pub class Test |py| {
|
|
def __getitem__(&self, idx: PyObject) -> PyResult<PyObject> {
|
|
if let Ok(slice) = PySlice::downcast_from(py, idx.clone_ref(py)) {
|
|
let indices = slice.indices(py, 1000)?;
|
|
if indices.start == 100 && indices.stop == 200 && indices.step == 1 {
|
|
return Ok("slice".to_py_object(py).into_object())
|
|
}
|
|
}
|
|
else if let Ok(idx) = idx.extract::<isize>(py) {
|
|
if idx == 1 {
|
|
return Ok("int".to_py_object(py).into_object())
|
|
}
|
|
}
|
|
Err(PyErr::new::<exc::ValueError, _>(py, "error"))
|
|
}
|
|
});
|
|
|
|
#[test]
|
|
fn test_cls_impl() {
|
|
let gil = Python::acquire_gil();
|
|
let py = gil.python();
|
|
|
|
let ob = Test::create_instance(py).unwrap();
|
|
let d = PyDict::new(py);
|
|
d.set_item(py, "ob", ob).unwrap();
|
|
|
|
py.run("assert ob[1] == 'int'", None, Some(&d)).unwrap();
|
|
py.run("assert ob[100:200:1] == 'slice'", None, Some(&d)).unwrap();
|
|
}
|