Skip to content

Commit 7fce7c8

Browse files
author
Vladlen Popolitov
committed
Test for hnsw_streaming added
1 parent 1caa395 commit 7fce7c8

File tree

2 files changed

+147
-0
lines changed

2 files changed

+147
-0
lines changed

‎test/expected/hnsw_streaming.out

Lines changed: 100 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,100 @@
1+
SELECT t,
2+
Array[t,t+1,t+2]::vector(3) AS val
3+
INTO test_where
4+
FROM generate_series(1, 2000) as t;
5+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
6+
t
7+
------
8+
1501
9+
1502
10+
(2 rows)
11+
12+
CREATE INDEX test_where_idx ON test_where USING hnsw (val vector_l2_ops) WITH (m=10);
13+
set hnsw.streaming=off;
14+
set hnsw.ef_search=100;
15+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
16+
t
17+
---
18+
(0 rows)
19+
20+
set hnsw.streaming=on;
21+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
22+
t
23+
------
24+
1501
25+
1502
26+
(2 rows)
27+
28+
SELECT t FROM test_where WHERE t>2000 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
29+
t
30+
---
31+
(0 rows)
32+
33+
DELETE FROM test_where;
34+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
35+
t
36+
---
37+
(0 rows)
38+
39+
TRUNCATE test_where;
40+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
41+
t
42+
---
43+
(0 rows)
44+
45+
DROP TABLE test_where;
46+
SET enable_seqscan = off;
47+
CREATE TABLE t (val vector(3));
48+
INSERT INTO t (val) VALUES ('[0,0,0]'), ('[1,2,3]'), ('[1,1,1]'), (NULL);
49+
CREATE INDEX t_idx ON t USING hnsw (val vector_l2_ops);
50+
INSERT INTO t (val) VALUES ('[1,2,4]');
51+
SELECT * FROM t ORDER BY val <-> '[3,3,3]';
52+
val
53+
---------
54+
[1,2,3]
55+
[1,2,4]
56+
[1,1,1]
57+
[0,0,0]
58+
(4 rows)
59+
60+
SELECT COUNT(*) FROM (SELECT * FROM t ORDER BY val <-> (SELECT NULL::vector)) t2;
61+
count
62+
-------
63+
4
64+
(1 row)
65+
66+
SELECT COUNT(*) FROM t;
67+
count
68+
-------
69+
5
70+
(1 row)
71+
72+
TRUNCATE t;
73+
SELECT * FROM t ORDER BY val <-> '[3,3,3]';
74+
val
75+
-----
76+
(0 rows)
77+
78+
DROP TABLE t;
79+
SET enable_seqscan = off;
80+
CREATE TABLE t (val vector(3));
81+
INSERT INTO t (val) VALUES ('[0,0,0]'), ('[1,2,3]'), ('[1,1,1]'), (NULL);
82+
CREATE INDEX t_idx ON t USING hnsw (val vector_ip_ops);
83+
INSERT INTO t (val) VALUES ('[1,2,4]');
84+
SELECT * FROM t ORDER BY val <#> '[3,3,3]';
85+
val
86+
---------
87+
[1,2,4]
88+
[1,2,3]
89+
[1,1,1]
90+
[0,0,0]
91+
(4 rows)
92+
93+
SELECT COUNT(*) FROM (SELECT * FROM t ORDER BY val <#> (SELECT NULL::vector)) t2;
94+
count
95+
-------
96+
4
97+
(1 row)
98+
99+
DROP TABLE t;
100+
set hnsw.streaming=off;

‎test/sql/hnsw_streaming.sql

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
SELECT t,
2+
Array[t,t+1,t+2]::vector(3) AS val
3+
INTO test_where
4+
FROM generate_series(1, 2000) as t;
5+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
6+
CREATE INDEX test_where_idx ON test_where USING hnsw (val vector_l2_ops) WITH (m=10);
7+
set hnsw.streaming=off;
8+
set hnsw.ef_search=100;
9+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
10+
set hnsw.streaming=on;
11+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
12+
SELECT t FROM test_where WHERE t>2000 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
13+
DELETE FROM test_where;
14+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
15+
TRUNCATE test_where;
16+
SELECT t FROM test_where WHERE t>1500 ORDER BY val <-> '[1000.2,1000,1000.1]' LIMIT 2;
17+
DROP TABLE test_where;
18+
SET enable_seqscan = off;
19+
20+
CREATE TABLE t (val vector(3));
21+
INSERT INTO t (val) VALUES ('[0,0,0]'), ('[1,2,3]'), ('[1,1,1]'), (NULL);
22+
CREATE INDEX t_idx ON t USING hnsw (val vector_l2_ops);
23+
24+
INSERT INTO t (val) VALUES ('[1,2,4]');
25+
26+
SELECT * FROM t ORDER BY val <-> '[3,3,3]';
27+
SELECT COUNT(*) FROM (SELECT * FROM t ORDER BY val <-> (SELECT NULL::vector)) t2;
28+
SELECT COUNT(*) FROM t;
29+
30+
TRUNCATE t;
31+
SELECT * FROM t ORDER BY val <-> '[3,3,3]';
32+
33+
DROP TABLE t;
34+
SET enable_seqscan = off;
35+
36+
CREATE TABLE t (val vector(3));
37+
INSERT INTO t (val) VALUES ('[0,0,0]'), ('[1,2,3]'), ('[1,1,1]'), (NULL);
38+
CREATE INDEX t_idx ON t USING hnsw (val vector_ip_ops);
39+
40+
INSERT INTO t (val) VALUES ('[1,2,4]');
41+
42+
SELECT * FROM t ORDER BY val <#> '[3,3,3]';
43+
SELECT COUNT(*) FROM (SELECT * FROM t ORDER BY val <#> (SELECT NULL::vector)) t2;
44+
45+
DROP TABLE t;
46+
47+
set hnsw.streaming=off;

0 commit comments

Comments
 (0)
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy