faiss-napi/test/IndexHNSW.test.js

JavaScript NAPI bindings for faiss

const { IndexHNSW, MetricType } = require('..');

describe('IndexHNSW', () => {
  describe('#constructor', () => {
    it('1 arg will result in index with default neighbors & metric', () => {
      const index = new IndexHNSW(2);
      expect(index.dims).toBe(2);
    });

    it('2 args will result in index with default metric', () => {
      const index = new IndexHNSW(2, 20);
      expect(index.dims).toBe(2);
    });

    it('3 args will result in index', () => {
      const index = new IndexHNSW(2, 20, MetricType.METRIC_INNER_PRODUCT);
      expect(index.dims).toBe(2);
    });
  });
});