Register mapping class in tests/test_proto_methods::mapping

PyO3 · Jun 24, 2022 · 6d17012 · 6d17012
1 parent 99413a9
commit 6d17012
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/tests/test_proto_methods.rs b/tests/test_proto_methods.rs
@@ -2,7 +2,7 @@
 
 use pyo3::exceptions::{PyAttributeError, PyIndexError, PyValueError};
 use pyo3::types::{PyDict, PyList, PyMapping, PySequence, PySlice, PyType};
-use pyo3::{prelude::*, py_run, PyCell};
+use pyo3::{prelude::*, py_run, PyCell, PyTypeInfo};
 use std::{isize, iter};
 
 mod common;
@@ -210,6 +210,15 @@ impl Mapping {
 #[test]
 fn mapping() {
     Python::with_gil(|py| {
+        PyModule::import(py, "collections.abc")
+            .unwrap()
+            .getattr("Mapping")
+            .unwrap()
+            .getattr("register")
+            .unwrap()
+            .call1((Mapping::type_object(py),))
+            .unwrap();
+
         let inst = Py::new(
             py,
             Mapping {
@@ -218,7 +227,6 @@ fn mapping() {
         )
         .unwrap();
 
-        //
         let mapping: &PyMapping = inst.as_ref(py).downcast().unwrap();
 
         py_assert!(py, inst, "len(inst) == 0");