def METHOD_NAME(self):
    tensors = [torch.ones([2]), torch.ones([4]), torch.ones([3])]
    max_lens = [6]
    expected_output = torch.tensor(
        [[1, 1, 0, 0, 0, 0], [1, 1, 1, 1, 0, 0], [1, 1, 1, 0, 0, 0]], dtype=torch.float32
    )
    stacked_tensor = stack_tensors(tensors=tensors, max_lens=max_lens)
    torch.testing.assert_close(stacked_tensor, expected_output)