Skip to content

Commit 11e6a93

Browse files
committed
add simple test
1 parent 657ce07 commit 11e6a93

1 file changed

Lines changed: 11 additions & 1 deletion

File tree

libsql-sqlite3/test/libsql_vector_index.test

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,9 +268,19 @@ do_execsql_test vector-transaction {
268268
SELECT * FROM vector_top_k('t_transaction_idx', vector('[1,2]'), 2);
269269
} {3 4 1 2}
270270

271+
do_execsql_test vector-1bit {
272+
CREATE TABLE t_1bit( v FLOAT32(3) );
273+
CREATE INDEX t_1bit_idx ON t_1bit( libsql_vector_idx(v, 'compress_neighbors=1bit') );
274+
INSERT INTO t_1bit VALUES (vector('[-1,-1,1]'));
275+
INSERT INTO t_1bit VALUES (vector('[-1,1,-1.5]'));
276+
INSERT INTO t_1bit VALUES (vector('[1,-1,-1]'));
277+
INSERT INTO t_1bit VALUES (vector('[-1,-1,-1]'));
278+
SELECT rowid FROM vector_top_k('t_1bit_idx', vector('[1,-1,-1]'), 4);
279+
} {3 4 2 1}
280+
271281
do_execsql_test vector-all-params {
272282
CREATE TABLE t_all_params ( emb FLOAT32(2) );
273-
CREATE INDEX t_all_params_idx ON t_all_params(libsql_vector_idx(emb, 'type=diskann', 'metric=cos', 'alpha=1.2', 'search_l=200', 'insert_l=70', 'max_neighbors=6'));
283+
CREATE INDEX t_all_params_idx ON t_all_params(libsql_vector_idx(emb, 'type=diskann', 'metric=cos', 'alpha=1.2', 'search_l=200', 'insert_l=70', 'max_neighbors=6', 'compress_neighbors=1bit'));
274284
INSERT INTO t_all_params VALUES (vector('[1,2]')), (vector('[3,4]'));
275285
SELECT * FROM vector_top_k('t_all_params_idx', vector('[1,2]'), 2);
276286
} {1 2}

0 commit comments

Comments
 (0)