Add pytest support

Author: yushulx

.github/workflows/main.yml

@@ -18,13 +18,13 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Run test.py in develop mode
+      - name: Run test in develop mode
         run: |
+          python -m pip install opencv-python pytest
           python setup.py develop
-          python -m pip install opencv-python
-          
           python --version
-          python test.py
+          pytest
+          
       - name: Build wheels for Linux
         if: matrix.os == 'ubuntu-latest'
         run: |

test.py

@@ -0,0 +1,104 @@
+from time import sleep
+import docscanner
+import numpy as np
+import cv2
+import time
+
+# set license
+docscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")
+
+scanner = docscanner.createInstance()
+
+ret = scanner.setParameters(docscanner.Templates.color)
+print(ret)
+
+def showNormalizedImage(name, normalized_image):
+    mat = docscanner.convertNormalizedImage2Mat(normalized_image)
+    cv2.imshow(name, mat)
+    return mat
+    
+# detectFile()
+def test_detectFile():
+    print('')
+    print('Test detectFile()')
+    results = scanner.detectFile("images/1.png")
+    assert len(results) > 0
+    image = cv2.imread("images/1.png")
+    for result in results:
+        x1 = result.x1
+        y1 = result.y1
+        x2 = result.x2
+        y2 = result.y2
+        x3 = result.x3
+        y3 = result.y3
+        x4 = result.x4
+        y4 = result.y4
+        
+        print(x1, y1, x2, y2, x3, y3, x4, y4)
+        
+        normalized_image = scanner.normalizeFile("images/1.png", x1, y1, x2, y2, x3, y3, x4, y4)
+        normalized_image.recycle()
+        
+    # assert len(results) > 0
+        # showNormalizedImage("Normalized Image", normalized_image)
+        # cv2.drawContours(image, [np.int0([(x1, y1), (x2, y2), (x3, y3), (x4, y4)])], 0, (0, 255, 0), 2)
+        
+    # cv2.imshow('Document Image', image)
+    # cv2.waitKey(0)
+
+
+# detectMat()
+def test_detectMat():
+    print('')
+    print('Test detectMat()')
+
+    image = cv2.imread("images/1.png")
+    results = scanner.detectMat(image)
+    assert len(results) > 0
+    for result in results:
+        x1 = result.x1
+        y1 = result.y1
+        x2 = result.x2
+        y2 = result.y2
+        x3 = result.x3
+        y3 = result.y3
+        x4 = result.x4
+        y4 = result.y4
+        
+        normalized_image = scanner.normalizeBuffer(image, x1, y1, x2, y2, x3, y3, x4, y4)
+        # showNormalizedImage("Normalized Image", normalized_image)
+        normalized_image.recycle()
+        cv2.drawContours(image, [np.int0([(x1, y1), (x2, y2), (x3, y3), (x4, y4)])], 0, (0, 255, 0), 2)
+        
+    # cv2.imshow('Document Image', image)
+    # cv2.waitKey(0)
+
+# detectMatAsync()
+def test_detectMatAsync():
+    print('')
+    print('Test detectMatAsync()')
+    def callback(results):
+        assert len(results) > 0
+        for result in results:
+            x1 = result.x1
+            y1 = result.y1
+            x2 = result.x2
+            y2 = result.y2
+            x3 = result.x3
+            y3 = result.y3
+            x4 = result.x4
+            y4 = result.y4
+            
+            cv2.drawContours(image, [np.int0([(x1, y1), (x2, y2), (x3, y3), (x4, y4)])], 0, (0, 255, 0), 2)
+        
+        # cv2.imshow('Document Image', image)  
+        # cv2.waitKey(0)
+        
+    import cv2
+    image = cv2.imread("images/1.png")
+    scanner.addAsyncListener(callback)
+    scanner.detectMatAsync(image)
+    sleep(3)
+
+
+