Use pyo3 as test harness

bschoenmaeckers · bschoenmaeckers · commit 16f80745adf3 · 2026-04-08T21:39:59.000+02:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/capi/.cargo/config.toml b/crates/capi/.cargo/config.toml
@@ -1,2 +1,2 @@
-[env]
-PYO3_CONFIG_FILE = { value = "pyo3-rustpython.config", relative = true }
+[profile.test.env]
+PYO3_CONFIG_FILE = { value = "pyo3-rustpython.config", relative = true }
diff --git a/crates/capi/Cargo.toml b/crates/capi/Cargo.toml
@@ -14,5 +14,8 @@ crate-type = ["staticlib"]
 [dependencies]
 rustpython-vm = { workspace = true, features = ["threading"]}
 
+[dev-dependencies]
+pyo3 = { version = "0.28.3", features = ["auto-initialize", "abi3-py314"] }
+
 [lints]
 workspace = true
diff --git a/crates/capi/pyo3-rustpython.config b/crates/capi/pyo3-rustpython.config
@@ -1,6 +1,6 @@
 implementation=CPython
 version=3.14
-shared=false
+shared=true
 abi3=true
 build_flags=
 suppress_build_script_link_lines=true
diff --git a/crates/capi/src/abstract_.rs b/crates/capi/src/abstract_.rs
@@ -75,3 +75,17 @@ pub extern "C" fn PyObject_VectorcallMethod(
             )
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::types::PyString;
+
+    #[test]
+    fn test_call_method1() {
+        Python::attach(|py| {
+            let string = PyString::new(py, "Hello, World!");
+            assert!(string.call_method1("endswith", ("!",)).unwrap().is_truthy().unwrap());
+        })
+    }
+}
diff --git a/crates/capi/src/bytesobject.rs b/crates/capi/src/bytesobject.rs
@@ -40,3 +40,17 @@ pub extern "C" fn PyBytes_AsString(bytes: *mut PyObject) -> *mut c_char {
         bytes.as_bytes().as_ptr() as _
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::types::PyBytes;
+
+    #[test]
+    fn test_bytes() {
+        Python::attach(|py| {
+            let bytes = PyBytes::new(py, b"Hello, World!");
+            assert_eq!(bytes.as_bytes(), b"Hello, World!");
+        })
+    }
+}
diff --git a/crates/capi/src/import.rs b/crates/capi/src/import.rs
@@ -15,3 +15,15 @@ pub extern "C" fn PyImport_Import(name: *mut PyObject) -> *mut PyObject {
         )
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+
+    #[test]
+    fn test_import() {
+        Python::attach(|py| {
+            let _module = py.import("sys").unwrap();
+        })
+    }
+}
diff --git a/crates/capi/src/longobject.rs b/crates/capi/src/longobject.rs
@@ -58,3 +58,18 @@ pub extern "C" fn PyLong_AsLong(obj: *mut PyObject) -> c_long {
         })
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::types::PyInt;
+
+    #[test]
+    fn test_py_int() {
+        Python::attach(|py| {
+            let number = PyInt::new(py, 123);
+            assert!(number.is_instance_of::<PyInt>());
+            assert_eq!(number.extract::<i32>().unwrap(), 123);
+        })
+    }
+}
diff --git a/crates/capi/src/object.rs b/crates/capi/src/object.rs
@@ -133,3 +133,31 @@ pub extern "C" fn PyObject_IsTrue(obj: *mut PyObject) -> c_int {
         )
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::types::PyString;
+
+    #[test]
+    fn test_is_truthy() {
+        Python::attach(|py| {
+            assert!(!py.None().is_truthy(py).unwrap());
+        })
+    }
+
+    #[test]
+    fn test_is_none() {
+        Python::attach(|py| {
+            assert!(py.None().is_none(py));
+        })
+    }
+
+    #[test]
+    fn test_type_name() {
+        Python::attach(|py| {
+            let string = PyString::new(py, "Hello, World!");
+            assert_eq!(string.get_type().name().unwrap().to_str().unwrap(), "str");
+        })
+    }
+}
diff --git a/crates/capi/src/pyerrors.rs b/crates/capi/src/pyerrors.rs
@@ -135,3 +135,17 @@ pub extern "C" fn PyException_GetTraceback(_exc: *mut PyObject) -> *mut PyObject
     crate::log_stub("PyException_GetTraceback");
     std::ptr::null_mut()
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::exceptions::PyTypeError;
+    use pyo3::prelude::*;
+    
+    #[test]
+    fn test_raised_exception() {
+        Python::attach(|py| {
+            PyTypeError::new_err("This is a type error").restore(py);
+            assert!(PyErr::take(py).is_some());
+        })
+    }
+}
diff --git a/crates/capi/src/pystate.rs b/crates/capi/src/pystate.rs
@@ -51,3 +51,25 @@ pub extern "C" fn PyEval_SaveThread() -> *mut PyThreadState {
 
 #[unsafe(no_mangle)]
 pub extern "C" fn PyEval_RestoreThread(_tstate: *mut PyThreadState) {}
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::types::PyInt;
+
+    #[test]
+    fn test_new_thread() {
+        Python::attach(|py| {
+            let number = PyInt::new(py, 123).unbind();
+            
+            std::thread::spawn(move || {
+                Python::attach(|py| {
+                    let number = number.bind(py);
+                    assert!(number.is_instance_of::<PyInt>());
+                });
+            })
+                .join()
+                .unwrap();
+        })
+    }
+}
diff --git a/crates/capi/src/refcount.rs b/crates/capi/src/refcount.rs
@@ -1,6 +1,7 @@
 use crate::PyObject;
 use rustpython_vm::PyObjectRef;
 use std::ptr::NonNull;
