Improved JDBC test [skip ci]

ankane · ankane · commit a86a6c1903a9 · 2024-05-20T18:37:54.000-04:00
diff --git a/src/test/java/com/pgvector/JDBCJavaTest.java b/src/test/java/com/pgvector/JDBCJavaTest.java
@@ -17,16 +17,16 @@
 
 public class JDBCJavaTest {
     @Test
-    void testReadText() throws SQLException {
-        example(false);
+    void testVectorReadText() throws SQLException {
+        vectorExample(false);
     }
 
     @Test
-    void testReadBinary() throws SQLException {
-        example(true);
+    void testVectorReadBinary() throws SQLException {
+        vectorExample(true);
     }
 
-    void example(boolean readBinary) throws SQLException {
+    void vectorExample(boolean readBinary) throws SQLException {
         Connection conn = DriverManager.getConnection("jdbc:postgresql://localhost:5432/pgvector_java_test");
         if (readBinary) {
             conn.unwrap(PGConnection.class).setPrepareThreshold(-1);
@@ -70,8 +70,20 @@ void example(boolean readBinary) throws SQLException {
     }
 
     @Test
-    void testHalfvec() throws SQLException {
+    void testHalfvecReadText() throws SQLException {
+        halfvecExample(false);
+    }
+
+    @Test
+    void testHalfvecReadBinary() throws SQLException {
+        halfvecExample(true);
+    }
+
+    void halfvecExample(boolean readBinary) throws SQLException {
         Connection conn = DriverManager.getConnection("jdbc:postgresql://localhost:5432/pgvector_java_test");
+        if (readBinary) {
+            conn.unwrap(PGConnection.class).setPrepareThreshold(-1);
+        }
 
         Statement setupStmt = conn.createStatement();
         setupStmt.executeUpdate("CREATE EXTENSION IF NOT EXISTS vector");
