MIRRORS/meta-intel
1
0
Fork 0
mirror of git://git.yoctoproject.org/meta-intel.git synced 2025-07-05 05:04:45 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

runtime/dldt_inference_engine: Check MYRIAD device before test

Browse Source 
Checking MYRIAD device availability before execute MYRIAD test.

Signed-off-by: Yeoh Ee Peng <ee.peng.yeoh@intel.com>
Signed-off-by: Anuj Mittal <anuj.mittal@intel.com>
This commit is contained in:
Yeoh Ee Peng 2020-04-09 15:40:11 +08:00 committed by Anuj Mittal
parent 3813e50e14
commit e0a8e6e48f
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/oeqa/runtime/cases/dldt_inference_engine.py
View File

@ -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))