liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-04 03:30: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
[*.{c,h}]
[*.{c,h,pl}]
indent_style = tab
indent_size = tab
tab_width = 4

7
test/t/001_wal.pl
View File

@@ -34,9 +34,10 @@ sub test_index_replay
my $r2 = rand();
my $r3 = rand();
my $queries = qq(SET enable_seqscan=off;
SELECT * FROM tst ORDER BY v <-> '[$r1,$r2,$r3]' LIMIT 10;
);
my $queries = qq(
SET enable_seqscan = off;
SELECT * FROM tst ORDER BY v <-> '[$r1,$r2,$r3]' LIMIT 10;
);
# Run test queries and compare their result
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
sub test_recall
{
my ($probes, $min) = @_;
my ($probes, $min) = @_;
my $actual = $node->safe_psql("postgres", qq(
SET enable_seqscan = off;
SET ivfflat.probes = $probes;
SELECT i FROM tst ORDER BY v <-> '[0.5,0.5,0.5]' LIMIT 10;
));
my @actual_ids = split("\n", $actual);
my %actual_set = map { $_ => 1 } @actual_ids;
my $actual = $node->safe_psql("postgres", qq(
SET enable_seqscan = off;
SET ivfflat.probes = $probes;
SELECT i FROM tst ORDER BY v <-> '[0.5,0.5,0.5]' LIMIT 10;
));
my @actual_ids = split("\n", $actual);
my %actual_set = map { $_ => 1 } @actual_ids;
my $count = 0;
for my $el (@expected_ids) {
if (exists($actual_set{$el})) {
$count++;
}
}
my $count = 0;
for my $el (@expected_ids) {
if (exists($actual_set{$el})) {
$count++;
}
}
cmp_ok($count, ">=", $min);
cmp_ok($count, ">=", $min);
}
# Initialize node
@@ -39,7 +39,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,100000) i;"
"INSERT INTO tst SELECT i, ARRAY[random(), random(), random()] FROM generate_series(1,100000) i;"
);
# Get exact results