[PyOV] Skip tests with bool for arm (#15208)

* [PyOV] Skip tests with bool for arm

* exec_net export test rewrite
This commit is contained in:
Anastasia Kuporosova 2023-01-19 20:12:14 +01:00 committed by GitHub
parent 53672e7896
commit fda677f153
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 7 deletions

View File

@ -49,6 +49,11 @@ def create_simple_request_and_inputs(device):
def concat_model_with_data(device, ov_type, numpy_dtype):
core = Core()
# todo: remove as 101726 will be fixed.
if ov_type == Type.boolean and device == "CPU" and "Intel" not in core.get_property(device, "FULL_DEVICE_NAME"):
pytest.skip("This test runs only on openvino intel cpu plugin")
input_shape = [5]
params = []
@ -59,7 +64,6 @@ def concat_model_with_data(device, ov_type, numpy_dtype):
params += [ops.parameter(input_shape, numpy_dtype)]
model = Model(ops.concat(params, 0), params)
core = Core()
compiled_model = core.compile_model(model, device)
request = compiled_model.create_infer_request()
tensor1 = Tensor(ov_type, input_shape)

View File

@ -343,9 +343,6 @@ def test_properties_hint_model():
def test_single_property_setting(device):
core = Core()
if device == "CPU" and "Intel" not in core.get_property(device, "FULL_DEVICE_NAME"):
pytest.skip("This test runs only on openvino intel cpu plugin")
core.set_property(device, properties.streams.num(properties.streams.Num.AUTO))
assert properties.streams.Num.AUTO.to_integer() == -1

View File

@ -222,10 +222,11 @@ def test_exec_graph(device):
del ie_core
@pytest.mark.skipif(os.environ.get("TEST_DEVICE", "CPU") != "CPU",
reason="Device specific test. Only CPU plugin implements network export")
def test_export_import():
def test_export_import(device):
ie_core = ie.IECore()
if "EXPORT_IMPORT" not in ie_core.get_metric(device, "OPTIMIZATION_CAPABILITIES"):
pytest.skip(f"{ie_core.get_metric(device, 'FULL_DEVICE_NAME')} plugin due-to export, import model API isn't implemented.")
net = ie_core.read_network(model=test_net_xml, weights=test_net_bin)
exec_net = ie_core.load_network(net, "CPU")
exported_net_file = 'exported_model.bin'