Exemplo n.º 1
0
def test_ngram2():
    result = list(ngrams(n=2, inp=range(4)))
    assert result == [[0, 1], [1, 2], [2, 3]]
Exemplo n.º 2
0
def test_ngram1():
    result = list(ngrams(n=1, inp=range(3)))
    assert result == [[0], [1], [2]]
Exemplo n.º 3
0
def test_ngram4():
    result = list(ngrams(n=3, inp=[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]))
    assert result == [[1, 1, 1]]
Exemplo n.º 4
0
def test_ngram_alpha():
    result = list(ngrams(n=3, inp="abc"))
    assert result == [
        ["a", "b", "c"],
    ]
Exemplo n.º 5
0
def test_ngram3():
    result = list(ngrams(n=3, inp=range(5)))
    assert result == [[0, 1, 2], [1, 2, 3], [2, 3, 4]]