IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix node name tests (#1507)

Browse Source
This commit is contained in:
Jan Iwaszkiewicz 2020-07-29 12:27:30 +02:00 committed by GitHub
parent eb8fe44de3
commit 13ae51930b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ngraph/python/tests/test_ngraph/test_basic.py
View File

@ -275,7 +275,6 @@ def test_result():
def test_node_friendly_name(): def test_node_friendly_name():
dummy_node = ng.parameter(shape=[1], name="dummy_name") dummy_node = ng.parameter(shape=[1], name="dummy_name")
assert(dummy_node.name == "Parameter_0")
assert(dummy_node.friendly_name == "dummy_name") assert(dummy_node.friendly_name == "dummy_name")
dummy_node.set_friendly_name("changed_name") dummy_node.set_friendly_name("changed_name")

2
ngraph/python/tests/test_ngraph/test_node_factory.py
View File

@ -44,7 +44,7 @@ def test_node_factory_wrapper_add():
assert node.get_type_name() == "Add" assert node.get_type_name() == "Add"
assert node.get_output_size() == 1 assert node.get_output_size() == 1
assert list(node.get_output_shape(0)) == [2, 2] assert list(node.get_output_shape(0)) == [2, 2]
assert node.name == "TestNode" assert node.friendly_name == "TestNode"
def test_node_factory_topk(): def test_node_factory_topk():