[JS API] Refactoring test utils (#28663)

### Details: - Utils moved to the `tests` directory for use from all tests folders. - Removed `getModelPath` from utils, model path is now available in `xml` and `bin` property. - Added `inputShape` and `outputShape` to model declaration and use them where needed. - Added `lengthFromShape` function to get required tensor size. --------- Signed-off-by: Kirill Suvorov <kirill.suvorov@intel.com>
openvinotoolkit · Jan 29, 2025 · fefbe56 · fefbe56
1 parent 90ae7ac
commit fefbe56
Show file tree

Hide file tree

Showing 13 changed files with 121 additions and 80 deletions.
diff --git a/src/bindings/js/.eslintrc-global.js b/src/bindings/js/.eslintrc-global.js
@@ -23,6 +23,7 @@ module.exports = {
     'key-spacing': ['error', { beforeColon: false }],
     'no-multiple-empty-lines': ['error', { max: 1, maxBOF: 0, maxEOF: 0 }],
     'keyword-spacing': ['error', { overrides: { catch: { after: false } } }],
+    'prefer-destructuring': ["error", { "object": true, "array": false }],
     '@typescript-eslint/no-var-requires': 0,
   }
 };
diff --git a/src/bindings/js/node/package.json b/src/bindings/js/node/package.json
@@ -18,7 +18,7 @@
     "build": "npm run tsc",
     "prepare": "npm run build",
     "lint": "eslint .",
-    "test_setup": "node ./tests/unit/setup.js",
+    "test_setup": "node ./tests/setup.js",
     "test": "npm run test_setup && node --test ./tests/unit/*.test.js",
     "test:e2e": "mocha ./tests/e2e/electron-app.test.js",
     "tsc": "tsc",

diff --git a/src/bindings/js/node/tests/e2e/electron-app.test.js b/src/bindings/js/node/tests/e2e/electron-app.test.js
@@ -4,7 +4,7 @@ const util = require('node:util');
 const assert = require('node:assert');
 const { exec } = require('child_process');
 const execPromise = util.promisify(exec);
