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