def test_init_with_numpy_shared_memory(ov_type, numpy_dtype):
arr = read_image().astype(numpy_dtype)
shape = arr.shape
arr = np.ascontiguousarray(arr)
ov_tensor = Tensor(array=arr, shared_memory=True)
assert tuple(ov_tensor.shape) == shape
assert ov_tensor.element_type == ov_type
assert isinstance(ov_tensor.data, np.ndarray)
assert ov_tensor.data.dtype == numpy_dtype
assert ov_tensor.data.shape == shape
assert np.shares_memory(arr, ov_tensor.data)
assert np.array_equal(ov_tensor.data, arr)
assert ov_tensor.size == arr.size
assert ov_tensor.byte_size == arr.nbytes
assert tuple(ov_tensor.strides) == arr.strides
assert tuple(ov_tensor.get_shape()) == shape
assert ov_tensor.get_element_type() == ov_type
assert ov_tensor.get_size() == arr.size
assert ov_tensor.get_byte_size() == arr.nbytes
assert tuple(ov_tensor.get_strides()) == arr.strides
def test_cannot_get_strides_for_packed_tensor(ov_type):
ov_tensor = Tensor(ov_type, [1, 3, 48, 48])
with pytest.raises(RuntimeError) as e:
ov_tensor.get_strides()
assert "Could not get strides for types with bitwidths less then 8 bit." in str(
e.value)