-const { testModels, downloadTestModel } = require('../unit/utils.js');
+const { testModels, downloadTestModel } = require('../utils.js');
 
 describe('E2E testing for OpenVINO as an Electron dependency.', function() {
   this.timeout(50000);

diff --git a/src/bindings/js/node/tests/unit/setup.js → src/bindings/js/node/tests/setup.js b/src/bindings/js/node/tests/unit/setup.js → src/bindings/js/node/tests/setup.js
diff --git a/src/bindings/js/node/tests/unit/basic.test.js b/src/bindings/js/node/tests/unit/basic.test.js
@@ -11,14 +11,14 @@ const { after, describe, it, before, beforeEach } = require('node:test');
 const {
   testModels,
   compareModels,
-  getModelPath,
   isModelAvailable,
   sleep,
-} = require('./utils.js');
+  lengthFromShape,
+} = require('../utils.js');
 const epsilon = 0.5;
 
 describe('ov basic tests.', () => {
-  let testXml = null;
+  const { testModelFP32 } = testModels;
   let core = null;
   let model = null;
   let compiledModel = null;
@@ -27,13 +27,12 @@ describe('ov basic tests.', () => {
 
   before(async () => {
     outDir = await fs.promises.mkdtemp(path.join(os.tmpdir(), 'ov_js_out_'));
-    await isModelAvailable(testModels.testModelFP32);
-    testXml = getModelPath().xml;
+    await isModelAvailable(testModelFP32);
   });
 
   beforeEach(() => {
     core = new ov.Core();
-    model = core.readModelSync(testXml);
+    model = core.readModelSync(testModelFP32.xml);
     compiledModel = core.compileModelSync(model, 'CPU');
     modelLike = [model, compiledModel];
   });
@@ -139,12 +138,12 @@ describe('ov basic tests.', () => {
     });
 
     it('compileModelSync(model_path, deviceName, config: {}) ', () => {
-      const cm = core.compileModelSync(testXml, 'CPU', tput);
+      const cm = core.compileModelSync(testModelFP32.xml, 'CPU', tput);
       assert.equal(cm.inputs.length, 1);
     });
 
     it('compileModelSync(model:model_path, deviceName: string) ', () => {
-      const cm = core.compileModelSync(testXml, 'CPU');
+      const cm = core.compileModelSync(testModelFP32.xml, 'CPU');
       assert.deepStrictEqual(cm.output(0).shape, [1, 10]);
     });
 
@@ -200,13 +199,13 @@ describe('ov basic tests.', () => {
     });
 
     it('compileModel(model_path, deviceName, config: {}) ', () => {
-      core.compileModel(testXml, 'CPU', tput).then((cm) => {
+      core.compileModel(testModelFP32.xml, 'CPU', tput).then((cm) => {
         assert.equal(cm.inputs.length, 1);
       });
     });
 
     it('compileModel(model:model_path, deviceName: string) ', () => {
-      core.compileModel(testXml, 'CPU').then((cm) => {
+      core.compileModel(testModelFP32.xml, 'CPU').then((cm) => {
         assert.deepStrictEqual(cm.output(0).shape, [1, 10]);
       });
     });
@@ -277,8 +276,11 @@ describe('ov basic tests.', () => {
         assert.strictEqual(obj.input().getAnyName(), 'data');
         assert.strictEqual(obj.input().anyName, 'data');
 
-        assert.deepStrictEqual(obj.input(0).shape, [1, 3, 32, 32]);
-        assert.deepStrictEqual(obj.input(0).getShape(), [1, 3, 32, 32]);
+        assert.deepStrictEqual(obj.input(0).shape, testModelFP32.inputShape);
+        assert.deepStrictEqual(
+          obj.input(0).getShape(),
+          testModelFP32.inputShape,
+        );
       });
     });
   });
@@ -290,11 +292,11 @@ describe('ov basic tests.', () => {
 
     before(() => {
       tensor = Float32Array.from(
-        { length: 3072 },
+        { length: lengthFromShape(testModelFP32.inputShape) },
         () => Math.random() + epsilon,
       );
       const core = new ov.Core();
-      const model = core.readModelSync(testXml);
+      const model = core.readModelSync(testModelFP32.xml);
       const compiledModel = core.compileModelSync(model, 'CPU');
       userStream = compiledModel.exportModelSync();
       const inferRequest = compiledModel.createInferRequest();

diff --git a/src/bindings/js/node/tests/unit/compiled_model.test.js b/src/bindings/js/node/tests/unit/compiled_model.test.js
@@ -5,24 +5,27 @@
 const { addon: ov } = require('../..');
 const assert = require('assert');
 const { describe, it, before, beforeEach } = require('node:test');
-const { testModels, getModelPath, isModelAvailable } = require('./utils.js');
+const { testModels, isModelAvailable } = require('../utils.js');
 
 describe('ov.CompiledModel tests', () => {
-  let testXml = null;
+  const { testModelFP32 } = testModels;
   let core = null;
   let compiledModel = null;
 
   before(async () => {
-    await isModelAvailable(testModels.testModelFP32);
-    testXml = getModelPath().xml;
+    await isModelAvailable(testModelFP32);
     core = new ov.Core();
   });
 
   beforeEach(() => {
     const properties = {
       AUTO_BATCH_TIMEOUT: '1',
     };
-    compiledModel = core.compileModelSync(testXml, 'BATCH:CPU', properties);
+    compiledModel = core.compileModelSync(
+      testModelFP32.xml,
+      'BATCH:CPU',
+      properties
+    );
   });
 
   describe('getProperty()', () => {

diff --git a/src/bindings/js/node/tests/unit/core.test.js b/src/bindings/js/node/tests/unit/core.test.js
@@ -5,12 +5,13 @@
 const { addon: ov } = require('../..');
 const assert = require('assert');
 const { describe, it, before, beforeEach } = require('node:test');
-const { testModels, isModelAvailable, getModelPath } = require('./utils.js');
+const { testModels, isModelAvailable } = require('../utils.js');
 
 describe('ov.Core tests', () => {
+  const { testModelFP32 } = testModels;
   let core = null;
   before(async () => {
-    await isModelAvailable(testModels.testModelFP32);
+    await isModelAvailable(testModelFP32);
   });
 
   beforeEach(() => {
@@ -95,15 +96,15 @@ describe('ov.Core tests', () => {
   });
 
   it('Core.queryModel() with incorrect arguments should throw an error', () => {
-    const model = core.readModelSync(getModelPath().xml);
+    const model = core.readModelSync(testModelFP32.xml);
     assert.throws(
       () => core.queryModel(model, 'arg1', 'arg2').then(),
       /'queryModel' method called with incorrect parameters./,
     );
   });
 
   it('Core.queryModel() should have device in the result values', () => {
-    const model = core.readModelSync(getModelPath().xml);
+    const model = core.readModelSync(testModelFP32.xml);
     const device = 'CPU';
     const queryModel = core.queryModel(model, device);
     assert(Object.values(queryModel).includes(device));

diff --git a/src/bindings/js/node/tests/unit/infer_request.test.js b/src/bindings/js/node/tests/unit/infer_request.test.js
@@ -5,31 +5,43 @@
 const { addon: ov } = require('../..');
 const assert = require('assert');
 const { describe, it, before, beforeEach } = require('node:test');
-const { testModels, isModelAvailable, getModelPath } = require('./utils.js');
+const {
+  testModels,
+  isModelAvailable,
+  lengthFromShape,
+} = require('../utils.js');
 
 const epsilon = 0.5; // To avoid very small numbers
-const testXml = getModelPath().xml;
 
 describe('ov.InferRequest tests', () => {
+  const { testModelFP32 } = testModels;
   let compiledModel = null;
   let tensorData = null;
   let tensor = null;
   let resTensor = null;
   let tensorLike = null;
 
   before(async () => {
-    await isModelAvailable(testModels.testModelFP32);
+    await isModelAvailable(testModelFP32);
 
     const core = new ov.Core();
-    const model = core.readModelSync(testXml);
+    const model = core.readModelSync(testModelFP32.xml);
     compiledModel = core.compileModelSync(model, 'CPU');
 
     tensorData = Float32Array.from(
-      { length: 3072 },
+      { length: lengthFromShape(testModelFP32.inputShape) },
       () => Math.random() + epsilon,
     );
-    tensor = new ov.Tensor(ov.element.f32, [1, 3, 32, 32], tensorData);
-    resTensor = new ov.Tensor(ov.element.f32, [1, 10], tensorData.slice(-10));
+    tensor = new ov.Tensor(
+      ov.element.f32,
+      testModelFP32.inputShape,
+      tensorData,
+    );
+    resTensor = new ov.Tensor(
+      ov.element.f32,
+      testModelFP32.outputShape,
+      tensorData.slice(-10),
+    );
     tensorLike = [tensor, tensorData];
   });
 
@@ -43,15 +55,21 @@ describe('ov.InferRequest tests', () => {
       tensorLike.forEach((tl) => {
         const result = inferRequest.infer({ data: tl });
         assert.deepStrictEqual(Object.keys(result), ['fc_out']);
-        assert.deepStrictEqual(result['fc_out'].data.length, 10);
+        assert.deepStrictEqual(
+          result['fc_out'].data.length,
+          lengthFromShape(testModelFP32.outputShape)
+        );
       });
     });
 
     it('Test infer(inputData: Tensor[]/TypedArray[])', () => {
       tensorLike.forEach((tl) => {
         const result = inferRequest.infer([tl]);
         assert.deepStrictEqual(Object.keys(result), ['fc_out']);
-        assert.deepStrictEqual(result['fc_out'].data.length, 10);
+        assert.deepStrictEqual(
+          result['fc_out'].data.length,
+          lengthFromShape(testModelFP32.outputShape),
+        );
       });
     });
 
@@ -102,15 +120,21 @@ describe('ov.InferRequest tests', () => {
       inferRequest.inferAsync({ data: tensor }).then((result) => {
         assert.ok(result['fc_out'] instanceof ov.Tensor);
         assert.deepStrictEqual(Object.keys(result), ['fc_out']);
-        assert.deepStrictEqual(result['fc_out'].data.length, 10);
+        assert.deepStrictEqual(
+          result['fc_out'].data.length,
+          lengthFromShape(testModelFP32.outputShape),
+        );
       });
     });
 
     it('Test inferAsync(inputData: Tensor[])', () => {
       inferRequest.inferAsync([tensor]).then((result) => {
         assert.ok(result['fc_out'] instanceof ov.Tensor);
         assert.deepStrictEqual(Object.keys(result), ['fc_out']);
-        assert.deepStrictEqual(result['fc_out'].data.length, 10);
+        assert.deepStrictEqual(
+          result['fc_out'].data.length,
+          lengthFromShape(testModelFP32.outputShape),
+        );
       });
     });
 

diff --git a/src/bindings/js/node/tests/unit/model.test.js b/src/bindings/js/node/tests/unit/model.test.js
@@ -5,21 +5,20 @@
 const { addon: ov } = require('../..');
 const assert = require('assert');
 const { describe, it, before, beforeEach } = require('node:test');
-const { testModels, getModelPath, isModelAvailable } = require('./utils.js');
+const { testModels, isModelAvailable } = require('../utils.js');
 
 describe('ov.Model tests', () => {
-  let testXml = null;
+  const { testModelFP32 } = testModels;
   let core = null;
   let model = null;
 
   before(async () => {
-    await isModelAvailable(testModels.testModelFP32);
-    testXml = getModelPath().xml;
+    await isModelAvailable(testModelFP32);
     core = new ov.Core();
   });
 
   beforeEach(() => {
-    model = core.readModelSync(testXml);
+    model = core.readModelSync(testModelFP32.xml);
   });
 
   describe('Model.isDynamic()', () => {

diff --git a/src/bindings/js/node/tests/unit/pre_post_processor.test.js b/src/bindings/js/node/tests/unit/pre_post_processor.test.js
@@ -5,21 +5,20 @@
 const { addon: ov } = require('../..');
 const assert = require('assert');
 const { describe, it, before, beforeEach } = require('node:test');
-const { testModels, getModelPath, isModelAvailable } = require('./utils.js');
+const { testModels, isModelAvailable } = require('../utils.js');
 
 describe('ov.preprocess.PrePostProcessor tests', () => {
-  let testXml = null;
+  const { testModelFP32 } = testModels;
   let core = null;
   let model = null;
 
   before(async () => {
-    await isModelAvailable(testModels.testModelFP32);
-    testXml = getModelPath().xml;
+    await isModelAvailable(testModelFP32);
     core = new ov.Core();
   });
 
   beforeEach(() => {
-    model = core.readModelSync(testXml);
+    model = core.readModelSync(testModelFP32.xml);
   });
 
   describe('PrePostProcess', () => {

diff --git a/src/bindings/js/node/tests/unit/read_model.test.js b/src/bindings/js/node/tests/unit/read_model.test.js
@@ -6,22 +6,21 @@ const fs = require('node:fs');
 const { addon: ov } = require('../..');
 const assert = require('assert');
 const { describe, it, before, beforeEach } = require('node:test');
-const { testModels, isModelAvailable, getModelPath } = require('./utils.js');
-
-const { xml: modelPath, bin: weightsPath } = getModelPath();
+const { testModels, isModelAvailable } = require('../utils.js');
 
 describe('Tests for reading model.', () => {
+  const { testModelFP32 } = testModels;
   let modelFile = null;
   let modelStr = null;
   let weightsFile = null;
   let weightsTensor = null;
   let core = null;
 
   before(async () => {
-    await isModelAvailable(testModels.testModelFP32);
-    modelFile = fs.readFileSync(modelPath);
-    modelStr = fs.readFileSync(modelPath, 'utf8');
-    weightsFile = fs.readFileSync(weightsPath);
+    await isModelAvailable(testModelFP32);
+    modelFile = fs.readFileSync(testModelFP32.xml);
+    modelStr = fs.readFileSync(testModelFP32.xml, 'utf8');
+    weightsFile = fs.readFileSync(testModelFP32.bin);
   });
 
   beforeEach(() => {
@@ -35,13 +34,13 @@ describe('Tests for reading model.', () => {
 
   describe('Core.readModeSync', () => {
     it('readModeSync(xmlPath) ', () => {
-      const model = core.readModelSync(modelPath);
+      const model = core.readModelSync(testModelFP32.xml);
       assert.ok(model instanceof ov.Model);
       assert.equal(model.inputs.length, 1);
     });
 
     it('readModeSync(xmlPath, weightsPath) ', () => {
-      const model = core.readModelSync(modelPath, weightsPath);
+      const model = core.readModelSync(testModelFP32.xml, testModelFP32.bin);
       assert.ok(model instanceof ov.Model);
       assert.equal(model.inputs.length, 1);
     });
@@ -71,12 +70,12 @@ describe('Tests for reading model.', () => {
 
   describe('Core.readModel', () => {
     it('readModel(xmlPath) ', async () => {
-      const model = await core.readModel(modelPath);
+      const model = await core.readModel(testModelFP32.xml);
       assert.equal(model.inputs.length, 1);
     });
 
     it('readModel(xmlPath, weightsPath) ', async () => {
-      const model = await core.readModel(modelPath, weightsPath);
+      const model = await core.readModel(testModelFP32.xml, testModelFP32.bin);
       assert.equal(model.inputs.length, 1);
     });