diff options
Diffstat (limited to 'lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py')
-rw-r--r-- | lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py b/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py new file mode 100644 index 00000000..a44f9027 --- /dev/null +++ b/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py | |||
@@ -0,0 +1,48 @@ | |||
1 | import os | ||
2 | script_path = os.path.dirname(os.path.realpath(__file__)) | ||
3 | files_path = os.path.join(script_path, '../../files/') | ||
4 | |||
5 | class DldtInferenceEngineTest(object): | ||
6 | ie_input_files = {'ie_python_sample': 'classification_sample.py', | ||
7 | 'input': 'chicky_512.png', | ||
8 | 'input_download': 'https://raw.githubusercontent.com/opencv/opencv/master/samples/data/chicky_512.png', | ||
9 | 'model': 'squeezenet_v1.1.xml'} | ||
10 | |||
11 | def __init__(self, target, work_dir): | ||
12 | self.target = target | ||
13 | self.work_dir = work_dir | ||
14 | |||
15 | def setup(self): | ||
16 | self.target.run('mkdir -p %s' % self.work_dir) | ||
17 | self.target.copy_to(os.path.join(files_path, 'dldt-inference-engine', self.ie_input_files['ie_python_sample']), | ||
18 | self.work_dir) | ||
19 | |||
20 | def tear_down(self): | ||
21 | self.target.run('rm -rf %s' % self.work_dir) | ||
22 | |||
23 | def test_can_download_input_file(self, proxy_port): | ||
24 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
25 | (self.work_dir, | ||
26 | self.ie_input_files['input_download'], | ||
27 | proxy_port)) | ||
28 | |||
29 | def test_dldt_ie_classification_with_device(self, device, ir_files_dir): | ||
30 | return self.target.run('classification_sample_async -d %s -i %s -m %s' % | ||
31 | (device, | ||
32 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
33 | os.path.join(ir_files_dir, self.ie_input_files['model']))) | ||
34 | |||
35 | def test_dldt_ie_classification_python_api_with_device(self, device, ir_files_dir, extension=''): | ||
36 | if extension: | ||
37 | return self.target.run('python3 %s -d %s -i %s -m %s -l %s' % | ||
38 | (os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']), | ||
39 | device, | ||
40 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
41 | os.path.join(ir_files_dir, self.ie_input_files['model']), | ||
42 | extension)) | ||
43 | else: | ||
44 | return self.target.run('python3 %s -d %s -i %s -m %s' % | ||
45 | (os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']), | ||
46 | device, | ||
47 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
48 | os.path.join(ir_files_dir, self.ie_input_files['model']))) | ||