From e4a764bb2185d6930d1e57750a4585878f7a7e99 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Tue, 31 Dec 2024 10:23:53 -0800 Subject: [PATCH] Added test for 2d arrays --- test/vector_test.rb | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/vector_test.rb b/test/vector_test.rb index a781928..9263b60 100644 --- 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