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

Improved test [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2022-12-30 19:28:12 -08:00
parent d4681dd898
commit 8dbd39f641
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/t/008_avg.pl
View File

@@ -2,7 +2,7 @@ use strict;
use warnings; use warnings;
use PostgresNode; use PostgresNode;
use TestLib; use TestLib;
use Test::More tests => 2; use Test::More tests => 5;
# Initialize node # Initialize node
my $node = get_new_node('node'); my $node = get_new_node('node');
@@ -14,7 +14,7 @@ $node->safe_psql("postgres", "CREATE EXTENSION vector;");
$node->safe_psql("postgres", "CREATE TABLE tst (r1 real, r2 real, r3 real, v vector(3));"); $node->safe_psql("postgres", "CREATE TABLE tst (r1 real, r2 real, r3 real, v vector(3));");
$node->safe_psql("postgres", $node->safe_psql("postgres",
"INSERT INTO tst SELECT r1, r2, r3, ARRAY[r1, r2, r3] FROM ( "INSERT INTO tst SELECT r1, r2, r3, ARRAY[r1, r2, r3] FROM (
SELECT random() + 1 AS r1, random() + 2 AS r2, random() + 3 AS r3 FROM generate_series(1, 1000000) t SELECT random() + 1.01 AS r1, random() + 2.01 AS r2, random() + 3.01 AS r3 FROM generate_series(1, 1000000) t
) i;" ) i;"
); );
@@ -24,6 +24,9 @@ my $r2 = $node->safe_psql("postgres", "SELECT AVG(r2)::float4 FROM tst;");
my $r3 = $node->safe_psql("postgres", "SELECT AVG(r3)::float4 FROM tst;"); my $r3 = $node->safe_psql("postgres", "SELECT AVG(r3)::float4 FROM tst;");
my $avg = $node->safe_psql("postgres", "SELECT AVG(v) FROM tst;"); my $avg = $node->safe_psql("postgres", "SELECT AVG(v) FROM tst;");
is($avg, "[$r1,$r2,$r3]"); is($avg, "[$r1,$r2,$r3]");
like($avg, qr/\[1\.5/);
like($avg, qr/,2\.5/);
like($avg, qr/,3\.5/);
# Test explain # Test explain
my $explain = $node->safe_psql("postgres", "EXPLAIN SELECT AVG(v) FROM tst;"); my $explain = $node->safe_psql("postgres", "EXPLAIN SELECT AVG(v) FROM tst;");