Skip to content

Commit e0035d6

Browse files
committed
add more tests
1 parent 1a704d8 commit e0035d6

2 files changed

Lines changed: 15 additions & 0 deletions

File tree

libsql-sqlite3/test/libsql_vector.test

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ do_execsql_test vector-1-func-valid {
5050
SELECT vector_distance_cos('[1,1]', '[-1,-1]');
5151
SELECT vector_distance_cos('[1,1]', '[-1,1]');
5252
SELECT vector_distance_cos('[1,2]', '[2,1]');
53+
SELECT vector_distance_cos(vector1bit('[10,-10]'), vector1bit('[-5,4]'));
54+
SELECT vector_distance_cos(vector1bit('[10,-10]'), vector1bit('[20,4]'));
55+
SELECT vector_distance_cos(vector1bit('[10,-10]'), vector1bit('[20,-2]'));
5356
} {
5457
{[]}
5558
{[]}
@@ -65,6 +68,9 @@ do_execsql_test vector-1-func-valid {
6568
{2.0}
6669
{1.0}
6770
{0.200000002980232}
71+
{2.0}
72+
{1.0}
73+
{0.0}
6874
}
6975

7076
do_execsql_test vector-1-conversion {

libsql-sqlite3/test/libsql_vector_index.test

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -327,6 +327,15 @@ do_execsql_test vector-partial {
327327
2 3 5 6 8 9
328328
}
329329

330+
do_execsql_test vector-1bit-table {
331+
CREATE TABLE t_1bit_table( v FLOAT1BIT(4) );
332+
INSERT INTO t_1bit_table VALUES ( vector1bit('[1,-1,1,-1]') );
333+
CREATE INDEX t_1bit_table_idx ON t_1bit_table( libsql_vector_idx(v) );
334+
INSERT INTO t_1bit_table VALUES ( vector1bit('[-1,1,1,-1]') );
335+
INSERT INTO t_1bit_table VALUES ( vector1bit('[1,-1,-1,1]') );
336+
SELECT * FROM vector_top_k('t_1bit_table_idx', vector1bit('[10,-10,-20,20]'), 4);
337+
} {3 1 2}
338+
330339
proc error_messages {sql} {
331340
set ret ""
332341
catch {

0 commit comments

Comments
 (0)