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

This commit is contained in:
Anastasia Kuporosova 2023-09-28 14:06:20 +02:00 committed by GitHub
parent 9a88eed9f2
commit 744f5d4130
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 1 deletions

View File

@ -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

View File

@ -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()