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