diff --git a/test/t/012_hnsw_vector_build_recall.pl b/test/t/012_hnsw_vector_build_recall.pl index 1298009..9ed8707 100644 --- a/test/t/012_hnsw_vector_build_recall.pl +++ b/test/t/012_hnsw_vector_build_recall.pl @@ -8,6 +8,7 @@ my $node; my @queries = (); my @expected; my $limit = 20; +my $array_sql = join(",", ('random() * random()') x 3); sub test_recall { @@ -54,7 +55,7 @@ $node->start; $node->safe_psql("postgres", "CREATE EXTENSION vector;"); $node->safe_psql("postgres", "CREATE TABLE tst (i int4, v vector(3));"); $node->safe_psql("postgres", - "INSERT INTO tst SELECT i, ARRAY[random(), random(), random()] FROM generate_series(1, 10000) i;" + "INSERT INTO tst SELECT i, ARRAY[$array_sql] FROM generate_series(1, 10000) i;" ); # Generate queries @@ -90,7 +91,7 @@ for my $i (0 .. $#operators) )); # Test approximate results - my $min = $operator eq "<#>" ? 0.80 : 0.99; + my $min = 0.99; test_recall($min, $operator); $node->safe_psql("postgres", "DROP INDEX idx;"); diff --git a/test/t/013_hnsw_vector_insert_recall.pl b/test/t/013_hnsw_vector_insert_recall.pl index 006231b..3ec2b5c 100644 --- a/test/t/013_hnsw_vector_insert_recall.pl +++ b/test/t/013_hnsw_vector_insert_recall.pl @@ -8,6 +8,7 @@ my $node; my @queries = (); my @expected; my $limit = 20; +my $array_sql = join(",", ('random() * random()') x 3); sub test_recall { @@ -83,7 +84,7 @@ for my $i (0 .. $#operators) [qr{^$}], "concurrent INSERTs", { - "013_hnsw_insert_recall_$opclass" => "INSERT INTO tst (v) VALUES (ARRAY[random(), random(), random()]);" + "013_hnsw_insert_recall_$opclass" => "INSERT INTO tst (v) VALUES (ARRAY[$array_sql]);" } ); @@ -99,7 +100,7 @@ for my $i (0 .. $#operators) } # Test approximate results - my $min = $operator eq "<#>" ? 0.80 : 0.99; + my $min = 0.99; test_recall($min, $operator); $node->safe_psql("postgres", "DROP INDEX idx;"); diff --git a/test/t/024_hnsw_halfvec_build_recall.pl b/test/t/024_hnsw_halfvec_build_recall.pl index f83a9b1..885e860 100644 --- a/test/t/024_hnsw_halfvec_build_recall.pl +++ b/test/t/024_hnsw_halfvec_build_recall.pl @@ -9,7 +9,7 @@ my @queries = (); my @expected; my $limit = 20; my $dim = 10; -my $array_sql = join(",", ('random()') x $dim); +my $array_sql = join(",", ('2 * random() * random()') x $dim); sub test_recall { @@ -94,7 +94,7 @@ for my $i (0 .. $#operators) )); # Test approximate results - my $min = $operator eq "<#>" ? 0.93 : 0.98; + my $min = 0.98; test_recall($min, $operator); $node->safe_psql("postgres", "DROP INDEX idx;"); diff --git a/test/t/025_hnsw_halfvec_insert_recall.pl b/test/t/025_hnsw_halfvec_insert_recall.pl index 6a7c501..c68d888 100644 --- a/test/t/025_hnsw_halfvec_insert_recall.pl +++ b/test/t/025_hnsw_halfvec_insert_recall.pl @@ -9,7 +9,7 @@ my @queries = (); my @expected; my $limit = 20; my $dim = 10; -my $array_sql = join(",", ('random()') x $dim); +my $array_sql = join(",", ('2 * random() * random()') x $dim); sub test_recall { @@ -103,7 +103,7 @@ for my $i (0 .. $#operators) } # Test approximate results - my $min = $operator eq "<#>" ? 0.94 : 0.98; + my $min = 0.98; test_recall($min, $operator); $node->safe_psql("postgres", "DROP INDEX idx;"); diff --git a/test/t/028_hnsw_sparsevec_build_recall.pl b/test/t/028_hnsw_sparsevec_build_recall.pl index 8533e2d..ec758e2 100644 --- a/test/t/028_hnsw_sparsevec_build_recall.pl +++ b/test/t/028_hnsw_sparsevec_build_recall.pl @@ -8,6 +8,7 @@ my $node; my @queries = (); my @expected; my $limit = 20; +my $array_sql = join(",", ('random() * random()') x 3); sub test_recall { @@ -54,7 +55,7 @@ $node->start; $node->safe_psql("postgres", "CREATE EXTENSION vector;"); $node->safe_psql("postgres", "CREATE TABLE tst (i int4, v sparsevec(3));"); $node->safe_psql("postgres", - "INSERT INTO tst SELECT i, ARRAY[random(), random(), random()]::vector::sparsevec FROM generate_series(1, 10000) i;" + "INSERT INTO tst SELECT i, ARRAY[$array_sql]::vector::sparsevec FROM generate_series(1, 10000) i;" ); # Generate queries @@ -90,7 +91,7 @@ for my $i (0 .. $#operators) )); # Test approximate results - my $min = $operator eq "<#>" ? 0.80 : 0.99; + my $min = 0.99; test_recall($min, $operator); $node->safe_psql("postgres", "DROP INDEX idx;"); diff --git a/test/t/029_hnsw_sparsevec_insert_recall.pl b/test/t/029_hnsw_sparsevec_insert_recall.pl index b6b8093..e87f672 100644 --- a/test/t/029_hnsw_sparsevec_insert_recall.pl +++ b/test/t/029_hnsw_sparsevec_insert_recall.pl @@ -8,6 +8,7 @@ my $node; my @queries = (); my @expected; my $limit = 20; +my $array_sql = join(",", ('random() * random()') x 3); sub test_recall { @@ -83,7 +84,7 @@ for my $i (0 .. $#operators) [qr{^$}], "concurrent INSERTs", { - "025_hnsw_sparsevec_insert_recall_$opclass" => "INSERT INTO tst (v) VALUES (ARRAY[random(), random(), random()]::vector::sparsevec);" + "025_hnsw_sparsevec_insert_recall_$opclass" => "INSERT INTO tst (v) VALUES (ARRAY[$array_sql]::vector::sparsevec);" } ); @@ -99,7 +100,7 @@ for my $i (0 .. $#operators) } # Test approximate results - my $min = $operator eq "<#>" ? 0.79 : 0.99; + my $min = 0.99; test_recall($min, $operator); $node->safe_psql("postgres", "DROP INDEX idx;");