From c198fd58ee2d4aeaa75d03c5a3b87c15b161cd9f Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Wed, 24 Apr 2024 01:31:50 -0700 Subject: [PATCH] Added more tests for subvector function [skip ci] --- test/expected/halfvec_functions.out | 6 ++++++ test/expected/vector_functions.out | 6 ++++++ test/sql/halfvec_functions.sql | 1 + test/sql/vector_functions.sql | 1 + 4 files changed, 14 insertions(+) diff --git a/test/expected/halfvec_functions.out b/test/expected/halfvec_functions.out index aaf3588..e7e3d76 100644 --- a/test/expected/halfvec_functions.out +++ b/test/expected/halfvec_functions.out @@ -414,6 +414,12 @@ SELECT subvector('[1,2,3,4,5]'::halfvec, 3, 2147483647); [3,4,5] (1 row) +SELECT subvector('[1,2,3,4,5]'::halfvec, -2147483644, 2147483647); + subvector +----------- + [1,2] +(1 row) + SELECT avg(v) FROM unnest(ARRAY['[1,2,3]'::halfvec, '[3,5,7]']) v; avg ----------- diff --git a/test/expected/vector_functions.out b/test/expected/vector_functions.out index 4b22927..dd3654c 100644 --- a/test/expected/vector_functions.out +++ b/test/expected/vector_functions.out @@ -438,6 +438,12 @@ SELECT subvector('[1,2,3,4,5]'::vector, 3, 2147483647); [3,4,5] (1 row) +SELECT subvector('[1,2,3,4,5]'::vector, -2147483644, 2147483647); + subvector +----------- + [1,2] +(1 row) + SELECT avg(v) FROM unnest(ARRAY['[1,2,3]'::vector, '[3,5,7]']) v; avg ----------- diff --git a/test/sql/halfvec_functions.sql b/test/sql/halfvec_functions.sql index 77a1a88..efd005a 100644 --- a/test/sql/halfvec_functions.sql +++ b/test/sql/halfvec_functions.sql @@ -92,6 +92,7 @@ SELECT subvector('[1,2,3,4,5]'::halfvec, 3, -1); SELECT subvector('[1,2,3,4,5]'::halfvec, -1, 2); SELECT subvector('[1,2,3,4,5]'::halfvec, 2147483647, 10); SELECT subvector('[1,2,3,4,5]'::halfvec, 3, 2147483647); +SELECT subvector('[1,2,3,4,5]'::halfvec, -2147483644, 2147483647); SELECT avg(v) FROM unnest(ARRAY['[1,2,3]'::halfvec, '[3,5,7]']) v; SELECT avg(v) FROM unnest(ARRAY['[1,2,3]'::halfvec, '[3,5,7]', NULL]) v; diff --git a/test/sql/vector_functions.sql b/test/sql/vector_functions.sql index 25eb810..7b4de45 100644 --- a/test/sql/vector_functions.sql +++ b/test/sql/vector_functions.sql @@ -96,6 +96,7 @@ SELECT subvector('[1,2,3,4,5]'::vector, 3, -1); SELECT subvector('[1,2,3,4,5]'::vector, -1, 2); SELECT subvector('[1,2,3,4,5]'::vector, 2147483647, 10); SELECT subvector('[1,2,3,4,5]'::vector, 3, 2147483647); +SELECT subvector('[1,2,3,4,5]'::vector, -2147483644, 2147483647); SELECT avg(v) FROM unnest(ARRAY['[1,2,3]'::vector, '[3,5,7]']) v; SELECT avg(v) FROM unnest(ARRAY['[1,2,3]'::vector, '[3,5,7]', NULL]) v;