[PyOV] Fix docs and add test for Model._get_raw_address (#20109)

This commit is contained in:
Anastasia Kuporosova authored and GitHub committed 2023-09-28 12:06:20 +00:00
1 parent 9a88eed9f2
commit 744f5d4130
2 files changed
+12 -1

No files matched your search

@@ -1004,7 +1004,9 @@ void regclass_graph_Model(py::module m) {
return reinterpret_cast<uint64_t>(&self);
},
R"(
Returns raw address of the Model object.
Returns a raw address of the Model object from C++.
Use this function in order to compare underlying C++ addresses instead of using `__eq__` in Python.
:return: a raw address of the Model object.
:rtype: int
@@ -657,3 +657,12 @@ def test_model_add_remove_result_parameter_sink():
assert ["Assign"] == [sink.get_type_name() for sink in assign_nodes]
model.remove_sink(assign)
assert len(model.sinks) == 0
def test_model_get_raw_address():
model = generate_add_model()
model_with_same_addr = model
model_different = generate_add_model()
assert model._get_raw_address() == model_with_same_addr._get_raw_address()
assert model._get_raw_address() != model_different._get_raw_address()