def test_fit_to_block_sequence_too_small(self):
        """Pad the sequence with 0 if the sequence is smaller than the block size."""
        sequence = [1, 2, 3, 4]
        expected_output = [1, 2, 3, 4, 0, 0, 0, 0, 0, 0]
        self.assertEqual(truncate_or_pad(sequence, self.block_size, 0), expected_output)
Pad the sequence with 0 if the sequence is smaller than the block size.
dpkd/transformers/examples/research_projects/bertabs/test_utils_summarization.py