fix hnsw_vector test

lanterndata · Nov 3, 2024 · 475d861 · 475d861
1 parent caebe5c
commit 475d861
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/lantern_hnsw/test/expected/hnsw_vector.out b/lantern_hnsw/test/expected/hnsw_vector.out
@@ -198,28 +198,30 @@ SELECT id FROM small_world_arr ORDER BY v <-> ARRAY[0,0,0];
   3
 
 DROP INDEX l2_idx;
-CREATE INDEX cos_idx ON small_world_arr USING lantern_hnsw(v) WITH (m=2);
+CREATE INDEX cos_idx ON small_world_arr USING lantern_hnsw(v dist_cos_ops) WITH (m=2);
 INFO:  done init usearch index
 INFO:  inserted 3 elements
 INFO:  done saving 3 vectors
 SELECT id FROM small_world_arr ORDER BY v <=> ARRAY[0,0,0];
  id 
 ----
   1
-  2
   3
+  2
 
 DROP INDEX cos_idx;
-CREATE INDEX ham_idx ON small_world_arr USING lantern_hnsw(v) WITH (m=3);
+ALTER TABLE small_world_arr ADD COLUMN v_int INT[];
+UPDATE small_world_arr SET v_int=v::INT[];
+CREATE INDEX ham_idx ON small_world_arr USING lantern_hnsw(v_int dist_hamming_ops) WITH (m=3);
 INFO:  done init usearch index
 INFO:  inserted 3 elements
 INFO:  done saving 3 vectors
-SELECT id FROM small_world_arr ORDER BY v::INT[] <+> ARRAY[0,0,0];
+SELECT id FROM small_world_arr ORDER BY v_int <+> ARRAY[0,0,0];
  id 
 ----
   1
-  2
   3
+  2
 
 DROP TABLE small_world;
 \ir utils/small_world_vector.sql

diff --git a/lantern_hnsw/test/sql/hnsw_vector.sql b/lantern_hnsw/test/sql/hnsw_vector.sql
@@ -101,11 +101,13 @@ CREATE INDEX l2_idx ON small_world_arr USING lantern_hnsw(v) WITH (dim=3, m=2);
 EXPLAIN (COSTS FALSE) SELECT id FROM small_world_arr ORDER BY v <-> ARRAY[0,0,0];
 SELECT id FROM small_world_arr ORDER BY v <-> ARRAY[0,0,0];
 DROP INDEX l2_idx;
-CREATE INDEX cos_idx ON small_world_arr USING lantern_hnsw(v) WITH (m=2);
+CREATE INDEX cos_idx ON small_world_arr USING lantern_hnsw(v dist_cos_ops) WITH (m=2);
 SELECT id FROM small_world_arr ORDER BY v <=> ARRAY[0,0,0];
 DROP INDEX cos_idx;
-CREATE INDEX ham_idx ON small_world_arr USING lantern_hnsw(v) WITH (m=3);
-SELECT id FROM small_world_arr ORDER BY v::INT[] <+> ARRAY[0,0,0];
+ALTER TABLE small_world_arr ADD COLUMN v_int INT[];
+UPDATE small_world_arr SET v_int=v::INT[];
+CREATE INDEX ham_idx ON small_world_arr USING lantern_hnsw(v_int dist_hamming_ops) WITH (m=3);
+SELECT id FROM small_world_arr ORDER BY v_int <+> ARRAY[0,0,0];
 
 DROP TABLE small_world;
 \ir utils/small_world_vector.sql