Added basic test for Postgres arrays

ankane · Dec 31, 2024 · 884db5e · 884db5e
1 parent cd6875b
commit 884db5e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/test/neighbor_test.rb b/test/neighbor_test.rb
@@ -14,6 +14,7 @@ def test_schema
     assert_match %{t.halfvec "half_embedding", limit: 3}, contents
     assert_match %{t.bit "binary_embedding", limit: 3}, contents
     assert_match %{t.sparsevec "sparse_embedding", limit: 3}, contents
+    assert_match %{t.vector "embeddings", limit: 3, array: true}, contents
   end
 
   def test_connection_leasing

diff --git a/test/support/postgresql.rb b/test/support/postgresql.rb
@@ -18,6 +18,7 @@ class PostgresRecord < ActiveRecord::Base
     t.bit :binary_embedding, limit: 3
     t.sparsevec :sparse_embedding, limit: 3
     t.sparsevec :sparse_factors, limit: 5
+    t.vector :embeddings, limit: 3, array: true
   end
   add_index :items, :cube_embedding, using: :gist
   add_index :items, :embedding, using: :hnsw, opclass: :vector_cosine_ops

diff --git a/test/vector_test.rb b/test/vector_test.rb
@@ -97,4 +97,10 @@ def test_nan
     end
     assert_equal "Validation failed: Embedding must have finite values", error.message
   end
+
+  def test_array
+    Item.connection.execute("INSERT INTO items (embeddings) VALUES (ARRAY['[1,2,3]', '[4,5,6]']::vector[])")
+    item = Item.last
+    assert_equal [[1, 2, 3], [4, 5, 6]], item.embeddings
+  end
 end