+use crate::pystate::with_vm;
 
 #[unsafe(no_mangle)]
 #[allow(clippy::not_unsafe_ptr_arg_deref)]
@@ -27,3 +28,29 @@ pub extern "C" fn _Py_IncRef(op: *mut PyObject) {
 
     std::mem::forget(owned);
 }
+
+#[unsafe(no_mangle)]
+pub extern "C" fn Py_REFCNT(op: *mut PyObject) -> isize {
+    with_vm(|vm| unsafe { &*op }
+        .strong_count() as _)
+}
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::PyTypeInfo;
+    use pyo3::types::PyInt;
+
+    #[test]
+    fn test_refcount() {
+        Python::attach(|py| {
+            let obj = PyInt::type_object(py);
+            unsafe { pyo3::ffi::Py_REFCNT(obj.as_ptr())};
+            let ref_count = obj.get_refcnt();
+            let obj_clone = obj.clone();
+            assert_eq!(obj.get_refcnt(), ref_count + 1);
+            drop(obj_clone);
+            assert_eq!(obj.get_refcnt(), ref_count);
+        });
+    }
+}
diff --git a/crates/capi/src/unicodeobject.rs b/crates/capi/src/unicodeobject.rs
@@ -64,3 +64,18 @@ pub extern "C" fn PyUnicode_AsEncodedString(
 pub extern "C" fn PyUnicode_InternInPlace(_string: *mut *mut PyObject) {
     crate::log_stub("PyUnicode_InternInPlace");
 }
+
+#[cfg(test)]
+mod tests {
+    use pyo3::prelude::*;
+    use pyo3::types::PyString;
+
+    #[test]
+    fn test_unicode() {
+        Python::attach(|py| {
+            let string = PyString::new(py, "Hello, World!");
+            assert!(string.is_instance_of::<PyString>());
+            assert_eq!(string.to_str().unwrap(), "Hello, World!");
+        })
+    }
+}
diff --git a/example_projects/pyo3_embed/Cargo.toml b/example_projects/pyo3_embed/Cargo.toml
diff --git a/example_projects/pyo3_embed/README.md b/example_projects/pyo3_embed/README.md
diff --git a/example_projects/pyo3_embed/src/main.rs b/example_projects/pyo3_embed/src/main.rs
