@@ -124,34 +124,47 @@ mod tests {
124124
125125 #[ tokio:: test]
126126 async fn tokio_works ( ) -> Result < ( ) , tokio_postgres:: Error > {
127- let ( client, connection) =
128- tokio_postgres:: connect ( "host=localhost dbname=pgvector_rust_test" , tokio_postgres:: NoTls ) . await ?;
127+ let ( client, connection) = tokio_postgres:: connect (
128+ "host=localhost dbname=pgvector_rust_test" ,
129+ tokio_postgres:: NoTls ,
130+ )
131+ . await ?;
129132
130133 tokio:: spawn ( async move {
131134 if let Err ( e) = connection. await {
132135 eprintln ! ( "connection error: {}" , e) ;
133136 }
134137 } ) ;
135138
136- client. execute ( "CREATE EXTENSION IF NOT EXISTS vector" , & [ ] ) . await ?;
137- client. execute ( "DROP TABLE IF EXISTS tokio_postgres_items" , & [ ] ) . await ?;
138- client. execute (
139- "CREATE TABLE tokio_postgres_items (id bigserial PRIMARY KEY, embedding vector(3))" ,
140- & [ ] ,
141- ) . await ?;
139+ client
140+ . execute ( "CREATE EXTENSION IF NOT EXISTS vector" , & [ ] )
141+ . await ?;
142+ client
143+ . execute ( "DROP TABLE IF EXISTS tokio_postgres_items" , & [ ] )
144+ . await ?;
145+ client
146+ . execute (
147+ "CREATE TABLE tokio_postgres_items (id bigserial PRIMARY KEY, embedding vector(3))" ,
148+ & [ ] ,
149+ )
150+ . await ?;
142151
143152 let vec = Vector :: from ( vec ! [ 1.0 , 2.0 , 3.0 ] ) ;
144153 let vec2 = Vector :: from ( vec ! [ 4.0 , 5.0 , 6.0 ] ) ;
145- client. execute (
146- "INSERT INTO tokio_postgres_items (embedding) VALUES ($1), ($2), (NULL)" ,
147- & [ & vec, & vec2] ,
148- ) . await ?;
154+ client
155+ . execute (
156+ "INSERT INTO tokio_postgres_items (embedding) VALUES ($1), ($2), (NULL)" ,
157+ & [ & vec, & vec2] ,
158+ )
159+ . await ?;
149160
150161 let query_vec = Vector :: from ( vec ! [ 3.0 , 1.0 , 2.0 ] ) ;
151- let row = client. query_one (
152- "SELECT embedding FROM tokio_postgres_items ORDER BY embedding <-> $1 LIMIT 1" ,
153- & [ & query_vec] ,
154- ) . await ?;
162+ let row = client
163+ . query_one (
164+ "SELECT embedding FROM tokio_postgres_items ORDER BY embedding <-> $1 LIMIT 1" ,
165+ & [ & query_vec] ,
166+ )
167+ . await ?;
155168 let res_vec: Vector = row. get ( 0 ) ;
156169 assert_eq ! ( vec, res_vec) ;
157170 assert_eq ! ( vec![ 1.0 , 2.0 , 3.0 ] , res_vec. to_vec( ) ) ;
0 commit comments