clustring/src/main.test.js

Algorithms for clustering strings

import { clusterByKey, clusterByKnn } from './main'
import fingerprint from './key/fingerprint'
import levenshtein from './knn/levenshtein'

describe('main.js', () => {
  it('should handle README clusterByKey example', async () => {
    const bucket = {
      "commonWord": 3,
      "CommonWord": 20,
      "SuperRareWord": 1
    }

    const bins = await clusterByKey(bucket, fingerprint()).cluster()
    expect(bins).toEqual([
      {
        "name": "CommonWord",
        "key": "commonword",
        "count": 23,
        "bucket": { "commonWord": 3, "CommonWord": 20 }
      }
    ])
  })

  it('should handle README clusterByKnn example', async () => {
    const bucket = {
      "commonWord": 3,
      "CommonWord": 20,
      "SuperRareWord": 1
    }

    const bins = await clusterByKnn(bucket, levenshtein(2), 2).cluster()
    expect(bins).toEqual([
      {
        "name": "CommonWord",
        "count": 23,
        "bucket": { "commonWord": 3, "CommonWord": 20 }
      }
    ])
  })
})