liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-14 00:26:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use consistent style in TAP tests [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2023-08-11 19:08:15 -07:00
parent 746dce72ef
commit aabbeac30c
9 changed files with 38 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/t/001_ivfflat_wal.pl
View File

@@ -26,7 +26,8 @@ sub test_index_replay
or die "Timed out while waiting for replica 1 to catch up"; or die "Timed out while waiting for replica 1 to catch up";
my @r = (); my @r = ();
for (1 .. $dim) { for (1 .. $dim)
{
push(@r, rand()); push(@r, rand());
} }
my $sql = join(",", @r); my $sql = join(",", @r);
@@ -67,8 +68,7 @@ $node_primary->backup($backup_name);
# Create streaming replica linking to primary # Create streaming replica linking to primary
$node_replica = get_new_node('replica'); $node_replica = get_new_node('replica');
$node_replica->init_from_backup($node_primary, $backup_name, $node_replica->init_from_backup($node_primary, $backup_name, has_streaming => 1);
has_streaming => 1);
$node_replica->start; $node_replica->start;
# Create ivfflat index on primary # Create ivfflat index on primary

3
test/t/002_ivfflat_vacuum.pl
View File

@@ -7,7 +7,8 @@ use Test::More;
my $dim = 3; my $dim = 3;
my @r = (); my @r = ();
for (1 .. $dim) { for (1 .. $dim)
{
my $v = int(rand(1000)) + 1; my $v = int(rand(1000)) + 1;
push(@r, "i % $v"); push(@r, "i % $v");
} }

12
test/t/003_ivfflat_recall.pl
View File

@@ -57,7 +57,8 @@ $node->safe_psql("postgres",
); );
# Generate queries # Generate queries
for (1..20) { for (1 .. 20)
{
my $r1 = rand(); my $r1 = rand();
my $r2 = rand(); my $r2 = rand();
my $r3 = rand(); my $r3 = rand();
@@ -67,7 +68,8 @@ for (1..20) {
# Check each index type # Check each index type
my @operators = ("<->", "<#>", "<=>"); my @operators = ("<->", "<#>", "<=>");
foreach (@operators) { foreach (@operators)
{
my $operator = $_; my $operator = $_;
# Get exact results # Get exact results
@@ -93,7 +95,8 @@ foreach (@operators) {
)); ));
# Test approximate results # Test approximate results
if ($operator ne "<#>") { if ($operator ne "<#>")
{
# TODO fix test # TODO fix test
test_recall(1, 0.73, $operator); test_recall(1, 0.73, $operator);
test_recall(10, 0.95, $operator); test_recall(10, 0.95, $operator);
@@ -113,7 +116,8 @@ foreach (@operators) {
like($stderr, qr/using \d+ parallel workers/); like($stderr, qr/using \d+ parallel workers/);
# Test approximate results # Test approximate results
if ($operator ne "<#>") { if ($operator ne "<#>")
{
# TODO fix test # TODO fix test
test_recall(1, 0.73, $operator); test_recall(1, 0.73, $operator);
test_recall(10, 0.95, $operator); test_recall(10, 0.95, $operator);

6
test/t/005_ivfflat_query_recall.pl
View File

@@ -18,7 +18,8 @@ $node->safe_psql("postgres",
# Check each index type # Check each index type
my @operators = ("<->", "<#>", "<=>"); my @operators = ("<->", "<#>", "<=>");
foreach (@operators) { foreach (@operators)
{
my $operator = $_; my $operator = $_;
# Add index # Add index
@@ -33,7 +34,8 @@ foreach (@operators) {
$node->safe_psql("postgres", "CREATE INDEX ON tst USING ivfflat (v $opclass);"); $node->safe_psql("postgres", "CREATE INDEX ON tst USING ivfflat (v $opclass);");
# Test 100% recall # Test 100% recall
for (1..20) { for (1 .. 20)
{
my $i = int(rand() * 100000); my $i = int(rand() * 100000);
my $query = $node->safe_psql("postgres", "SELECT v FROM tst WHERE i = $i;"); my $query = $node->safe_psql("postgres", "SELECT v FROM tst WHERE i = $i;");
my $res = $node->safe_psql("postgres", qq( my $res = $node->safe_psql("postgres", qq(

3
test/t/008_aggregates.pl
View File

@@ -30,7 +30,8 @@ sub test_aggregate
# Test matches real for avg # Test matches real for avg
# Cannot test sum since sum(real) varies between calls # Cannot test sum since sum(real) varies between calls
if ($agg eq 'avg') { if ($agg eq 'avg')
{
my $r1 = $node->safe_psql("postgres", "SELECT $agg(r1)::float4 FROM tst;"); my $r1 = $node->safe_psql("postgres", "SELECT $agg(r1)::float4 FROM tst;");
my $r2 = $node->safe_psql("postgres", "SELECT $agg(r2)::float4 FROM tst;"); my $r2 = $node->safe_psql("postgres", "SELECT $agg(r2)::float4 FROM tst;");
my $r3 = $node->safe_psql("postgres", "SELECT $agg(r3)::float4 FROM tst;"); my $r3 = $node->safe_psql("postgres", "SELECT $agg(r3)::float4 FROM tst;");

9
test/t/010_hnsw_wal.pl
View File

@@ -52,11 +52,13 @@ my $array_sql = join(",", ('random()') x $dim);
# Initialize primary node # Initialize primary node
$node_primary = get_new_node('primary'); $node_primary = get_new_node('primary');
$node_primary->init(allows_streaming => 1); $node_primary->init(allows_streaming => 1);
if ($dim > 32) { if ($dim > 32)
{
# TODO use wal_keep_segments for Postgres < 13 # TODO use wal_keep_segments for Postgres < 13
$node_primary->append_conf('postgresql.conf', qq(wal_keep_size = 1GB)); $node_primary->append_conf('postgresql.conf', qq(wal_keep_size = 1GB));
} }
if ($dim > 1500) { if ($dim > 1500)
{
$node_primary->append_conf('postgresql.conf', qq(maintenance_work_mem = 128MB)); $node_primary->append_conf('postgresql.conf', qq(maintenance_work_mem = 128MB));
} }
$node_primary->start; $node_primary->start;
@@ -67,8 +69,7 @@ $node_primary->backup($backup_name);
# Create streaming replica linking to primary # Create streaming replica linking to primary
$node_replica = get_new_node('replica'); $node_replica = get_new_node('replica');
$node_replica->init_from_backup($node_primary, $backup_name, $node_replica->init_from_backup($node_primary, $backup_name, has_streaming => 1);
has_streaming => 1);
$node_replica->start; $node_replica->start;
# Create hnsw index on primary # Create hnsw index on primary

9
test/t/012_hnsw_build_recall.pl
View File

@@ -55,7 +55,8 @@ $node->safe_psql("postgres",
); );
# Generate queries # Generate queries
for (1..20) { for (1 .. 20)
{
my $r1 = rand(); my $r1 = rand();
my $r2 = rand(); my $r2 = rand();
my $r3 = rand(); my $r3 = rand();
@@ -65,12 +66,14 @@ for (1..20) {
# Check each index type # Check each index type
my @operators = ("<->", "<#>", "<=>"); my @operators = ("<->", "<#>", "<=>");
foreach (@operators) { foreach (@operators)
{
my $operator = $_; my $operator = $_;
# Get exact results # Get exact results
@expected = (); @expected = ();
foreach (@queries) { foreach (@queries)
{
my $res = $node->safe_psql("postgres", "SELECT i FROM tst ORDER BY v $operator '$_' LIMIT $limit;"); my $res = $node->safe_psql("postgres", "SELECT i FROM tst ORDER BY v $operator '$_' LIMIT $limit;");
push(@expected, $res); push(@expected, $res);
} }

9
test/t/013_hnsw_insert_recall.pl
View File

@@ -52,7 +52,8 @@ $node->safe_psql("postgres", "CREATE EXTENSION vector;");
$node->safe_psql("postgres", "CREATE TABLE tst (i serial, v vector(3));"); $node->safe_psql("postgres", "CREATE TABLE tst (i serial, v vector(3));");
# Generate queries # Generate queries
for (1..20) { for (1 .. 20)
{
my $r1 = rand(); my $r1 = rand();
my $r2 = rand(); my $r2 = rand();
my $r3 = rand(); my $r3 = rand();
@@ -62,7 +63,8 @@ for (1..20) {
# Check each index type # Check each index type
my @operators = ("<->", "<#>", "<=>"); my @operators = ("<->", "<#>", "<=>");
foreach (@operators) { foreach (@operators)
{
my $operator = $_; my $operator = $_;
# Add index # Add index
@@ -90,7 +92,8 @@ foreach (@operators) {
# Get exact results # Get exact results
@expected = (); @expected = ();
foreach (@queries) { foreach (@queries)
{
my $res = $node->safe_psql("postgres", qq( my $res = $node->safe_psql("postgres", qq(
SET enable_indexscan = off; SET enable_indexscan = off;
SELECT i FROM tst ORDER BY v $operator '$_' LIMIT $limit; SELECT i FROM tst ORDER BY v $operator '$_' LIMIT $limit;

3
test/t/015_hnsw_duplicates.pl
View File

@@ -15,7 +15,8 @@ $node->safe_psql("postgres", "CREATE TABLE tst (v vector(3));");
sub insert_vectors sub insert_vectors
{ {
for my $i (1 .. 20) { for my $i (1 .. 20)
{
$node->safe_psql("postgres", "INSERT INTO tst VALUES ('[1,1,1]');"); $node->safe_psql("postgres", "INSERT INTO tst VALUES ('[1,1,1]');");
} }
} }