4f0225014d
* Deprecated getInputTo, getCreatorLayer * Fixes * Fixed ie_layers moving to legacy * Fixed onnx importer dependency * Fixed python * Fix python API compilation * Added comments not to use _impl from Data Co-authored-by: Nadezhda Ageeva <nadezhda.ageeva@intel.com>
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import pytest
|
|
|
|
from openvino.inference_engine import IECore, IENetLayer, DataPtr
|
|
from conftest import model_path
|
|
|
|
|
|
test_net_xml, test_net_bin = model_path()
|
|
|
|
|
|
def layer_out_data():
|
|
ie = IECore()
|
|
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
|
|
return net.layers['19/Fused_Add_'].out_data[0]
|
|
|
|
|
|
def test_name():
|
|
assert layer_out_data().name == "19/Fused_Add_", "Incorrect name for layer '19/Fused_Add_'"
|
|
|
|
|
|
def test_precision():
|
|
assert layer_out_data().precision == "FP32", "Incorrect precision for layer '19/Fused_Add_'"
|
|
|
|
|
|
def test_precision_setter():
|
|
ie = IECore()
|
|
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
|
|
net.layers['19/Fused_Add_'].out_data[0].precision = "I8"
|
|
assert net.layers['19/Fused_Add_'].out_data[0].precision == "I8", "Incorrect precision for layer '19/Fused_Add_'"
|
|
|
|
|
|
def test_incorrect_precision_setter():
|
|
with pytest.raises(ValueError) as e:
|
|
layer_out_data().precision = "123"
|
|
assert "Unsupported precision 123! List of supported precisions:" in str(e.value)
|
|
|
|
|
|
def test_layout():
|
|
assert layer_out_data().layout == "NCHW", "Incorrect layout for layer '19/Fused_Add_'"
|
|
|
|
|
|
def test_initialized():
|
|
assert layer_out_data().initialized, "Incorrect value for initialized property for layer '19/Fused_Add_'"
|
|
|
|
|
|
def test_input_to():
|
|
ie = IECore()
|
|
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
|
|
input_to = net.layers['26'].out_data[0].input_to
|
|
assert len(input_to) == 1
|
|
assert input_to[0].name == '27'
|
|
|
|
def test_creator_layer():
|
|
ie = IECore()
|
|
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
|
|
outputs = net.outputs
|
|
assert len(outputs) == 1
|
|
creator_layer = outputs['fc_out'].creator_layer
|
|
params = creator_layer.params
|
|
params['originalLayersNames'] == 'fc_out'
|
|
params['axis'] == '1' |