added tests and docs for deep neural nets + added tests for new Mat utilities

Author: justadudewhohacks

test/tests/core/Mat/Mat.test.js

@@ -450,5 +450,77 @@ describe('Mat', () => {
       });
     });
   });
+
+  describe('findNonZero', () => {
+    const expectOutput = (res) => {
+      expect(res).to.be.an('array').lengthOf(3);
+    };
+
+    const mat = new cv.Mat([
+      [1, 0, 1],
+      [0, 1, 0]
+    ], cv.CV_8U);
+
+    generateAPITests({
+      getDut: () => mat,
+      methodName: 'findNonZero',
+      methodNameSpace: 'Mat',
+      expectOutput
+    });
+  });
+
+  describe('padToSquare', () => {
+    const expectOutput = (res) => {
+      expect(res).to.be.instanceOf(cv.Mat);
+      assertMetaData(res)(3, 3, cv.CV_8UC3);
+    };
+
+    describe('cols > rows', () => {
+      const mat = new cv.Mat([
+        [[255, 255, 255], [0, 0, 0], [255, 255, 255]],
+        [[0, 0, 0], [255, 255, 255], [0, 0, 0]]
+      ], cv.CV_8UC3);
+
+      generateAPITests({
+        getDut: () => mat,
+        getOptionalArgs: () => [new cv.Vec(255, 255, 255)],
+        methodName: 'padToSquare',
+        methodNameSpace: 'Mat',
+        expectOutput
+      });
+    });
+
+    describe('rows > cols', () => {
+      const mat = new cv.Mat([
+        [[255, 255, 255], [0, 0, 0]],
+        [[0, 0, 0], [255, 255, 255]],
+        [[0, 0, 0], [255, 255, 255]]
+      ], cv.CV_8UC3);
+
+      generateAPITests({
+        getDut: () => mat,
+        getOptionalArgs: () => [new cv.Vec(255, 255, 255)],
+        methodName: 'padToSquare',
+        methodNameSpace: 'Mat',
+        expectOutput
+      });
+    });
+
+    describe('rows === cols', () => {
+      const mat = new cv.Mat([
+        [[255, 255, 255], [0, 0, 0], [0, 0, 0]],
+        [[0, 0, 0], [255, 255, 255], [0, 0, 0]],
+        [[0, 0, 0], [255, 255, 255], [0, 0, 0]]
+      ], cv.CV_8UC3);
+
+      generateAPITests({
+        getDut: () => mat,
+        getOptionalArgs: () => [new cv.Vec(255, 255, 255)],
+        methodName: 'padToSquare',
+        methodNameSpace: 'Mat',
+        expectOutput
+      });
+    });
+  });
 });
 

test/tests/modules/dnn/NetTests.js

@@ -0,0 +1,40 @@
+const cv = global.dut;
+const { generateAPITests } = global.utils;
+const { expect } = require('chai');
+
+module.exports = (getTestImg) => {
+  let blob;
+  before(() => {
+    blob = cv.blobFromImage(getTestImg());
+  });
+
+  describe('setInput', () => {
+    const expectOutput = () => {
+      // expect to not throw
+    };
+
+    generateAPITests({
+      getDut: () => new cv.Net(),
+      methodName: 'setInput',
+      methodNameSpace: 'Net',
+      getRequiredArgs: () => ([
+        blob
+      ]),
+      expectOutput
+    });
+  });
+
+  // TODO: load an actual model in tests
+  describe.skip('forward', () => {
+    const expectOutput = (res) => {
+      expect(res).to.be.instanceOf(cv.Mat);
+    };
+
+    generateAPITests({
+      getDut: () => new cv.Net(),
+      methodName: 'forward',
+      methodNameSpace: 'Net',
+      expectOutput
+    });
+  });
+};

test/tests/modules/dnn/dnn.test.js

@@ -0,0 +1,66 @@
+const cv = global.dut;
+const {
+  readTestImage,
+  generateAPITests,
+  assertMetaData
+} = global.utils;
+const { expect } = require('chai');
+const NetTests = require('./NetTests');
+
+describe('dnn', () => {
+  if (!cv.xmodules.dnn) {
+    it('compiled without dnn');
+    return;
+  }
+
+  let testImg;
+
+  before(() => {
+    testImg = readTestImage().resizeToMax(250);
+  });
+
+  NetTests(() => testImg);
+
+  describe('blobFromImage', () => {
+    const expectOutput = (res) => {
+      expect(res).to.be.instanceOf(cv.Mat);
+      assertMetaData(res)(-1, -1, cv.CV_32F);
+    };
+
+    const getOptionalArgsMap = () => ([
+      ['scalefactor', 0.8],
+      ['size', new cv.Size(3, 3)],
+      ['mean', new cv.Vec(0.5, 0.5, 0.5)],
+      ['swapRB', true]
+    ]);
+
+    describe('blobFromImage', () => {
+      generateAPITests({
+        getDut: () => cv,
+        methodName: 'blobFromImage',
+        getRequiredArgs: () => ([
+          testImg
+        ]),
+        getOptionalArgsMap: () => ([
+          ['scalefactor', 0.8],
+          ['size', new cv.Size(3, 3)],
+          ['mean', new cv.Vec(0.5, 0.5, 0.5)],
+          ['swapRB', true]
+        ]),
+        expectOutput
+      });
+    });
+
+    describe('blobFromImages', () => {
+      generateAPITests({
+        getDut: () => cv,
+        methodName: 'blobFromImages',
+        getRequiredArgs: () => ([
+          [testImg, testImg]
+        ]),
+        getOptionalArgsMap,
+        expectOutput
+      });
+    });
+  });
+});