From 3813e50e14a8bed891bd26a2a69fdef558408932 Mon Sep 17 00:00:00 2001 From: Yeoh Ee Peng Date: Thu, 9 Apr 2020 15:40:10 +0800 Subject: runtime/tests/dldt_inference_engine_test: Enable check available ie devices Enable checking of available inference engine devices. Signed-off-by: Yeoh Ee Peng Signed-off-by: Anuj Mittal --- lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py b/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py index a44f9027..31bfb539 100644 --- a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py +++ b/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py @@ -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, -- cgit v1.2.3-54-g00ecf