mirror of
https://github.com/asg017/sqlite-vec.git
synced 2026-04-26 01:06:27 +02:00
stress test
This commit is contained in:
Alex Garcia
parent
fef4997b68
commit
b04b64759c
2 changed files with 512 additions and 0 deletions
|
|
@ -328,3 +328,443 @@
|
|||
]),
|
||||
})
|
||||
# ---
|
||||
# name: test_stress
|
||||
dict({
|
||||
'vec_movies_auxiliary': OrderedDict({
|
||||
'sql': 'select * from vec_movies_auxiliary',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'rowid': 1,
|
||||
'value00': 'The Conjuring',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 2,
|
||||
'value00': 'Dumb and Dumber',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 3,
|
||||
'value00': 'Interstellar',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 4,
|
||||
'value00': 'The Lord of the Rings: The Fellowship of the Ring',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 5,
|
||||
'value00': 'An Inconvenient Truth',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 6,
|
||||
'value00': 'Hereditary',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 7,
|
||||
'value00': 'Anchorman: The Legend of Ron Burgundy',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 8,
|
||||
'value00': 'Blade Runner 2049',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 9,
|
||||
'value00': "Harry Potter and the Sorcerer's Stone",
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 10,
|
||||
'value00': 'Free Solo',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 11,
|
||||
'value00': 'Get Out',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 12,
|
||||
'value00': 'The Hangover',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 13,
|
||||
'value00': 'The Matrix',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 14,
|
||||
'value00': "Pan's Labyrinth",
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 15,
|
||||
'value00': '13th',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 16,
|
||||
'value00': 'It Follows',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 17,
|
||||
'value00': 'Step Brothers',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 18,
|
||||
'value00': 'Inception',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 19,
|
||||
'value00': 'The Shape of Water',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 20,
|
||||
'value00': "Won't You Be My Neighbor?",
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 21,
|
||||
'value00': 'Gravity',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 22,
|
||||
'value00': 'Dune',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 23,
|
||||
'value00': 'The Martian',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 24,
|
||||
'value00': 'A Quiet Place',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 25,
|
||||
'value00': 'The Chronicles of Narnia: The Lion, the Witch and the Wardrobe',
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
'vec_movies_chunks': OrderedDict({
|
||||
'sql': 'select * from vec_movies_chunks',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'chunk_id': 1,
|
||||
'size': 8,
|
||||
'validity': b'\xff',
|
||||
'rowids': b'\x01\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00\x00\x00\x00\x00\x07\x00\x00\x00\x00\x00\x00\x00\x08\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'chunk_id': 2,
|
||||
'size': 8,
|
||||
'validity': b'\xff',
|
||||
'rowids': b'\t\x00\x00\x00\x00\x00\x00\x00\n\x00\x00\x00\x00\x00\x00\x00\x0b\x00\x00\x00\x00\x00\x00\x00\x0c\x00\x00\x00\x00\x00\x00\x00\r\x00\x00\x00\x00\x00\x00\x00\x0e\x00\x00\x00\x00\x00\x00\x00\x0f\x00\x00\x00\x00\x00\x00\x00\x10\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'chunk_id': 3,
|
||||
'size': 8,
|
||||
'validity': b'\xff',
|
||||
'rowids': b'\x11\x00\x00\x00\x00\x00\x00\x00\x12\x00\x00\x00\x00\x00\x00\x00\x13\x00\x00\x00\x00\x00\x00\x00\x14\x00\x00\x00\x00\x00\x00\x00\x15\x00\x00\x00\x00\x00\x00\x00\x16\x00\x00\x00\x00\x00\x00\x00\x17\x00\x00\x00\x00\x00\x00\x00\x18\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'chunk_id': 4,
|
||||
'size': 8,
|
||||
'validity': b'\x01',
|
||||
'rowids': b'\x19\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
'vec_movies_metadata_chunks00': OrderedDict({
|
||||
'sql': 'select * from vec_movies_metadata_chunks00',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'rowid': 1,
|
||||
'data': b'\x06\x00\x00\x00horror\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00comedy\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00\x07\x00\x00\x00fantasy\x00\x00\x00\x00\x00\x0b\x00\x00\x00documentary\x00\x06\x00\x00\x00horror\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00comedy\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 2,
|
||||
'data': b'\x07\x00\x00\x00fantasy\x00\x00\x00\x00\x00\x0b\x00\x00\x00documentary\x00\x06\x00\x00\x00horror\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00comedy\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00\x07\x00\x00\x00fantasy\x00\x00\x00\x00\x00\x0b\x00\x00\x00documentary\x00\x06\x00\x00\x00horror\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 3,
|
||||
'data': b'\x06\x00\x00\x00comedy\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00\x07\x00\x00\x00fantasy\x00\x00\x00\x00\x00\x0b\x00\x00\x00documentary\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00\x05\x00\x00\x00scifi\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00horror\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 4,
|
||||
'data': b'\x07\x00\x00\x00fantasy\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
'vec_movies_metadata_chunks01': OrderedDict({
|
||||
'sql': 'select * from vec_movies_metadata_chunks01',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'rowid': 1,
|
||||
'data': b'\x99\x00\x00\x00~\x01\x00\x005\x00\x00\x00\xd2\x00\x00\x00]\x00\x00\x00\xa7\x00\x00\x00\xe2\x01\x00\x00-\x01\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 2,
|
||||
'data': b'\x86\x00\x00\x00B\x00\x00\x00X\x00\x00\x00;\x00\x00\x00\xa7\x01\x00\x00\x13\x01\x00\x00\xbf\x00\x00\x00:\x01\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 3,
|
||||
'data': b'J\x00\x00\x00\xc9\x00\x00\x00\x8f\x01\x00\x00\xba\x00\x00\x00V\x01\x00\x00\xc3\x01\x00\x00\n\x02\x00\x00\x0f\x01\x00\x00',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 4,
|
||||
'data': b'6\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
'vec_movies_metadata_chunks02': OrderedDict({
|
||||
'sql': 'select * from vec_movies_metadata_chunks02',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'rowid': 1,
|
||||
'data': b'33\x93@ff&@\x00\x00\xa0@ff\x86@\x9a\x99Y@ff\x96@\x9a\x999@\x00\x00\xa0@',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 2,
|
||||
'data': b'33\x83@\xcd\xccL@\xcd\xcc\x9c@333@\x00\x00\x90@fff@\xcd\xcc\x8c@\x9a\x99\x89@',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 3,
|
||||
'data': b'\x00\x00@@\x00\x00\xa0@\xcd\xcc,@\x9a\x99\x99@\x00\x00\x80@\xcd\xcc\x8c@33\x93@\x9a\x99\x89@',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 4,
|
||||
'data': b'\x9a\x99y@\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
'vec_movies_rowids': OrderedDict({
|
||||
'sql': 'select * from vec_movies_rowids',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'rowid': 1,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 0,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 2,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 1,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 3,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 2,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 4,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 3,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 5,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 4,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 6,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 7,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 6,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 8,
|
||||
'id': None,
|
||||
'chunk_id': 1,
|
||||
'chunk_offset': 7,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 9,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 0,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 10,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 1,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 11,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 2,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 12,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 3,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 13,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 4,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 14,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 15,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 6,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 16,
|
||||
'id': None,
|
||||
'chunk_id': 2,
|
||||
'chunk_offset': 7,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 17,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 0,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 18,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 1,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 19,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 2,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 20,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 3,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 21,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 4,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 22,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 23,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 6,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 24,
|
||||
'id': None,
|
||||
'chunk_id': 3,
|
||||
'chunk_offset': 7,
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 25,
|
||||
'id': None,
|
||||
'chunk_id': 4,
|
||||
'chunk_offset': 0,
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
'vec_movies_vector_chunks00': OrderedDict({
|
||||
'sql': 'select * from vec_movies_vector_chunks00',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'rowid': 1,
|
||||
'vectors': b'\x00\x00\x80?\x00\x00\x00@\x00\x00@@\x00\x00\x80@\x00\x00\xa0@\x00\x00\xc0@\x00\x00\xe0@\x00\x00\x00A',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 2,
|
||||
'vectors': b'\x00\x00\x10A\x00\x00 A\x00\x000A\x00\x00@A\x00\x00PA\x00\x00`A\x00\x00pA\x00\x00\x80A',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 3,
|
||||
'vectors': b'\x00\x00\x88A\x00\x00\x90A\x00\x00\x98A\x00\x00\xa0A\x00\x00\xa8A\x00\x00\xb0A\x00\x00\xb8A\x00\x00\xc0A',
|
||||
}),
|
||||
OrderedDict({
|
||||
'rowid': 4,
|
||||
'vectors': b'\x00\x00\xc8A\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
|
||||
}),
|
||||
]),
|
||||
}),
|
||||
})
|
||||
# ---
|
||||
# name: test_stress.1
|
||||
OrderedDict({
|
||||
'sql': '''
|
||||
|
||||
select
|
||||
movie_id,
|
||||
title,
|
||||
genre,
|
||||
num_reviews,
|
||||
mean_rating,
|
||||
distance
|
||||
from vec_movies
|
||||
where synopsis_embedding match '[15.5]'
|
||||
and genre = 'scifi'
|
||||
and num_reviews between 100 and 500
|
||||
and mean_rating > 3.5
|
||||
and k = 5;
|
||||
|
||||
''',
|
||||
'rows': list([
|
||||
OrderedDict({
|
||||
'movie_id': 13,
|
||||
'title': 'The Matrix',
|
||||
'genre': 'scifi',
|
||||
'num_reviews': 423,
|
||||
'mean_rating': 4.5,
|
||||
'distance': 2.5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'movie_id': 18,
|
||||
'title': 'Inception',
|
||||
'genre': 'scifi',
|
||||
'num_reviews': 201,
|
||||
'mean_rating': 5.0,
|
||||
'distance': 2.5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'movie_id': 21,
|
||||
'title': 'Gravity',
|
||||
'genre': 'scifi',
|
||||
'num_reviews': 342,
|
||||
'mean_rating': 4.0,
|
||||
'distance': 5.5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'movie_id': 22,
|
||||
'title': 'Dune',
|
||||
'genre': 'scifi',
|
||||
'num_reviews': 451,
|
||||
'mean_rating': 4.400000095367432,
|
||||
'distance': 6.5,
|
||||
}),
|
||||
OrderedDict({
|
||||
'movie_id': 8,
|
||||
'title': 'Blade Runner 2049',
|
||||
'genre': 'scifi',
|
||||
'num_reviews': 301,
|
||||
'mean_rating': 5.0,
|
||||
'distance': 7.5,
|
||||
}),
|
||||
]),
|
||||
})
|
||||
# ---
|
||||
|
|
|
|||
|
|
@ -80,6 +80,78 @@ def test_knn(db, snapshot):
|
|||
)
|
||||
|
||||
|
||||
def test_stress(db, snapshot):
|
||||
db.execute(
|
||||
"""
|
||||
create virtual table vec_movies using vec0(
|
||||
movie_id integer primary key,
|
||||
synopsis_embedding float[1],
|
||||
+title text,
|
||||
genre text,
|
||||
num_reviews int,
|
||||
mean_rating float,
|
||||
chunk_size=8
|
||||
);
|
||||
"""
|
||||
)
|
||||
|
||||
db.execute(
|
||||
"""
|
||||
INSERT INTO vec_movies(movie_id, synopsis_embedding, genre, title, num_reviews, mean_rating)
|
||||
VALUES
|
||||
(1, '[1]', 'horror', 'The Conjuring', 153, 4.6),
|
||||
(2, '[2]', 'comedy', 'Dumb and Dumber', 382, 2.6),
|
||||
(3, '[3]', 'scifi', 'Interstellar', 53, 5.0),
|
||||
(4, '[4]', 'fantasy', 'The Lord of the Rings: The Fellowship of the Ring', 210, 4.2),
|
||||
(5, '[5]', 'documentary', 'An Inconvenient Truth', 93, 3.4),
|
||||
(6, '[6]', 'horror', 'Hereditary', 167, 4.7),
|
||||
(7, '[7]', 'comedy', 'Anchorman: The Legend of Ron Burgundy', 482, 2.9),
|
||||
(8, '[8]', 'scifi', 'Blade Runner 2049', 301, 5.0),
|
||||
(9, '[9]', 'fantasy', 'Harry Potter and the Sorcerer''s Stone', 134, 4.1),
|
||||
(10, '[10]', 'documentary', 'Free Solo', 66, 3.2),
|
||||
(11, '[11]', 'horror', 'Get Out', 88, 4.9),
|
||||
(12, '[12]', 'comedy', 'The Hangover', 59, 2.8),
|
||||
(13, '[13]', 'scifi', 'The Matrix', 423, 4.5),
|
||||
(14, '[14]', 'fantasy', 'Pan''s Labyrinth', 275, 3.6),
|
||||
(15, '[15]', 'documentary', '13th', 191, 4.4),
|
||||
(16, '[16]', 'horror', 'It Follows', 314, 4.3),
|
||||
(17, '[17]', 'comedy', 'Step Brothers', 74, 3.0),
|
||||
(18, '[18]', 'scifi', 'Inception', 201, 5.0),
|
||||
(19, '[19]', 'fantasy', 'The Shape of Water', 399, 2.7),
|
||||
(20, '[20]', 'documentary', 'Won''t You Be My Neighbor?', 186, 4.8),
|
||||
(21, '[21]', 'scifi', 'Gravity', 342, 4.0),
|
||||
(22, '[22]', 'scifi', 'Dune', 451, 4.4),
|
||||
(23, '[23]', 'scifi', 'The Martian', 522, 4.6),
|
||||
(24, '[24]', 'horror', 'A Quiet Place', 271, 4.3),
|
||||
(25, '[25]', 'fantasy', 'The Chronicles of Narnia: The Lion, the Witch and the Wardrobe', 310, 3.9);
|
||||
|
||||
"""
|
||||
)
|
||||
|
||||
assert vec0_shadow_table_contents(db, "vec_movies") == snapshot()
|
||||
assert (
|
||||
exec(
|
||||
db,
|
||||
"""
|
||||
select
|
||||
movie_id,
|
||||
title,
|
||||
genre,
|
||||
num_reviews,
|
||||
mean_rating,
|
||||
distance
|
||||
from vec_movies
|
||||
where synopsis_embedding match '[15.5]'
|
||||
and genre = 'scifi'
|
||||
and num_reviews between 100 and 500
|
||||
and mean_rating > 3.5
|
||||
and k = 5;
|
||||
""",
|
||||
)
|
||||
== snapshot()
|
||||
)
|
||||
|
||||
|
||||
def exec(db, sql, parameters=[]):
|
||||
try:
|
||||
rows = db.execute(sql, parameters).fetchall()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue