Added support for objects to SparseVector

Author: ankane

src/utils/sparse-vector.js

@@ -62,7 +62,8 @@ class SparseVector {
     this.indices = [];
     this.values = [];
 
-    for (const [i, v] of map.entries()) {
+    const entries = map instanceof Map ? map.entries() : Object.entries(map);
+    for (const [i, v] of entries) {
       if (v != 0) {
         this.indices.push(Number(i));
         this.values.push(Number(v));

tests/utils/sparse-vector.test.mjs

@@ -28,6 +28,14 @@ test('fromMap', () => {
   expect(vec.values).toStrictEqual([2, 3, 1]);
 });
 
+test('fromMapObject', () => {
+  const map = {2: 2, 4: 3, 0: 1, 3: 0};
+  const vec = new SparseVector(map, 6);
+  expect(vec.dimensions).toStrictEqual(6);
+  expect(vec.indices).toStrictEqual([0, 2, 4]);
+  expect(vec.values).toStrictEqual([1, 2, 3]);
+});
+
 test('toPostgres', () => {
   const vec = new SparseVector([1.23456789]);
   expect(vec.toPostgres()).toStrictEqual('{1:1.23456789}/1');