77 lines
2.8 KiB
Rust
77 lines
2.8 KiB
Rust
|
#[macro_use] extern crate cpython;
|
||
|
|
|
||
|
|
use cpython::{PyResult, Python, NoArgs, ToPyObject, ObjectProtocol, PyDict, PyTuple};
|
||
|
|
use std::sync::atomic;
|
||
|
|
use std::sync::atomic::Ordering::Relaxed;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn no_args() {
|
||
|
|
static CALL_COUNT: atomic::AtomicUsize = atomic::ATOMIC_USIZE_INIT;
|
||
|
|
|
||
|
|
fn f(_py: Python) -> PyResult<usize> {
|
||
|
|
Ok(CALL_COUNT.fetch_add(1, Relaxed))
|
||
|
|
}
|
||
|
|
|
||
|
|
let gil = Python::acquire_gil();
|
||
|
|
let py = gil.python();
|
||
|
|
let obj = py_fn!(f()).to_py_object(py);
|
||
|
|
|
||
|
|
assert_eq!(CALL_COUNT.load(Relaxed), 0);
|
||
|
|
assert_eq!(obj.call(py, NoArgs, None).unwrap().extract::<i32>(py).unwrap(), 0);
|
||
|
|
assert_eq!(obj.call(py, NoArgs, None).unwrap().extract::<i32>(py).unwrap(), 1);
|
||
|
|
assert_eq!(CALL_COUNT.load(Relaxed), 2);
|
||
|
|
assert!(obj.call(py, (1,), None).is_err());
|
||
|
|
assert_eq!(CALL_COUNT.load(Relaxed), 2);
|
||
|
|
assert_eq!(obj.call(py, NoArgs, Some(&PyDict::new(py))).unwrap().extract::<i32>(py).unwrap(), 2);
|
||
|
|
assert_eq!(CALL_COUNT.load(Relaxed), 3);
|
||
|
|
let dict = PyDict::new(py);
|
||
|
|
dict.set_item(py, "param", 42).unwrap();
|
||
|
|
assert!(obj.call(py, NoArgs, Some(&dict)).is_err());
|
||
|
|
assert_eq!(CALL_COUNT.load(Relaxed), 3);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn one_arg() {
|
||
|
|
fn f(_py: Python, i: usize) -> PyResult<usize> {
|
||
|
|
Ok(i * 2)
|
||
|
|
}
|
||
|
|
|
||
|
|
let gil = Python::acquire_gil();
|
||
|
|
let py = gil.python();
|
||
|
|
let obj = py_fn!(f(i: usize)).to_py_object(py);
|
||
|
|
|
||
|
|
assert!(obj.call(py, NoArgs, None).is_err());
|
||
|
|
assert_eq!(obj.call(py, (1,), None).unwrap().extract::<i32>(py).unwrap(), 2);
|
||
|
|
assert!(obj.call(py, (1, 2), None).is_err());
|
||
|
|
|
||
|
|
let dict = PyDict::new(py);
|
||
|
|
dict.set_item(py, "i", 42).unwrap();
|
||
|
|
assert_eq!(obj.call(py, NoArgs, Some(&dict)).unwrap().extract::<i32>(py).unwrap(), 84);
|
||
|
|
assert!(obj.call(py, (1,), Some(&dict)).is_err());
|
||
|
|
dict.set_item(py, "j", 10).unwrap();
|
||
|
|
assert!(obj.call(py, NoArgs, Some(&dict)).is_err());
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn flexible_sig() {
|
||
|
|
fn f(py: Python, args: &PyTuple, kwargs: Option<&PyDict>) -> PyResult<usize> {
|
||
|
|
Ok(args.len(py) + 100 * kwargs.map_or(0, |kwargs| kwargs.len(py)))
|
||
|
|
}
|
||
|
|
|
||
|
|
let gil = Python::acquire_gil();
|
||
|
|
let py = gil.python();
|
||
|
|
let obj = py_fn!(f).to_py_object(py);
|
||
|
|
|
||
|
|
assert_eq!(obj.call(py, NoArgs, None).unwrap().extract::<i32>(py).unwrap(), 0);
|
||
|
|
assert_eq!(obj.call(py, (1,), None).unwrap().extract::<i32>(py).unwrap(), 1);
|
||
|
|
assert_eq!(obj.call(py, (1,2), None).unwrap().extract::<i32>(py).unwrap(), 2);
|
||
|
|
|
||
|
|
let dict = PyDict::new(py);
|
||
|
|
dict.set_item(py, "i", 42).unwrap();
|
||
|
|
assert_eq!(obj.call(py, NoArgs, Some(&dict)).unwrap().extract::<i32>(py).unwrap(), 100);
|
||
|
|
assert_eq!(obj.call(py, (1,2), Some(&dict)).unwrap().extract::<i32>(py).unwrap(), 102);
|
||
|
|
dict.set_item(py, "j", 10).unwrap();
|
||
|
|
assert_eq!(obj.call(py, (1,2,3), Some(&dict)).unwrap().extract::<i32>(py).unwrap(), 203);
|
||
|
|
}
|
||
|
|
|