def test_iter_ngram_pad2(): pad = [8, 9] seq = [0, 1, 2, 3, 4] lst = list(iter_ngram_pad(seq, 3, 1, padding=pad)) assert_equal(lst, [[0, 8, 9], [1, 8, 9], [0, 1, 9], [2, 8, 9], [1, 2, 9], [0, 1, 2], [3, 8, 9], [2, 3, 9], [1, 2, 3], [4, 8, 9], [3, 4, 9], [2, 3, 4]])
def test_iter_ngram_pad3(): pad = [9] seq = [0, 1, 2, 3, 4] lst = list(iter_ngram_pad(seq, 3, 1, padding=pad))