diff --git a/tests/__snapshots__/test-metadata.ambr b/tests/__snapshots__/test-metadata.ambr index 537df26..d9892f5 100644 --- a/tests/__snapshots__/test-metadata.ambr +++ b/tests/__snapshots__/test-metadata.ambr @@ -619,45 +619,45 @@ # --- # name: test_long_text_knn[eq-bbbb] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name = ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?", 'rows': list([ OrderedDict({ 'rowid': 3, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbb', + 'distance': 97.0, }), ]), }) # --- # name: test_long_text_knn[eq-bbbbbbbbbbbb_aaa] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name = ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?", 'rows': list([ ]), }) # --- # name: test_long_text_knn[eq-bbbbbbbbbbbb_bbb] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name = ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?", 'rows': list([ OrderedDict({ 'rowid': 4, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbbbbbbbbbb_bbb', + 'distance': 96.0, }), ]), }) # --- # name: test_long_text_knn[eq-bbbbbbbbbbbb_ccc] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name = ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?", 'rows': list([ ]), }) # --- # name: test_long_text_knn[eq-longlonglonglonglonglonglong] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name = ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?", 'rows': list([ ]), }) @@ -784,160 +784,160 @@ # --- # name: test_long_text_knn[ne-bbbb] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?", 'rows': list([ OrderedDict({ 'rowid': 6, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccccccccccc_ccc', + 'distance': 94.0, }), OrderedDict({ 'rowid': 5, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccc', + 'distance': 95.0, }), OrderedDict({ 'rowid': 4, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbbbbbbbbbb_bbb', + 'distance': 96.0, }), OrderedDict({ 'rowid': 2, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaaaaaaaaaa_aaa', + 'distance': 98.0, }), OrderedDict({ 'rowid': 1, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaa', + 'distance': 99.0, }), ]), }) # --- # name: test_long_text_knn[ne-bbbbbbbbbbbb_aaa] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?", 'rows': list([ OrderedDict({ 'rowid': 6, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccccccccccc_ccc', + 'distance': 94.0, }), OrderedDict({ 'rowid': 5, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccc', + 'distance': 95.0, }), OrderedDict({ 'rowid': 4, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbbbbbbbbbb_bbb', + 'distance': 96.0, }), OrderedDict({ 'rowid': 3, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbb', + 'distance': 97.0, }), OrderedDict({ 'rowid': 2, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaaaaaaaaaa_aaa', + 'distance': 98.0, }), ]), }) # --- # name: test_long_text_knn[ne-bbbbbbbbbbbb_bbb] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?", 'rows': list([ OrderedDict({ 'rowid': 6, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccccccccccc_ccc', + 'distance': 94.0, }), OrderedDict({ 'rowid': 5, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccc', + 'distance': 95.0, }), OrderedDict({ 'rowid': 3, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbb', + 'distance': 97.0, }), OrderedDict({ 'rowid': 2, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaaaaaaaaaa_aaa', + 'distance': 98.0, }), OrderedDict({ 'rowid': 1, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaa', + 'distance': 99.0, }), ]), }) # --- # name: test_long_text_knn[ne-bbbbbbbbbbbb_ccc] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?", 'rows': list([ OrderedDict({ 'rowid': 6, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccccccccccc_ccc', + 'distance': 94.0, }), OrderedDict({ 'rowid': 5, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccc', + 'distance': 95.0, }), OrderedDict({ 'rowid': 4, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbbbbbbbbbb_bbb', + 'distance': 96.0, }), OrderedDict({ 'rowid': 3, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbb', + 'distance': 97.0, }), OrderedDict({ 'rowid': 2, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaaaaaaaaaa_aaa', + 'distance': 98.0, }), ]), }) # --- # name: test_long_text_knn[ne-longlonglonglonglonglonglong] OrderedDict({ - 'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?", + 'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?", 'rows': list([ OrderedDict({ 'rowid': 6, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccccccccccc_ccc', + 'distance': 94.0, }), OrderedDict({ 'rowid': 5, - 'vector': b'\x11\x11\x11\x11', 'name': 'cccc', + 'distance': 95.0, }), OrderedDict({ 'rowid': 4, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbbbbbbbbbb_bbb', + 'distance': 96.0, }), OrderedDict({ 'rowid': 3, - 'vector': b'\x11\x11\x11\x11', 'name': 'bbbb', + 'distance': 97.0, }), OrderedDict({ 'rowid': 2, - 'vector': b'\x11\x11\x11\x11', 'name': 'aaaaaaaaaaaa_aaa', + 'distance': 98.0, }), ]), }) diff --git a/tests/test-metadata.py b/tests/test-metadata.py index ed7fd2a..edf37d2 100644 --- a/tests/test-metadata.py +++ b/tests/test-metadata.py @@ -138,12 +138,12 @@ def test_long_text_knn(db, snapshot): "create virtual table v using vec0(vector float[1], name text, chunk_size=8)" ) INSERT = "insert into v(vector, name) values (?, ?)" - exec(db, INSERT, [b"\x11\x11\x11\x11", "aaaa"]) - exec(db, INSERT, [b"\x11\x11\x11\x11", "aaaaaaaaaaaa_aaa"]) - exec(db, INSERT, [b"\x11\x11\x11\x11", "bbbb"]) - exec(db, INSERT, [b"\x11\x11\x11\x11", "bbbbbbbbbbbb_bbb"]) - exec(db, INSERT, [b"\x11\x11\x11\x11", "cccc"]) - exec(db, INSERT, [b"\x11\x11\x11\x11", "cccccccccccc_ccc"]) + exec(db, INSERT, ["[1]", "aaaa"]) + exec(db, INSERT, ["[2]", "aaaaaaaaaaaa_aaa"]) + exec(db, INSERT, ["[3]", "bbbb"]) + exec(db, INSERT, ["[4]", "bbbbbbbbbbbb_bbb"]) + exec(db, INSERT, ["[5]", "cccc"]) + exec(db, INSERT, ["[6]", "cccccccccccc_ccc"]) tests = [ "bbbb", @@ -159,7 +159,7 @@ def test_long_text_knn(db, snapshot): for op, op_name in zip(ops, op_names): assert exec( db, - f"select * from v where vector match X'11111111' and k = 5 and name {op} ?", + f"select rowid, name, distance from v where vector match '[100]' and k = 5 and name {op} ?", [test], ) == snapshot(name=f"{op_name}-{test}")