def METHOD_NAME(
    self, config, input_ids, input_mask, decoder_input_ids, decoder_attention_mask
):
    model = ProphetNetForConditionalGeneration(config)
    model.eval()
    result = model(
        input_ids,
        attention_mask=input_mask,
        decoder_input_ids=decoder_input_ids,
        decoder_attention_mask=decoder_attention_mask,
    )
    self.parent.assertEqual(result[0].shape, [self.batch_size, self.seq_length, self.vocab_size])