mirror of
https://github.com/pgvector/pgvector.git
synced 2026-06-06 05:51:21 +08:00
59 lines
1.2 KiB
Perl
59 lines
1.2 KiB
Perl
use strict;
|
|
use warnings;
|
|
use PostgreSQL::Test::Cluster;
|
|
use PostgreSQL::Test::Utils;
|
|
use Test::More;
|
|
|
|
# Initialize node
|
|
my $node = PostgreSQL::Test::Cluster->new('node');
|
|
$node->init;
|
|
$node->start;
|
|
|
|
# Create table
|
|
$node->safe_psql("postgres", "CREATE EXTENSION vector;");
|
|
$node->safe_psql("postgres", "CREATE TABLE tst (v vector(3));");
|
|
|
|
sub insert_vectors
|
|
{
|
|
for my $i (1 .. 20)
|
|
{
|
|
$node->safe_psql("postgres", "INSERT INTO tst VALUES ('[1,1,1]');");
|
|
}
|
|
}
|
|
|
|
sub test_duplicates
|
|
{
|
|
my $res = $node->safe_psql("postgres", qq(
|
|
SET enable_seqscan = off;
|
|
SET hnsw.ef_search = 1;
|
|
SELECT COUNT(*) FROM (SELECT * FROM tst ORDER BY v <-> '[1,1,1]') t;
|
|
));
|
|
is($res, 10);
|
|
}
|
|
|
|
# Test duplicates with build
|
|
insert_vectors();
|
|
$node->safe_psql("postgres", "CREATE INDEX idx ON tst USING hnsw (v vector_l2_ops);");
|
|
test_duplicates();
|
|
|
|
# Reset
|
|
$node->safe_psql("postgres", "TRUNCATE tst;");
|
|
|
|
# Test duplicates with inserts
|
|
insert_vectors();
|
|
test_duplicates();
|
|
|
|
# Test fallback path for inserts
|
|
$node->pgbench(
|
|
"--no-vacuum --client=5 --transactions=100",
|
|
0,
|
|
[qr{actually processed}],
|
|
[qr{^$}],
|
|
"concurrent INSERTs",
|
|
{
|
|
"015_hnsw_duplicates" => "INSERT INTO tst VALUES ('[1,1,1]');"
|
|
}
|
|
);
|
|
|
|
done_testing();
|