1
0
mirror of https://github.com/sqlite/sqlite.git synced 2026-10-23 21:49:02 +03:00
Files
sqlite/ext/vec1/test/vec1aa.test
dan e32fbe0c23 Import vec1 sources into directory ext/vec1/ of this project.
FossilOrigin-Name: 3a0876d973868be7b6f4156a434cd09098c7ab1abdc7f0ef4f3d12a03de68b24
2026-09-30 14:49:16 +00:00

265 lines
6.7 KiB
Plaintext

# 2026-01-17
#
# The author disclaims copyright to this source code. In place of
# a legal notice, here is a blessing:
#
# May you do good and not evil.
# May you find forgiveness for yourself and forgive others.
# May you share freely, never taking more than you give.
#
#***********************************************************************
#
#
source [file join [file dirname [info script]] vec1_common.tcl]
set testprefix vec1aa
vec1_load db
do_execsql_test 1.0 {
CREATE VIRTUAL TABLE v1 USING vec1;
}
do_execsql_test 1.0.1 { SELECT vec1_config('nthread'); } {1}
do_execsql_test 1.0.2 { SELECT vec1_config('nprobe'); } {0.05}
do_execsql_test 1.1 {
INSERT INTO v1(cmd, vector) VALUES('rebuild', '{index:"flat"}');
INSERT INTO v1(rowid, vector) VALUES(
1, vec1_from_json('[1,2,3,4,5,6,7,8]')
);
}
do_execsql_test 1.2 { PRAGMA integrity_check } {ok}
do_execsql_test 1.3 {
DELETE FROM v1_base;
PRAGMA integrity_check;
} {{v1: %_idx row 1 missing from %_base}}
do_execsql_test 1.4 {
DROP TABLE v1;
}
do_execsql_test 1.5 {
CREATE VIRTUAL TABLE v1 USING vec1;
INSERT INTO v1(cmd, vector) VALUES('rebuild', '{index:"flat"}');
INSERT INTO v1(rowid, vector) VALUES(1, vec1_from_json('[1,2,3,4]'));
BEGIN;
SAVEPOINT aaa;
INSERT INTO v1(rowid, vector) VALUES(2, vec1_from_json('[1,2,3,4]'));
ROLLBACK TO aaa;
COMMIT;
PRAGMA integrity_check;
} {ok}
do_execsql_test 1.6 {
BEGIN;
INSERT INTO v1(rowid, vector) VALUES(3, vec1_from_json('[1,2,3,4]'));
ROLLBACK;
INSERT INTO v1(rowid, vector) VALUES(4, vec1_from_json('[1,2,3,4]'));
PRAGMA integrity_check;
} {ok}
#-------------------------------------------------------------------------
#reset_db
#vec1_load db
#do_execsql_test 2.0 {
# CREATE VIRTUAL TABLE v1 USING vec1;
# INSERT INTO v1(cmd, vector) VALUES('rebuild', 'flat');
# INSERT INTO v1(rowid, vector) VALUES(1, vec1_from_json('[1,2,3,4]'));
# INSERT INTO v1(rowid, vector) VALUES(2, vec1_from_json('[4,3,2,1]'));
# SELECT count(*) fROM v1_idx;
#} {1}
#-------------------------------------------------------------------------
reset_db
vec1_load db
do_execsql_test 3.0 {
CREATE VIRTUAL TABLE v1 USING vec1;
INSERT INTO v1(cmd, vector) VALUES('rebuild', '{index:"flat"}');
BEGIN;
INSERT INTO v1(rowid, vector) VALUES(1, random_vector(8));
INSERT INTO v1(rowid, vector) VALUES(2, random_vector(8));
INSERT INTO v1(rowid, vector) VALUES(3, random_vector(8));
COMMIT;
PRAGMA integrity_check;
} {ok}
do_execsql_test 3.1 {
DELETE FROM v1 WHERE rowid=2;
PRAGMA integrity_check;
} {ok}
do_execsql_test 3.2 {
SELECT rowid FROM v1
} {1 3}
do_execsql_test 3.3 {
SELECT rowid FROM v1( random_vector(8), 5 ) ORDER BY 1;
} {1 3}
do_execsql_test 3.4 {
WITH s(i) AS ( SELECT 1 UNION ALL SELECT i+1 FROM s WHERE i<100 )
INSERT INTO v1 SELECT random_vector(8);
WITH s(i) AS ( SELECT 1 UNION ALL SELECT i+1 FROM s WHERE i<100 )
INSERT INTO v1 SELECT random_vector(8);
WITH s(i) AS ( SELECT 1 UNION ALL SELECT i+1 FROM s WHERE i<100 )
INSERT INTO v1 SELECT random_vector(8);
}
do_execsql_test 3.5 {
DELETE FROM v1 WHERE rowid IN (5, 105, 205, 77, 98, 222);
PRAGMA integrity_check;
} {ok}
#-------------------------------------------------------------------------
reset_db
vec1_load db
do_execsql_test 4.0 {
CREATE VIRTUAL TABLE v1 USING vec1;
INSERT INTO v1(cmd, vector) VALUES('rebuild', '{index:"flat"}');
}
do_execsql_test 4.1 {
BEGIN;
INSERT INTO v1 (rowid, vector) VALUES( 0x1_0000_0001, random_vector(8) );
INSERT INTO v1 (rowid, vector) VALUES( 0x1_0000_0002, random_vector(8) );
INSERT INTO v1 (rowid, vector) VALUES( 0x1_0000_0003, random_vector(8) );
INSERT INTO v1 (rowid, vector) VALUES( 0x1_0000_0004, random_vector(8) );
COMMIT;
}
do_execsql_test 4.2 {
PRAGMA integrity_check;
} {ok}
do_execsql_test 4.3 {
DELETE FROM v1 WHERE rowid = 0x1_0000_0003;
} {}
do_execsql_test 4.4 {
PRAGMA integrity_check;
} {ok}
#-------------------------------------------------------------------------
reset_db
vec1_load db
do_execsql_test 4.0 {
CREATE VIRTUAL TABLE v1 USING vec1;
INSERT INTO v1(cmd, vector) VALUES('rebuild', '{index:"flat"}');
}
do_execsql_test 4.1 {
INSERT INTO v1 VALUES( random_vector( 16 ) );
}
do_execsql_test 4.2 {
INSERT INTO v1 VALUES( random_vector( 16 ) );
SELECT count(*) FROM v1_idx;
} {1}
do_execsql_test 4.3 {
PRAGMA integrity_check
} {ok}
#-------------------------------------------------------------------------
#
reset_db
vec1_load db
do_execsql_test 5.0 {
CREATE VIRTUAL TABLE v1 USING vec1;
INSERT INTO v1(cmd, vector) VALUES('rebuild', '{index:"flat"}');
}
do_execsql_test 5.1 {
INSERT INTO v1 VALUES( random_vector( 16 ) );
INSERT INTO v1 VALUES( random_vector( 16 ) );
INSERT INTO v1 VALUES( random_vector( 16 ) );
}
do_execsql_test 5.2 {
BEGIN;
INSERT INTO v1(rowid, vector) VALUES(11, random_vector( 16 ) );
INSERT INTO v1(rowid, vector) VALUES(12, random_vector( 16 ) );
DELETE FROM v1 WHERE rowid=11;
COMMIT;
} {}
do_execsql_test 5.3 {
PRAGMA integrity_check
} {ok}
#-------------------------------------------------------------------------
#
reset_db
vec1_load db
do_execsql_test 6.0 {
CREATE VIRTUAL TABLE v1 USING vec1(x);
INSERT INTO v1 VALUES( random_vector(22) );
}
do_catchsql_test 6.1 {
INSERT INTO v1 VALUES( random_vector(23) );
} {1 {vec1: unexpected vector blob size 92 bytes, expected 88}}
#-------------------------------------------------------------------------
#
reset_db
vec1_load db
do_execsql_test 6.0 {
CREATE VIRTUAL TABLE v1 USING vec1(x);
}
do_catchsql_test 6.1 {
INSERT INTO v1(cmd, arg) VALUES('rebuild', '{distance:"l2"}');
} {1 {vec1: required parameter missing: index}}
do_execsql_test 6.2 {
INSERT INTO v1 VALUES(random_vector(16));
INSERT INTO v1 VALUES(random_vector(16));
INSERT INTO v1 VALUES(random_vector(16));
INSERT INTO v1 VALUES(random_vector(16));
INSERT INTO v1 VALUES(random_vector(16));
UPDATE v1_base SET vector = random_vector(15) WHERE id=4;
}
do_catchsql_test 6.1 {
INSERT INTO v1(cmd, arg) VALUES('rebuild', '{index:"flat",distance:"l2"}');
} {1 {vec1: unexpected vector size in v1_base: 60}}
#-------------------------------------------------------------------------
#
reset_db
vec1_load db
do_execsql_test 7.0 {
CREATE VIRTUAL TABLE v1 USING vec1(x);
}
do_catchsql_test 7.1 {
INSERT INTO v1(x) VALUES('sometextvalu');
} {1 {vec1: unexpected vector type TEXT, expected BLOB}}
do_execsql_test 7.2 {
INSERT INTO v1(x) VALUES( random_vector(4) );
} {}
do_catchsql_test 7.3 {
INSERT INTO v1(x) VALUES('sometextvaluABCD');
} {1 {vec1: unexpected vector type TEXT, expected BLOB}}
do_catchsql_test 7.4 {
UPDATE v1 SET x='sometextvaluABCD';
} {1 {vec1: unexpected vector type TEXT, expected BLOB}}
finish_test