Esempio n. 1
0
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]])
Esempio n. 2
0
def test_iter_ngram_pad3():
    pad = [9]
    seq = [0, 1, 2, 3, 4]

    lst = list(iter_ngram_pad(seq, 3, 1, padding=pad))