From c282627ce5a51a17fdd7917a532c611ed9b14daa Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Mon, 15 Apr 2024 10:51:52 -0700 Subject: [PATCH] Improved cosine distance tests for halfvec [skip ci] --- test/expected/halfvec_functions.out | 12 ++++++++++++ test/sql/halfvec_functions.sql | 2 ++ 2 files changed, 14 insertions(+) diff --git a/test/expected/halfvec_functions.out b/test/expected/halfvec_functions.out index 02c1bfd..0646dca 100644 --- a/test/expected/halfvec_functions.out +++ b/test/expected/halfvec_functions.out @@ -272,6 +272,18 @@ SELECT cosine_distance('[1,1]'::halfvec, '[-1.1,-1.1]'); 2 (1 row) +SELECT cosine_distance('[1,2,3,4,5,6,7,8,9]'::halfvec, '[1,2,3,4,5,6,7,8,9]'); + cosine_distance +----------------- + 0 +(1 row) + +SELECT cosine_distance('[1,2,3,4,5,6,7,8,9]'::halfvec, '[-1,-2,-3,-4,-5,-6,-7,-8,-9]'); + cosine_distance +----------------- + 2 +(1 row) + SELECT '[1,2]'::halfvec <=> '[2,4]'; ?column? ---------- diff --git a/test/sql/halfvec_functions.sql b/test/sql/halfvec_functions.sql index ce8fc48..fcf1ad4 100644 --- a/test/sql/halfvec_functions.sql +++ b/test/sql/halfvec_functions.sql @@ -61,6 +61,8 @@ SELECT cosine_distance('[1,1]'::halfvec, '[-1,-1]'); SELECT cosine_distance('[1,2]'::halfvec, '[3]'); SELECT cosine_distance('[1,1]'::halfvec, '[1.1,1.1]'); SELECT cosine_distance('[1,1]'::halfvec, '[-1.1,-1.1]'); +SELECT cosine_distance('[1,2,3,4,5,6,7,8,9]'::halfvec, '[1,2,3,4,5,6,7,8,9]'); +SELECT cosine_distance('[1,2,3,4,5,6,7,8,9]'::halfvec, '[-1,-2,-3,-4,-5,-6,-7,-8,-9]'); SELECT '[1,2]'::halfvec <=> '[2,4]'; SELECT l1_distance('[0,0]'::halfvec, '[3,4]');