Improved schema test

ankane · Sep 1, 2024 · b71e9a7 · b71e9a7
1 parent 0d33151
commit b71e9a7
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/neighbor_test.rb b/test/neighbor_test.rb
@@ -8,11 +8,11 @@ def test_schema
     file.rewind
     contents = file.read
     refute_match "Could not dump table", contents
-    assert_match "t.cube", contents
-    assert_match "t.vector", contents
-    assert_match "t.halfvec", contents
-    assert_match "t.bit", contents
-    assert_match "t.sparsevec", contents
+    assert_match %{t.cube "cube_embedding"}, contents
+    assert_match %{t.vector "embedding", limit: 3}, contents
+    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
     load(file.path)
   end