Improved test [skip ci]

ankane · ankane · commit 27272961e525 · 2023-10-28T16:27:00.000-07:00
diff --git a/src/test/java/com/pgvector/JDBCJavaTest.java b/src/test/java/com/pgvector/JDBCJavaTest.java
@@ -54,6 +54,8 @@ private void example(boolean read_binary) throws SQLException {
         }
         assertEquals(Arrays.asList(new Long[]{1L, 3L, 2L, 4L}), ids);
         assertArrayEquals(new float[] {1, 1, 1}, embeddings.get(0).toArray());
+        assertArrayEquals(new float[] {1, 1, 2}, embeddings.get(1).toArray());
+        assertArrayEquals(new float[] {2, 2, 2}, embeddings.get(2).toArray());
 
         Statement indexStmt = conn.createStatement();
         indexStmt.executeUpdate("CREATE INDEX ON jdbc_items USING ivfflat (embedding vector_l2_ops) WITH (lists = 100)");

Original file line number	Diff line number	Diff line change
`@@ -54,6 +54,8 @@ private void example(boolean read_binary) throws SQLException {`
`54`	`54`	`}`
`55`	`55`	`assertEquals(Arrays.asList(new Long[]{1L, 3L, 2L, 4L}), ids);`
`56`	`56`	`assertArrayEquals(new float[] {1, 1, 1}, embeddings.get(0).toArray());`
	`57`	`+ assertArrayEquals(new float[] {1, 1, 2}, embeddings.get(1).toArray());`
	`58`	`+ assertArrayEquals(new float[] {2, 2, 2}, embeddings.get(2).toArray());`
`57`	`59`
`58`	`60`	`Statement indexStmt = conn.createStatement();`
`59`	`61`	`indexStmt.executeUpdate("CREATE INDEX ON jdbc_items USING ivfflat (embedding vector_l2_ops) WITH (lists = 100)");`