Added test for type

ankane · Dec 11, 2023 · fa294f4 · fa294f4
1 parent cb186c4
commit fa294f4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/test/neighbor_test.rb b/test/neighbor_test.rb
@@ -215,6 +215,16 @@ def test_neighbor_attributes
     assert_equal Item.neighbor_attributes.keys, [:embedding, :neighbor_vector]
   end
 
+  def test_type
+    if vector?
+      Item.create!(factors: "[1,2,3]")
+      assert_equal "[1,2,3]", Item.last.factors
+    else
+      Item.create!(factors: "(1,2,3)")
+      assert_equal "(1, 2, 3)", Item.last.factors
+    end
+  end
+
   def create_items(cls, attribute = :embedding)
     vectors = [
       [1, 1, 1],

diff --git a/test/support/cube.rb b/test/support/cube.rb
@@ -4,6 +4,7 @@
   create_table :items, force: true do |t|
     t.cube :embedding
     t.cube :neighbor_vector
+    t.cube :factors
   end
 end
 

diff --git a/test/support/vector.rb b/test/support/vector.rb
@@ -4,6 +4,7 @@
   create_table :items, force: true do |t|
     t.vector :embedding, limit: 3
     t.vector :neighbor_vector, limit: 3
+    t.vector :factors, limit: 3
   end
 end