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

Fixed test style

Browse Source
This commit is contained in:
Andrew Kane
2022-02-06 20:19:20 -08:00
parent d5da22f24b
commit da88217fd8
3 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.editorconfig
View File

@@ -1,6 +1,6 @@
root = true root = true
[*.{c,h}] [*.{c,h,pl}]
indent_style = tab indent_style = tab
indent_size = tab indent_size = tab
tab_width = 4 tab_width = 4

7
test/t/001_wal.pl
View File

@@ -34,9 +34,10 @@ sub test_index_replay
my $r2 = rand(); my $r2 = rand();
my $r3 = rand(); my $r3 = rand();
my $queries = qq(SET enable_seqscan=off; my $queries = qq(
SELECT * FROM tst ORDER BY v <-> '[$r1,$r2,$r3]' LIMIT 10; SET enable_seqscan = off;
); SELECT * FROM tst ORDER BY v <-> '[$r1,$r2,$r3]' LIMIT 10;
);
# Run test queries and compare their result # Run test queries and compare their result
my $primary_result = $node_primary->safe_psql("postgres", $queries); my $primary_result = $node_primary->safe_psql("postgres", $queries);

32
test/t/003_recall.pl
View File

@@ -10,24 +10,24 @@ my @expected_ids;
# TODO Run more queries to prevent flakiness # TODO Run more queries to prevent flakiness
sub test_recall sub test_recall
{ {
my ($probes, $min) = @_; my ($probes, $min) = @_;
my $actual = $node->safe_psql("postgres", qq( my $actual = $node->safe_psql("postgres", qq(
SET enable_seqscan = off; SET enable_seqscan = off;
SET ivfflat.probes = $probes; SET ivfflat.probes = $probes;
SELECT i FROM tst ORDER BY v <-> '[0.5,0.5,0.5]' LIMIT 10; SELECT i FROM tst ORDER BY v <-> '[0.5,0.5,0.5]' LIMIT 10;
)); ));
my @actual_ids = split("\n", $actual); my @actual_ids = split("\n", $actual);
my %actual_set = map { $_ => 1 } @actual_ids; my %actual_set = map { $_ => 1 } @actual_ids;
my $count = 0; my $count = 0;
for my $el (@expected_ids) { for my $el (@expected_ids) {
if (exists($actual_set{$el})) { if (exists($actual_set{$el})) {
$count++; $count++;
} }
} }
cmp_ok($count, ">=", $min); cmp_ok($count, ">=", $min);
} }
# Initialize node # Initialize node
@@ -39,7 +39,7 @@ $node->start;
$node->safe_psql("postgres", "CREATE EXTENSION vector;"); $node->safe_psql("postgres", "CREATE EXTENSION vector;");
$node->safe_psql("postgres", "CREATE TABLE tst (i int4, v vector(3));"); $node->safe_psql("postgres", "CREATE TABLE tst (i int4, v vector(3));");
$node->safe_psql("postgres", $node->safe_psql("postgres",
"INSERT INTO tst SELECT i, ARRAY[random(), random(), random()] FROM generate_series(1,100000) i;" "INSERT INTO tst SELECT i, ARRAY[random(), random(), random()] FROM generate_series(1,100000) i;"
); );
# Get exact results # Get exact results