securityos/node_modules/lunr/test/vector_test.js

suite('lunr.Vector', function () {
  var vectorFromArgs = function () {
    var vector = new lunr.Vector

    Array.prototype.slice.call(arguments)
      .forEach(function (el, i) {
        vector.insert(i, el)
      })

    return vector
  }

  suite('#magnitude', function () {
    test('calculates magnitude of a vector', function () {
      var vector = vectorFromArgs(4,5,6)
      assert.equal(Math.sqrt(77), vector.magnitude())
    })
  })

  suite('#dot', function () {
    test('calculates dot product of two vectors', function () {
      var v1 = vectorFromArgs(1, 3, -5),
          v2 = vectorFromArgs(4, -2, -1)

      assert.equal(3, v1.dot(v2))
    })
  })

  suite('#similarity', function () {
    test('calculates the similarity between two vectors', function () {
      var v1 = vectorFromArgs(1, 3, -5),
          v2 = vectorFromArgs(4, -2, -1)

      assert.approximately(v1.similarity(v2), 0.5, 0.01)
    })

    test('empty vector', function () {
      var vEmpty = new lunr.Vector,
          v1 = vectorFromArgs(1)

      assert.equal(0, vEmpty.similarity(v1))
      assert.equal(0, v1.similarity(vEmpty))
    })

    test('non-overlapping vector', function () {
      var v1 = new lunr.Vector([1, 1]),
          v2 = new lunr.Vector([2, 1])

      assert.equal(0, v1.similarity(v2))
      assert.equal(0, v2.similarity(v1))
    })
  })

  suite('#insert', function () {
    test('invalidates magnitude cache', function () {
      var vector = vectorFromArgs(4,5,6)

      assert.equal(Math.sqrt(77), vector.magnitude())

      vector.insert(3, 7)

      assert.equal(Math.sqrt(126), vector.magnitude())
    })

    test('keeps items in index specified order', function () {
      var vector = new lunr.Vector

      vector.insert(2, 4)
      vector.insert(1, 5)
      vector.insert(0, 6)

      assert.deepEqual([6,5,4], vector.toArray())
    })

    test('fails when duplicate entry', function () {
      var vector = vectorFromArgs(4, 5, 6)
      assert.throws(function () { vector.insert(0, 44) })
    })
  })

  suite('#upsert', function () {
    test('invalidates magnitude cache', function () {
      var vector = vectorFromArgs(4,5,6)

      assert.equal(Math.sqrt(77), vector.magnitude())

      vector.upsert(3, 7)

      assert.equal(Math.sqrt(126), vector.magnitude())
    })

    test('keeps items in index specified order', function () {
      var vector = new lunr.Vector

      vector.upsert(2, 4)
      vector.upsert(1, 5)
      vector.upsert(0, 6)

      assert.deepEqual([6,5,4], vector.toArray())
    })

    test('calls fn for value on duplicate', function () {
      var vector = vectorFromArgs(4, 5, 6)
      vector.upsert(0, 4, function (current, passed) { return current + passed })
      assert.deepEqual([8, 5, 6], vector.toArray())
    })
  })

  suite('#positionForIndex', function () {
    var vector = new lunr.Vector ([
        1, 'a',
        2, 'b',
        4, 'c',
        7, 'd',
        11, 'e'
    ])

    test('at the beginning', function () {
      assert.equal(0, vector.positionForIndex(0))
    })

    test('at the end', function () {
      assert.equal(10, vector.positionForIndex(20))
    })

    test('consecutive', function () {
      assert.equal(4, vector.positionForIndex(3))
    })

    test('non-consecutive gap after', function () {
      assert.equal(6, vector.positionForIndex(5))
    })

    test('non-consecutive gap before', function () {
      assert.equal(6, vector.positionForIndex(6))
    })

    test('non-consecutive gave before and after', function () {
      assert.equal(8, vector.positionForIndex(9))
    })

    test('duplicate at the beginning', function () {
      assert.equal(0, vector.positionForIndex(1))
    })

    test('duplicate at the end', function () {
      assert.equal(8, vector.positionForIndex(11))
    })

    test('duplicate consecutive', function () {
      assert.equal(4, vector.positionForIndex(4))
    })
  })
})