Updated tests for JRuby

Author: ankane

Gemfile

@@ -4,7 +4,8 @@ gemspec
 
 gem "rake"
 gem "minitest"
-gem "numo-narray"
-gem "pg"
+gem "numo-narray", platform: :ruby
+gem "pg", platform: :ruby
+gem "jdbc-postgres", platform: :jruby
 gem "sequel"
 gem "logger"

lib/sequel/plugins/pgvector.rb

@@ -76,7 +76,8 @@ def []=(k, v)
 
         def [](k)
           if self.class.vector_columns.key?(k.to_sym)
-            ::Pgvector.decode(super)
+            # to_s needed for JRuby
+            ::Pgvector.decode(super.to_s)
           else
             super
           end

test/pg_test.rb

@@ -2,6 +2,8 @@
 
 class PgTest < Minitest::Test
   def setup
+    skip if RUBY_PLATFORM == "java"
+
     conn.exec("DELETE FROM pg_items")
   end
 

test/sequel_test.rb

@@ -1,6 +1,7 @@
 require_relative "test_helper"
 
-DB = Sequel.connect("postgres://localhost/pgvector_ruby_test")
+adapter = RUBY_PLATFORM == "java" ? "jdbc:postgresql" : "postgres"
+DB = Sequel.connect("#{adapter}://localhost/pgvector_ruby_test")
 
 if ENV["VERBOSE"]
   require "logger"
@@ -35,14 +36,14 @@ def test_dataset
     items.insert(embedding: Pgvector.encode([1, 1, 1]))
     items.multi_insert([{embedding: "[2,2,2]"}, {embedding: "[1,1,2]"}])
     results = items.order(Sequel.lit("embedding <-> ?", Pgvector.encode([1, 1, 1]))).limit(5)
-    assert_equal [[1, 1, 1], [1, 1, 2], [2, 2, 2]], results.map { |r| Pgvector.decode(r[:embedding]) }
+    assert_equal [[1, 1, 1], [1, 1, 2], [2, 2, 2]], results.map { |r| Pgvector.decode(r[:embedding].to_s) }
   end
 
   def test_extension
     items.insert(embedding: Pgvector.encode([1, 1, 1]))
     items.multi_insert([{embedding: "[2,2,2]"}, {embedding: "[1,1,2]"}])
     results = items.extension(:pgvector).nearest_neighbors(:embedding, [1, 1, 1], distance: "euclidean").limit(5)
-    assert_equal [[1, 1, 1], [1, 1, 2], [2, 2, 2]], results.map { |r| Pgvector.decode(r[:embedding]) }
+    assert_equal [[1, 1, 1], [1, 1, 2], [2, 2, 2]], results.map { |r| Pgvector.decode(r[:embedding].to_s) }
   end
 
   def test_model_vector_euclidean

test/vector_test.rb

@@ -12,6 +12,8 @@ def test_to_a
   end
 
   def test_numo
+    skip if RUBY_PLATFORM == "java"
+
     a = Pgvector::Vector.new([1, 2, 3])
     b = Pgvector::Vector.new(Numo::NArray.cast([1, 2, 3]))
     assert_equal a.to_s, b.to_s