Skip to content

Commit 5804516

Browse files
committed
Ran cargo fmt
1 parent 4d266d1 commit 5804516

1 file changed

Lines changed: 29 additions & 16 deletions

File tree

src/postgres_ext/vector.rs

Lines changed: 29 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)