def METHOD_NAME(self) -> None:
    generation_model = GenerationUtils(self.model)
    scores = torch.arange(0, 20).reshape(2, -1)
    top_k = 5
    expected = torch.zeros_like(scores)
    expected[0, 5:] = torch.arange(5, 10)
    expected[1, 5:] = torch.arange(15, 20)
    output = generation_model._get_top_k_restriction(scores, top_k)
    self.assertEqual(output, expected)
    top_k = 11
    output = generation_model._get_top_k_restriction(scores, top_k)
    self.assertEqual(output, scores)
    top_k = 0
    with self.assertRaises(ValueError):
        generation_model._get_top_k_restriction(scores, top_k)