fix

Author: olpipi

src/bindings/python/tests/test_runtime/test_core.py

@@ -15,7 +15,6 @@
     PartialShape,
     CompiledModel,
     tensor_from_file,
-    read_tensor_data,
     compile_model,
     serialize,
 )
@@ -176,28 +175,6 @@ def test_read_model_from_tensor(request, tmp_path):
     assert isinstance(model, Model)
 
 
-def test_read_tensor_data_defaults_and_args(tmp_path):
-    data = np.arange(16, dtype=np.uint8)
-    bin_path = tmp_path / "tensor.bin"
-    data.tofile(bin_path)
-
-    tensor = read_tensor_data(bin_path)
-    assert isinstance(tensor, Tensor)
-    assert tensor.get_element_type() == Tensor(data).get_element_type()
-    assert tensor.get_shape() == [data.size]
-    assert np.array_equal(tensor.data, data)
-
-    tensor_offset = read_tensor_data(
-        bin_path,
-        element_type=Tensor(data).get_element_type(),
-        shape=PartialShape.dynamic(1),
-        offset_in_bytes=4,
-        mmap=False,
-    )
-    assert tensor_offset.get_shape() == [data.size - 4]
-    assert np.array_equal(tensor_offset.data, data[4:])
-
-
 def test_read_model_with_wrong_input():
     core = Core()
     with pytest.raises(TypeError) as e:

src/bindings/python/tests/test_runtime/test_read_tensor_data.py

@@ -95,10 +95,14 @@ def test_read_tensor_data_too_big_offset_throws(tmp_path: Path, mmap: bool) -> N
     file_size = path.stat().st_size
     assert file_size == data.nbytes
 
-    # offset + data_size > file_size
     with pytest.raises(RuntimeError):
-        ov.read_tensor_data(path, element_type=ov.Type.f32, shape=ov.PartialShape(list(shape)), offset_in_bytes=1, mmap=mmap)
-    # offset == file_size
+        ov.read_tensor_data(
+            path,
+            element_type=ov.Type.f32,
+            shape=ov.PartialShape(list(shape)),
+            offset_in_bytes=1,
+            mmap=mmap)
+
     with pytest.raises(RuntimeError):
         ov.read_tensor_data(
             path,
@@ -107,7 +111,7 @@ def test_read_tensor_data_too_big_offset_throws(tmp_path: Path, mmap: bool) -> N
             offset_in_bytes=file_size,
             mmap=mmap,
         )
-    # offset > file_size
+
     with pytest.raises(RuntimeError):
         ov.read_tensor_data(
             path,
@@ -118,6 +122,22 @@ def test_read_tensor_data_too_big_offset_throws(tmp_path: Path, mmap: bool) -> N
         )
 
 
+def test_read_tensor_data_default_all_args(tmp_path: Path) -> None:
+    """Test main use case: only path specified, all other args use defaults."""
+    data = np.arange(24, dtype=np.uint8)
+    path = tmp_path / "tensor.bin"
+    data.tofile(path)
+
+    # Main use case - only specify path
+    tensor = ov.read_tensor_data(path)
+
+    assert isinstance(tensor, ov.Tensor)
+    assert tensor.get_element_type() == ov.Type.u8  # default element type
+    assert tensor.get_shape() == [data.size]  # dynamic shape inferred from file
+    assert not tensor.data.flags.writeable  # read-only
+    assert np.array_equal(tensor.data, data)
+
+
 @pytest.mark.parametrize("mmap", [True, False])
 def test_read_tensor_data_dynamic_shape(tmp_path: Path, mmap: bool) -> None:
     shape = (1, 2, 3, 4)