initial commit

modules/python/test/test_copytomask.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+
+'''
+Test for copyto with mask
+'''
+
+# Python 2/3 compatibility
+from __future__ import print_function
+
+import cv2 as cv
+import numpy as np
+import sys
+
+from tests_common import NewOpenCVTests
+
+class copytomask_test(NewOpenCVTests):
+    def test_copytomask(self):
+
+        img = self.get_sample('python/images/baboon.png', cv.IMREAD_COLOR)
+        eps = 0.
+
+        #Create mask using inRange
+        valeurBGRinf = np.array([0,0,100])
+        valeurBGRSup = np.array([70, 70,255])
+        maskRed = cv.inRange(img, valeurBGRinf, valeurBGRSup)
+        #New binding
+        dstcv = np.full(np.array((2, 2, 1))*img.shape, 255, dtype=img.dtype)
+        cv.copyTo(img, maskRed, dstcv[:img.shape[0],:img.shape[1],:])
+        #using numpy
+        dstnp = np.full(np.array((2, 2, 1))*img.shape, 255, dtype=img.dtype)
+        mask2=maskRed.astype(bool)
+        _, mask_b = np.broadcast_arrays(img, mask2[..., None])
+        np.copyto(dstnp[:img.shape[0],:img.shape[1],:], img, where=mask_b)
+        self.assertEqual(cv.norm(dstnp ,dstcv), eps)
+
+
+if __name__ == '__main__':
+    NewOpenCVTests.bootstrap()