Added array_contains test to regress 1115

lancedb · Jan 8, 2024 · c07a427 · c07a427
1 parent 7440e4c
commit c07a427
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/python/tests/test_filter.py b/python/python/tests/test_filter.py
@@ -195,12 +195,13 @@ def test_functions(tmp_path: Path):
     expected = table.slice(1, 2)
     dataset = lance.write_dataset(table, tmp_path / "test_neg_expr")
     assert (
-        dataset.scanner(
-            filter="array_has_any(genres, Array['anime', 'adventure'])"
-        ).to_table()
+        dataset.to_table(filter="array_has_any(genres, Array['anime', 'adventure'])")
         == expected
     )
 
+    expected = table.slice(0, 1)
+    assert dataset.to_table(filter="array_contains(genres, 'comedy')") == expected
+
 
 def test_negative_expressions(tmp_path: Path):
     table = pa.table({"x": [-1, 0, 1, 1], "y": [1, 2, 3, 4]})