def test_get_metric_list_of_str():
ie = ov.Core()
param = ie.get_metric("CPU", "OPTIMIZATION_CAPABILITIES")
assert isinstance(param, list), "Parameter value for 'OPTIMIZATION_CAPABILITIES' " \
f"metric must be a list but {type(param)} is returned"
assert all(isinstance(v, str) for v in param), \
"Not all of the parameter values for 'OPTIMIZATION_CAPABILITIES' metric are strings!"
def test_register_plugin():
ie = ov.Core()
ie.register_plugin("MKLDNNPlugin", "BLA")
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
exec_net = ie.load_network(net, "BLA")
assert isinstance(exec_net, ov.ExecutableNetwork), \
"Cannot load the network to the registered plugin with name 'BLA'"
def test_ie_core_class():
input_shape = [1, 3, 4, 4]
param = ng.parameter(input_shape, np.float32, name="parameter")
relu = ng.relu(param, name="relu")
func = Function([relu], [param], "test")
func.get_ordered_ops()[2].friendly_name = "friendly"
cnn_network = ov.IENetwork(func)
ie_core = ov.Core()
ie_core.set_config({}, device_name="CPU")
executable_network = ie_core.load_network(cnn_network, "CPU", {})
td = TensorDesc("FP32", input_shape, "NCHW")
# from IPython import embed; embed()
request = executable_network.create_infer_request()
input_data = np.random.rand(*input_shape) - 0.5
expected_output = np.maximum(0.0, input_data)
input_blob = Blob(td, input_data)
request.set_input({"parameter": input_blob})
request.infer()
result = request.get_blob("relu").buffer
assert np.allclose(result, expected_output)
def test_read_network():
ie_core = ov.Core()
net = ie_core.read_network(model=test_net_xml, weights=test_net_bin)
assert isinstance(net, ov.IENetwork)
net = ie_core.read_network(model=test_net_xml)
assert isinstance(net, ov.IENetwork)
def test_get_metric_tuple_of_three_ints():
ie = ov.Core()
param = ie.get_metric("CPU", "RANGE_FOR_ASYNC_INFER_REQUESTS")
assert isinstance(param, tuple), "Parameter value for 'RANGE_FOR_ASYNC_INFER_REQUESTS' " \
f"metric must be tuple but {type(param)} is returned"
assert all(isinstance(v, int) for v in param), "Not all of the parameter values for " \
"'RANGE_FOR_ASYNC_INFER_REQUESTS' metric are integers!"
def test_read_net_from_buffer():
ie_core = ov.Core()
with open(test_net_bin, "rb") as f:
bin = f.read()
with open(model_path()[0], "rb") as f:
xml = f.read()
net = ie_core.read_network(model=xml, weights=bin)
assert isinstance(net, ov.IENetwork)
def test_query_network(device):
ie = ov.Core()
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
query_res = ie.query_network(network=net, device_name=device)
func_net = net.get_function()
ops_net = func_net.get_ordered_ops()
ops_net_names = [op.friendly_name for op in ops_net]
assert [key for key in query_res.keys() if key not in ops_net_names] == [], \
"Not all network layers present in query_network results"
assert next(iter(set(
query_res.values()))) == device, "Wrong device for some layers"
def test_read_network_as_path():
ie_core = ov.Core()
net = ie_core.read_network(model=Path(test_net_xml),
weights=Path(test_net_bin))
assert isinstance(net, ov.IENetwork)
net = ie_core.read_network(model=test_net_xml, weights=Path(test_net_bin))
assert isinstance(net, ov.IENetwork)
net = ie_core.read_network(model=Path(test_net_xml))
assert isinstance(net, ov.IENetwork)
def test_read_network_from_blob_valid():
ie_core = ov.Core()
model = open(test_net_xml).read()
blob = ov.blob_from_file(test_net_bin)
net = ie_core.read_network(model=model, blob=blob)
ref_net = ie_core.read_network(model=test_net_xml, weights=test_net_bin)
assert net.name == ref_net.name
assert net.batch_size == ref_net.batch_size
ii_net = net.input_info
ii_net2 = ref_net.input_info
o_net = net.outputs
o_net2 = ref_net.outputs
assert ii_net.keys() == ii_net2.keys()
assert o_net.keys() == o_net2.keys()
def test_get_version(device):
ie = ov.Core()
version = ie.get_versions(device)
assert isinstance(version, dict), "Returned version must be a dictionary"
assert device in version, "{} plugin version wasn't found in versions"
assert hasattr(version[device],
"major"), "Returned version has no field 'major'"
assert hasattr(version[device],
"minor"), "Returned version has no field 'minor'"
assert hasattr(
version[device],
"description"), "Returned version has no field 'description'"
assert hasattr(
version[device],
"build_number"), "Returned version has no field 'build_number'"
def test_register_plugins():
ie = ov.Core()
if platform == "linux" or platform == "linux2":
ie.register_plugins(plugins_xml)
elif platform == "darwin":
ie.register_plugins(plugins_osx_xml)
elif platform == "win32":
ie.register_plugins(plugins_win_xml)
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
exec_net = ie.load_network(net, "CUSTOM")
assert isinstance(exec_net,
ov.ExecutableNetwork), "Cannot load the network to " \
"the registered plugin with name 'CUSTOM' " \
"registred in the XML file"
def test_net_from_buffer_valid():
ie_core = ov.Core()
with open(test_net_bin, "rb") as f:
bin = f.read()
with open(model_path()[0], "rb") as f:
xml = f.read()
net = ie_core.read_network(model=xml, weights=bin)
ref_net = ie_core.read_network(model=test_net_xml, weights=test_net_bin)
assert net.name == ref_net.name
assert net.batch_size == ref_net.batch_size
ii_net = net.input_info
ii_net2 = ref_net.input_info
o_net = net.outputs
o_net2 = ref_net.outputs
assert ii_net.keys() == ii_net2.keys()
assert o_net.keys() == o_net2.keys()
def test_get_metric_str():
ie = ov.Core()
param = ie.get_metric("CPU", "FULL_DEVICE_NAME")
assert isinstance(param, str), "Parameter value for 'FULL_DEVICE_NAME' " \
f"metric must be string but {type(param)} is returned"
def test_read_network_from_blob():
ie_core = ov.Core()
model = open(test_net_xml).read()
blob = ov.blob_from_file(test_net_bin)
net = ie_core.read_network(model=model, blob=blob)
assert isinstance(net, ov.IENetwork)
def test_load_network(device):
ie = ov.Core()
net = ie.read_network(model=test_net_xml, weights=test_net_bin)
exec_net = ie.load_network(net, device)
assert isinstance(exec_net, ov.ExecutableNetwork)
def test_available_devices(device):
ie = ov.Core()
devices = ie.available_devices
assert device in devices, f"Current device '{device}' is not listed in " \
f"available devices '{', '.join(devices)}'"
def test_get_config():
ie = ov.Core()
conf = ie.get_config("CPU", "CPU_BIND_THREAD")
assert conf == "YES"
def test_read_network_from_onnx_as_path():
ie_core = ov.Core()
net = ie_core.read_network(model=Path(test_net_onnx))
assert isinstance(net, ov.IENetwork)
