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

Improved test

Browse Source
This commit is contained in:
Andrew Kane
2022-11-02 12:50:50 -07:00
parent 059e88dda5
commit 3f8407460e
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/t/007_inserts.pl
View File

@@ -4,6 +4,10 @@ use PostgresNode;
use TestLib; use TestLib;
use Test::More tests => 3; use Test::More tests => 3;
# TODO fix randomness
my $dim = 768;
# Initialize node # Initialize node
my $node = get_new_node('node'); my $node = get_new_node('node');
$node->init; $node->init;
@@ -11,9 +15,9 @@ $node->start;
# Create table and index # Create table and index
$node->safe_psql("postgres", "CREATE EXTENSION vector;"); $node->safe_psql("postgres", "CREATE EXTENSION vector;");
$node->safe_psql("postgres", "CREATE TABLE tst (v vector(768));"); $node->safe_psql("postgres", "CREATE TABLE tst (v vector($dim));");
$node->safe_psql("postgres", $node->safe_psql("postgres",
"INSERT INTO tst SELECT (SELECT array_agg(random()) FROM generate_series(1, 768)) FROM generate_series(1, 10000) i;" "INSERT INTO tst SELECT (SELECT array_agg(random()) FROM generate_series(1, $dim)) FROM generate_series(1, 10000) i;"
); );
$node->safe_psql("postgres", "CREATE INDEX ON tst USING ivfflat (v);"); $node->safe_psql("postgres", "CREATE INDEX ON tst USING ivfflat (v);");
@@ -24,10 +28,6 @@ $node->pgbench(
[qr{^$}], [qr{^$}],
"concurrent INSERTs", "concurrent INSERTs",
{ {
"007_concurrent" => q( "007_concurrent" => "INSERT INTO tst SELECT (SELECT array_agg(random()) FROM generate_series(1, $dim)) FROM generate_series(1, 10) i;"
BEGIN;
INSERT INTO tst SELECT (SELECT array_agg(random()) FROM generate_series(1, 768)) FROM generate_series(1, 10) i;
COMMIT;
),
} }
); );