Added test for Sequel models

test/sequel_test.rb

@@ -10,19 +10,29 @@
   column :embedding, "vector(3)"
 end
 
+class Item < Sequel::Model(DB[:sequel_items])
+end
+
 class TestSequel < Minitest::Test
   def setup
     items.delete
   end
 
-  def test_works
+  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).all
-    assert_equal [1, 3, 2], results.map { |r| r[:id] }
     assert_equal ["[1,1,1]", "[1,1,2]", "[2,2,2]"], results.map { |r| r[:embedding] }
   end
 
+  def test_model
+    Item.create(embedding: Pgvector.encode([1, 1, 1]))
+    Item.create(embedding: Pgvector.encode([2, 2, 2]))
+    Item.create(embedding: Pgvector.encode([1, 1, 2]))
+    results = Item.order(Sequel.lit("embedding <-> ?", Pgvector.encode([1, 1, 1]))).limit(5)
+    assert_equal ["[1,1,1]", "[1,1,2]", "[2,2,2]"], results.map(&:embedding)
+  end
+
   def items
     DB[:sequel_items]
   end