apunkt/sqlite-vec
1
0
Fork 0
mirror of https://github.com/asg017/sqlite-vec.git synced 2026-04-25 08:46:49 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

boolean comparison handling

Browse source
This commit is contained in:
Alex Garcia 2024-11-14 16:36:53 -08:00
parent 3965029726
commit a1a64427fc
4 changed files with 189 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

131
tests/__snapshots__/test-metadata.ambr
View file

@ -1638,13 +1638,140 @@
]),
})
# ---
# name: test_stress.8
# name: test_stress[bool-eq-false]
OrderedDict({
'sql': "select movie_id, mean_rating, distance from vec_movies where synopsis_embedding match '[100]' and k = 5 and is_favorited = TRUE",
'sql': "select movie_id, is_favorited, distance from vec_movies where synopsis_embedding match '[100]' and k = 5 and is_favorited = FALSE",
'rows': list([
OrderedDict({
'movie_id': 16,
'is_favorited': 0,
'distance': 84.0,
}),
OrderedDict({
'movie_id': 14,
'is_favorited': 0,
'distance': 86.0,
}),
OrderedDict({
'movie_id': 12,
'is_favorited': 0,
'distance': 88.0,
}),
OrderedDict({
'movie_id': 10,
'is_favorited': 0,
'distance': 90.0,
}),
OrderedDict({
'movie_id': 8,
'is_favorited': 0,
'distance': 92.0,
}),
]),
})
# ---
# name: test_stress[bool-eq-true]
OrderedDict({
'sql': "select movie_id, is_favorited, distance from vec_movies where synopsis_embedding match '[100]' and k = 5 and is_favorited = TRUE",
'rows': list([
OrderedDict({
'movie_id': 25,
'is_favorited': 1,
'distance': 75.0,
}),
OrderedDict({
'movie_id': 24,
'is_favorited': 1,
'distance': 76.0,
}),
OrderedDict({
'movie_id': 23,
'is_favorited': 1,
'distance': 77.0,
}),
OrderedDict({
'movie_id': 22,
'is_favorited': 1,
'distance': 78.0,
}),
OrderedDict({
'movie_id': 21,
'is_favorited': 1,
'distance': 79.0,
}),
]),
})
# ---
# name: test_stress[bool-ne-false]
OrderedDict({
'sql': "select movie_id, is_favorited, distance from vec_movies where synopsis_embedding match '[100]' and k = 5 and is_favorited != FALSE",
'rows': list([
OrderedDict({
'movie_id': 25,
'is_favorited': 1,
'distance': 75.0,
}),
OrderedDict({
'movie_id': 24,
'is_favorited': 1,
'distance': 76.0,
}),
OrderedDict({
'movie_id': 23,
'is_favorited': 1,
'distance': 77.0,
}),
OrderedDict({
'movie_id': 22,
'is_favorited': 1,
'distance': 78.0,
}),
OrderedDict({
'movie_id': 21,
'is_favorited': 1,
'distance': 79.0,
}),
]),
})
# ---
# name: test_stress[bool-ne-true]
OrderedDict({
'sql': "select movie_id, is_favorited, distance from vec_movies where synopsis_embedding match '[100]' and k = 5 and is_favorited != TRUE",
'rows': list([
OrderedDict({
'movie_id': 16,
'is_favorited': 0,
'distance': 84.0,
}),
OrderedDict({
'movie_id': 14,
'is_favorited': 0,
'distance': 86.0,
}),
OrderedDict({
'movie_id': 12,
'is_favorited': 0,
'distance': 88.0,
}),
OrderedDict({
'movie_id': 10,
'is_favorited': 0,
'distance': 90.0,
}),
OrderedDict({
'movie_id': 8,
'is_favorited': 0,
'distance': 92.0,
}),
]),
})
# ---
# name: test_stress[bool-other-op]
dict({
'error': 'OperationalError',
'message': 'ONLY EQUALS (=) or NOT_EQUALS (!=) operators are allowed on boolean metadata columns.',
})
# ---
# name: test_types[illegal-boolean]
dict({
'error': 'OperationalError',