def test_model_cut_and_add_new_output():
model = init_model()
place = model.get_place_by_operation_name("")
model.cut_and_add_new_output(place=place, newName="5555")
stat = get_mdl_stat(model)
assert stat.cut_and_add_new_output == 1
assert stat.lastArgString == "5555"
assert stat.lastArgPlace == place
model.cut_and_add_new_output(place=place)
stat = get_mdl_stat(model)
assert stat.cut_and_add_new_output == 2
assert stat.lastArgString == ""
assert stat.lastArgPlace == place
def test_model_remove_output():
model = init_model()
place = model.get_place_by_operation_name("")
model.remove_output(place=place)
stat = get_mdl_stat(model)
assert stat.remove_output == 1
assert stat.lastArgPlace == place
def test_model_set_name_for_operation():
model = init_model()
place = model.get_place_by_operation_name(operationName="")
model.set_name_for_operation(operation=place, newName="1111")
stat = get_mdl_stat(model)
assert stat.set_name_for_operation == 1
assert stat.lastArgString == "1111"
assert stat.lastArgPlace == place
def test_model_add_name_for_tensor():
model = init_model()
place = model.get_place_by_tensor_name(tensorName="")
model.add_name_for_tensor(tensor=place, newName="1234")
stat = get_mdl_stat(model)
assert stat.add_name_for_tensor == 1
assert stat.lastArgString == "1234"
assert stat.lastArgPlace == place
def test_model_get_place_by_operation_name():
model = init_model()
for i in range(1, 10):
name = str(i)
model.get_place_by_operation_name(operationName=name)
stat = get_mdl_stat(model)
assert stat.get_place_by_operation_name == i
assert stat.lastArgString == name
def test_model_set_element_type():
model = init_model()
place = model.get_place_by_tensor_name(tensorName="")
model.set_element_type(place=place, type=get_element_type(np.int32))
stat = get_mdl_stat(model)
assert stat.set_element_type == 1
assert stat.lastArgPlace == place
assert stat.lastArgElementType == get_element_type(np.int32)
def test_model_get_place_by_tensor_name():
model = init_model()
for i in range(1, 10):
name = str(i)
model.get_place_by_tensor_name(tensor_name=name)
stat = get_mdl_stat()
assert stat.get_place_by_tensor_name == i
assert stat.lastArgString == name
def test_model_add_output():
model = init_model()
place = model.get_place_by_operation_name("")
place2 = model.add_output(place=place)
assert place2 is not None
stat = get_mdl_stat(model)
assert stat.add_output == 1
assert stat.lastArgPlace == place
def test_model_get_partial_shape():
model = init_model()
place = model.get_place_by_tensor_name(tensorName="")
shape = model.get_partial_shape(place=place)
assert shape is not None
stat = get_mdl_stat(model)
assert stat.get_partial_shape == 1
assert stat.lastArgPlace == place
def test_model_set_name_for_dimension():
model = init_model()
place = model.get_place_by_operation_name(operationName="")
model.set_name_for_dimension(place=place, dimIndex=123, dimName="4444")
stat = get_mdl_stat(model)
assert stat.set_name_for_dimension == 1
assert stat.lastArgString == "4444"
assert stat.lastArgInt == 123
assert stat.lastArgPlace == place
def test_model_set_partial_shape():
model = init_model()
place = model.get_place_by_tensor_name(tensorName="")
test_shape = PartialShape([1, 2, 3, 4])
model.set_partial_shape(place=place, shape=test_shape)
stat = get_mdl_stat(model)
assert stat.set_partial_shape == 1
assert stat.lastArgPlace == place
assert stat.lastArgPartialShape == test_shape
def test_model_get_place_by_operation_name_and_output_port():
model = init_model()
for i in range(1, 10):
name = str(i)
model.get_place_by_operation_name_and_output_port(operationName=name, outputPortIndex=i * 2)
stat = get_mdl_stat(model)
assert stat.get_place_by_operation_and_output_port == i
assert stat.lastArgString == name
assert stat.lastArgInt == i * 2
def test_model_override_all_outputs():
model = init_model()
place1 = model.get_place_by_tensor_name(tensorName="p1")
place2 = model.get_place_by_tensor_name(tensorName="p2")
model.override_all_outputs(outputs=[place1, place2])
stat = get_mdl_stat(model)
assert stat.override_all_outputs == 1
assert len(stat.lastArgOutputPlaces) == 2
assert stat.lastArgOutputPlaces[0] == place1
assert stat.lastArgOutputPlaces[1] == place2
def test_model_extract_subgraph():
model = init_model()
place1 = model.get_place_by_tensor_name(tensorName="p1")
place2 = model.get_place_by_tensor_name(tensorName="p2")
place3 = model.get_place_by_tensor_name(tensorName="p3")
place4 = model.get_place_by_tensor_name(tensorName="p4")
model.extract_subgraph(inputs=[place1, place2], outputs=[place3, place4])
stat = get_mdl_stat(model)
assert stat.extract_subgraph == 1
assert len(stat.lastArgInputPlaces) == 2
assert stat.lastArgInputPlaces[0] == place1
assert stat.lastArgInputPlaces[1] == place2
assert len(stat.lastArgOutputPlaces) == 2
assert stat.lastArgOutputPlaces[0] == place3
assert stat.lastArgOutputPlaces[1] == place4
def test_model_get_outputs():
model = init_model()
for i in range(1, 10):
model.get_outputs()
stat = get_mdl_stat(model)
assert stat.get_outputs == i
def test_model_free_name_for_tensor():
model = init_model()
model.free_name_for_tensor(name="2222")
stat = get_mdl_stat(model)
assert stat.free_name_for_tensor == 1
assert stat.lastArgString == "2222"
def test_model_free_name_for_operation():
model = init_model()
model.free_name_for_operation(name="3333")
stat = get_mdl_stat(model)
assert stat.free_name_for_operation == 1
assert stat.lastArgString == "3333"
