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

Fixed tests

Browse Source
This commit is contained in:
Andrew Kane
2022-02-06 20:14:21 -08:00
parent 062e220936
commit d5da22f24b
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/t/003_recall.pl
View File

@@ -10,9 +10,13 @@ my @expected_ids;
# TODO Run more queries to prevent flakiness # TODO Run more queries to prevent flakiness
sub test_recall sub test_recall
{ {
my ($min) = @_; my ($probes, $min) = @_;
my $actual = $node->safe_psql("postgres", "SELECT i FROM tst ORDER BY v <-> '[0.5,0.5,0.5]' LIMIT 10;"); 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_ids = split("\n", $actual);
my %actual_set = map { $_ => 1 } @actual_ids; my %actual_set = map { $_ => 1 } @actual_ids;
@@ -46,9 +50,7 @@ my $expected = $node->safe_psql("postgres", "SELECT i FROM tst ORDER BY v <-> '[
$node->safe_psql("postgres", "CREATE INDEX ON tst USING ivfflat (v);"); $node->safe_psql("postgres", "CREATE INDEX ON tst USING ivfflat (v);");
# Test approximate results # Test approximate results
$node->safe_psql("postgres", "SET enable_seqscan = off;"); test_recall(1, 5);
test_recall(5);
# Test probes # Test probes
$node->safe_psql("postgres", "SET ivfflat.probes = 100"); test_recall(100, 10);
test_recall(10);