adding test_promotion_of_disowned_to_shared

pybind · Dec 4, 2020 · 1f4f5e2 · 1f4f5e2
1 parent b96ceac
commit 1f4f5e2
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/tests/test_variant_unique_shared.cpp b/tests/test_variant_unique_shared.cpp
@@ -40,8 +40,13 @@ TEST_SUBMODULE(variant_unique_shared, m) {
                  v.get_unique();
                  return;
              })
-        .def("get_shared", [](vptr<double> &v) {
-            v.get_shared();
+        .def("get_shared",
+             [](vptr<double> &v) {
+                 v.get_shared();
+                 return;
+             })
+        .def("disown_unique", [](vptr<double> &v) {
+            v.get_unique().reset();
             return;
         });
 }

diff --git a/tests/test_variant_unique_shared.py b/tests/test_variant_unique_shared.py
@@ -48,3 +48,14 @@ def test_shared_from_birth():
         v.get_unique()
     assert str(exc_info.value) == "get_unique failure."
     v.get_shared()  # Still works.
+
+
+def test_promotion_of_disowned_to_shared():
+    v = m.from_unique()
+    assert v.get_value() == 5
+    v.disown_unique()
+    assert v.ownership_type() == 0
+    assert v.get_value() == -1
+    v.get_shared()  # Promotion of disowned to shared_ptr.
+    assert v.ownership_type() == 1
+    assert v.get_value() == -1