Added test for 2d arrays

ankane · Dec 31, 2024 · e4a764b · e4a764b
1 parent 1648762
commit e4a764b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/vector_test.rb b/test/vector_test.rb
@@ -103,4 +103,10 @@ def test_array
     assert_equal [[1, 2, 3], [4, 5, 6]], item.embeddings
     assert_equal [[1, 2, 3], [4, 5, 6]], Item.last.embeddings
   end
+
+  def test_array_2d
+    item = Item.create!(embeddings: [[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
+    assert_equal [[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]], item.embeddings
+    assert_equal [[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]], Item.last.embeddings
+  end
 end