feat: add test for return by argument with an allocation

Author: edmBernard

tests/cpp/test.cpp

@@ -179,6 +179,14 @@ void returnByArgumentPointer(cv::Mat *mat) {
   mat->at<cv::Vec3w>(0, 0) = cv::Vec3w(4,5,6);
 }
 
+void returnByArgumentValueWithAllocation(cv::Mat mat) {
+  mat = cv::Mat::ones(cv::Size(3, 3), CV_8UC1);
+}
+
+void returnByArgumentRefWithAllocation(cv::Mat & mat) {
+  mat = cv::Mat::ones(cv::Size(3, 3), CV_8UC1);
+}
+
 PYBIND11_MODULE(test_module, m) {
 
   NDArrayConverter::init_numpy();
@@ -228,4 +236,8 @@ PYBIND11_MODULE(test_module, m) {
 
   m.def("returnByArgumentPointer", &returnByArgumentPointer);
 
+  m.def("returnByArgumentValueWithAllocation", &returnByArgumentValueWithAllocation);
+
+  m.def("returnByArgumentRefWithAllocation", &returnByArgumentRefWithAllocation);
+
 }

tests/test_return.py

@@ -53,11 +53,24 @@ def test_return_by_argument_by_value():
 def test_return_by_argument_by_ref():
   mat = generate_matrix()
   tm.returnByArgumentRef(mat)
-  assert(np.any(mat != generate_matrix()))
   assert(not check_matrix_content(mat))
 
 
 def test_return_by_argument_by_pointer():
   mat = generate_matrix()
   tm.returnByArgumentPointer(mat)
   assert(not check_matrix_content(mat))
+
+
+def test_return_by_argument_by_value():
+  mat = generate_matrix()
+  # During allocation we create a new buffer the data are not linked anymore
+  tm.returnByArgumentValueWithAllocation(mat)
+  assert(check_matrix_content(mat))
+
+
+def test_return_by_argument_by_ref():
+  mat = generate_matrix()
+  # During allocation we create a new buffer the data are not linked anymore
+  tm.returnByArgumentRefWithAllocation(mat)
+  assert(check_matrix_content(mat))