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

runtime/tests/dldt_inference_engine_test: Enable check available ie devices

Browse Source 
Enable checking of available inference engine devices.

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:10 +08:00 committed by Anuj Mittal
parent bd539ea962
commit 3813e50e14
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py
View File

@ -16,10 +16,18 @@ class DldtInferenceEngineTest(object):
self.target.run('mkdir -p %s' % self.work_dir)
self.target.copy_to(os.path.join(files_path, 'dldt-inference-engine', self.ie_input_files['ie_python_sample']),
self.work_dir)
python_cmd = 'from openvino.inference_engine import IENetwork, IECore; ie = IECore(); print(ie.available_devices)'
__, output = self.target.run('python3 -c "%s"' % python_cmd)
self.available_devices = output
def tear_down(self):
self.target.run('rm -rf %s' % self.work_dir)
def test_check_if_openvino_device_available(self, device):
if device not in self.available_devices:
return False, self.available_devices
return True, self.available_devices
def test_can_download_input_file(self, proxy_port):
return self.target.run('cd %s; wget %s -e https_proxy=%s' %
(self.work_dir,