Add testcase for unsendable, dict, weakref pyclass.

PyO3 · Jul 21, 2020 · 4563e00 · 4563e00
1 parent 43d1f43
commit 4563e00
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/tests/test_unsendable_dict.rs b/tests/test_unsendable_dict.rs
@@ -19,3 +19,27 @@ fn test_unsendable_dict() {
     let inst = Py::new(py, UnsendableDictClass {}).unwrap();
     py_run!(py, inst, "assert inst.__dict__ == {}");
 }
+
+#[pyclass(dict, unsendable, weakref)]
+struct UnsendableDictClassWithWeakRef {}
+
+#[pymethods]
+impl UnsendableDictClassWithWeakRef {
+    #[new]
+    fn new() -> Self {
+        Self {}
+    }
+}
+
+#[test]
+fn test_unsendable_dict_with_weakref() {
+    let gil = Python::acquire_gil();
+    let py = gil.python();
+    let inst = Py::new(py, UnsendableDictClassWithWeakRef {}).unwrap();
+    py_run!(py, inst, "assert inst.__dict__ == {}");
+    py_run!(
+        py,
+        inst,
+        "import weakref; assert weakref.ref(inst)() is inst; inst.a = 1; assert inst.a == 1"
+    );
+}