def test_decode_name_with_port_noname(self):
with self.assertRaisesRegex(Error,
'No\\ node\\ with\\ name.*mocknoname*'):
decode_name_with_port(self.model, 'mocknoname')
model_stat = get_model_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 1
def test_decode_name_with_port_collision_op_tensor(self):
with self.assertRaisesRegex(Error, 'Name\\ collision.*tensorAndOp*'):
decode_name_with_port(self.model, 'tensorAndOp')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 1
assert place_stat.is_equal_data > 0
def test_decode_name_with_port_delim_no_port_in(self):
with self.assertRaisesRegex(Error, 'No\\ node\\ with\\ name.*1234\\:operation*'):
decode_name_with_port(self.model, '1234:operation')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 2
assert place_stat.get_input_port == 1
assert place_stat.lastArgInt == 1234
def test_decode_name_with_port_delim_op_collision_in(self):
with self.assertRaisesRegex(Error, 'Name\\ collision(?!.*Tensor.*).*0\\:operation*'):
decode_name_with_port(self.model, '0:operation')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 2
assert place_stat.is_equal_data > 0
assert place_stat.get_input_port == 1
assert place_stat.lastArgInt == 0
def test_decode_name_with_port_op(self):
node = decode_name_with_port(self.model, "operation")
model_stat = get_model_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 1
assert node
def test_decode_name_with_port_delim_tensor_no_collision_in(self):
node = decode_name_with_port(self.model, '0:tensor')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 2
assert place_stat.get_input_port == 0
assert node
def test_decode_name_with_port_delim_op_out(self):
node = decode_name_with_port(self.model, 'operation:7')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 2
assert place_stat.get_output_port == 1
assert place_stat.lastArgInt == 7
assert node
def test_decode_name_with_port_delim_equal_data_in(self):
set_equal_data('conv2d', 'conv2d')
node = decode_name_with_port(self.model, '0:conv2d')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 2
assert place_stat.get_input_port == 1
assert place_stat.is_equal_data > 0
assert node
def test_decode_name_with_port_delim_all_same_data(self):
set_equal_data('8', '9')
node = decode_name_with_port(self.model, '8:9')
model_stat = get_model_statistic()
place_stat = get_place_statistic()
assert model_stat.get_place_by_tensor_name == 1
assert model_stat.get_place_by_operation_name == 3
assert place_stat.get_input_port == 1
assert place_stat.get_output_port == 1
# At least 3 comparisons of places are expected
assert place_stat.is_equal_data > 2
assert node
