From e5535615c0b933afaa9dfc7fde2e72de07e36c77 Mon Sep 17 00:00:00 2001 From: Kirill Suvorov Date: Wed, 29 Jan 2025 16:26:53 +0100 Subject: [PATCH] fix tests --- src/bindings/js/node/src/node_output.cpp | 4 ++-- .../js/node/tests/unit/infer_request.test.js | 14 +++++--------- src/bindings/js/node/tests/utils.js | 4 ++-- 3 files changed, 9 insertions(+), 13 deletions(-) diff --git a/src/bindings/js/node/src/node_output.cpp b/src/bindings/js/node/src/node_output.cpp index d7ad1f1ebf8edb..3b3087f463a38c 100644 --- a/src/bindings/js/node/src/node_output.cpp +++ b/src/bindings/js/node/src/node_output.cpp @@ -45,7 +45,7 @@ Napi::Value Output::get_partial_shape(const Napi::CallbackInfo& info) } Napi::Value Output::get_any_name(const Napi::CallbackInfo& info) { - return Napi::String::New(info.Env(), _output.get_node()->get_name()); + return Napi::String::New(info.Env(), _output.get_any_name()); } Output::Output(const Napi::CallbackInfo& info) @@ -88,5 +88,5 @@ Napi::Value Output::get_partial_shape(const Napi::CallbackInfo& } Napi::Value Output::get_any_name(const Napi::CallbackInfo& info) { - return Napi::String::New(info.Env(), _output.get_node()->get_name()); + return Napi::String::New(info.Env(), _output.get_any_name()); } diff --git a/src/bindings/js/node/tests/unit/infer_request.test.js b/src/bindings/js/node/tests/unit/infer_request.test.js index a0a98952a6f2dd..b0593dba81e31d 100644 --- a/src/bindings/js/node/tests/unit/infer_request.test.js +++ b/src/bindings/js/node/tests/unit/infer_request.test.js @@ -338,8 +338,8 @@ describe('ov.InferRequest tests with missing outputs names', () => { const core = new ov.Core(); - let modelData = await fs.readFile(getModelPath(modelV3Small).xml, 'utf8'); - const weights = await fs.readFile(getModelPath(modelV3Small).bin); + let modelData = await fs.readFile(modelV3Small.xml, 'utf8'); + const weights = await fs.readFile(modelV3Small.bin); modelData = modelData.replace( 'names="MobilenetV3/Predictions/Softmax:0"', '' @@ -350,24 +350,20 @@ describe('ov.InferRequest tests with missing outputs names', () => { inferRequest = compiledModel.createInferRequest(); tensorData = Float32Array.from( - { length: 150528 }, + { length: lengthFromShape(modelV3Small.inputShape) }, () => Math.random() + epsilon, ); tensor = new ov.Tensor(ov.element.f32, modelV3Small.inputShape, tensorData); }); it('Test infer(inputData: Tensor[])', () => { - const outputLayer = compiledModel.outputs[0]; const result = inferRequest.infer([tensor]); - assert.deepStrictEqual(Object.keys(result), [outputLayer.toString()]); - assert.ok(result[outputLayer] instanceof ov.Tensor); + assert.deepStrictEqual(Object.keys(result).length, 1); }); it('Test inferAsync(inputData: Tensor[])', () => { inferRequest.inferAsync([tensor]).then((result) => { - const outputLayer = compiledModel.outputs[0]; - assert.deepStrictEqual(Object.keys(result), [outputLayer.toString()]); - assert.ok(result[outputLayer] instanceof ov.Tensor); + assert.deepStrictEqual(Object.keys(result).length, 1); }); }); }); diff --git a/src/bindings/js/node/tests/utils.js b/src/bindings/js/node/tests/utils.js index cd142e2d9dc8d9..243a39cef74264 100644 --- a/src/bindings/js/node/tests/utils.js +++ b/src/bindings/js/node/tests/utils.js @@ -27,8 +27,8 @@ const testModels = { 'https://media.githubusercontent.com/media/openvinotoolkit/testdata/master/models/test_model/test_model_fp32.bin', }, modelV3Small: { - xml: 'v3-small_224_1.0_float.xml', - bin: 'v3-small_224_1.0_float.bin', + xml: getModelPath('v3-small_224_1.0_float.xml'), + bin: getModelPath('v3-small_224_1.0_float.bin'), inputShape: [1, 224, 224, 3], outputShape: [1, 1001], xmlURL: