fixup! test add

Author: himself65

test/parallel/test-sqlite.js

@@ -1,6 +1,7 @@
 // Flags: --experimental-sqlite
 'use strict';
 const { spawnPromisified } = require('../common');
+const assert = require('node:assert');
 const tmpdir = require('../common/tmpdir');
 const { existsSync } = require('node:fs');
 const { join } = require('node:path');
@@ -161,6 +162,50 @@ suite('DatabaseSync.prototype.loadExtension()', () => {
     const ext = getExtension(platform, arch);
     const filePath = path('sqlite', `vec0${ext}`);
     t.assert.strictEqual(db.loadExtension(filePath), undefined);
+
+    const { vec_version } = db.prepare(
+      'select vec_version() as vec_version;'
+    ).get();
+    assert.strictEqual(vec_version, 'v0.1.1');
+
+    const items = [
+      [1, [0.1, 0.1, 0.1, 0.1]],
+      [2, [0.2, 0.2, 0.2, 0.2]],
+      [3, [0.3, 0.3, 0.3, 0.3]],
+      [4, [0.4, 0.4, 0.4, 0.4]],
+      [5, [0.5, 0.5, 0.5, 0.5]],
+    ];
+    const query = [0.3, 0.3, 0.3, 0.3];
+
+    db.exec('CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4])');
+
+    const insertStmt = db.prepare(
+      'INSERT INTO vec_items(rowid, embedding) VALUES (?, ?)'
+    );
+
+    for (const [id, vector] of items) {
+      const rowId = BigInt(id);
+      const embedding = new Uint8Array(Float32Array.from(vector).buffer);
+      insertStmt.run(rowId, embedding);
+    }
+
+    const rows = db.prepare(
+      `
+  SELECT
+    rowid,
+    distance
+  FROM vec_items
+  WHERE embedding MATCH ?
+  ORDER BY distance
+  LIMIT 3
+`
+    ).all(new Uint8Array(Float32Array.from(query).buffer));
+
+    assert.deepStrictEqual(rows, [
+      { rowid: 3, distance: 0 },
+      { rowid: 4, distance: 0.19999998807907104 },
+      { rowid: 2, distance: 0.20000001788139343 },
+    ]);
   });
 });