Files
2026-07-13 12:48:55 +08:00

25 lines
798 B
JavaScript

import test from 'tape'
import nlp from './_lib.js'
test('misc ngrams', function (t) {
let doc = nlp(`quickly, suddenly`)
t.equal(doc.ngrams().length, 3, 'found three ngrams')
doc = nlp(`john, bill, joe`)
t.equal(doc.unigrams().length, 3, 'found three unigrams')
doc = nlp(`john, bill, joe`)
t.equal(doc.bigrams().length, 2, 'found 2 bigrams')
doc = nlp(`john, bill, joe`)
t.equal(doc.trigrams().length, 1, 'found 1 trigrams')
doc = nlp('i am in houston texas. i am a good person. so i think he is a good person.')
const arr = doc.endgrams({ size: 2 }) || []
t.equal(arr.length, 2, 'found 2 endgrams of size-2')
const first = arr[0] || {}
t.equal(first.normal, 'good person', 'found good person')
t.equal(first.count, 2, 'found 2 good person results')
t.end()
})