Checking MYRIAD device availability before execute MYRIAD test.
Signed-off-by: Yeoh Ee Peng <[email protected]>
---
lib/oeqa/runtime/cases/dldt_inference_engine.py | 12 ++++++++++--
1 file changed, 10 insertions(+), 2 deletions(-)
diff --git a/lib/oeqa/runtime/cases/dldt_inference_engine.py
b/lib/oeqa/runtime/cases/dldt_inference_engine.py
index 2e96997..05aff61 100644
--- a/lib/oeqa/runtime/cases/dldt_inference_engine.py
+++ b/lib/oeqa/runtime/cases/dldt_inference_engine.py
@@ -67,7 +67,11 @@ class DldtInferenceEngine(OERuntimeTestCase):
@OEHasPackage(['dldt-inference-engine-samples'])
@OEHasPackage(['dldt-inference-engine-vpu-firmware'])
def test_dldt_ie_classification_with_myriad(self):
- (status, output) =
self.dldt_ie.test_dldt_ie_classification_with_device('MYRIAD',
self.ir_files_dir)
+ device = 'MYRIAD'
+ (status, output) =
self.dldt_ie.test_check_if_openvino_device_available(device)
+ if not status:
+ self.skipTest('OpenVINO %s device not available on target
machine(availalbe devices: %s)' % (device, output))
+ (status, output) =
self.dldt_ie.test_dldt_ie_classification_with_device(device, self.ir_files_dir)
self.assertEqual(status, 0, msg='status and output: %s and %s' %
(status, output))
@OETestDepends(['dldt_inference_engine.DldtInferenceEngine.test_dldt_ie_can_create_ir_and_download_input'])
@@ -97,5 +101,9 @@ class DldtInferenceEngine(OERuntimeTestCase):
@OEHasPackage(['python3-opencv'])
@OEHasPackage(['python3-numpy'])
def test_dldt_ie_classification_python_api_with_myriad(self):
- (status, output) =
self.dldt_ie.test_dldt_ie_classification_python_api_with_device('MYRIAD',
self.ir_files_dir)
+ device = 'MYRIAD'
+ (status, output) =
self.dldt_ie.test_check_if_openvino_device_available(device)
+ if not status:
+ self.skipTest('OpenVINO %s device not available on target
machine(availalbe devices: %s)' % (device, output))
+ (status, output) =
self.dldt_ie.test_dldt_ie_classification_python_api_with_device(device,
self.ir_files_dir)
self.assertEqual(status, 0, msg='status and output: %s and %s' %
(status, output))
--
2.7.4
-=-=-=-=-=-=-=-=-=-=-=-
Links: You receive all messages sent to this group.
View/Reply Online (#6470):
https://lists.yoctoproject.org/g/meta-intel/message/6470
Mute This Topic: https://lists.yoctoproject.org/mt/72891785/21656
Group Owner: [email protected]
Unsubscribe: https://lists.yoctoproject.org/g/meta-intel/unsub
[[email protected]]
-=-=-=-=-=-=-=-=-=-=-=-
