Merge pull request #2973 from mabel-dev/copilot/fix-cannot-negate-operator

Fix NOT operator with array containment operators (@>, @>>)

Author: joocer

opteryx/managers/expression/__init__.py

@@ -267,6 +267,13 @@ def _inner_evaluate(root: Node, table: Table):
                 if root.centre
                 else pyarrow.nulls(1, type=pyarrow.bool_())
             )
+            # Convert to numpy array if it's not already a PyArrow array
+            # This handles memoryviews, Cython memoryviewslices, and other array-like objects
+            if not isinstance(centre, pyarrow.Array):
+                centre = numpy.asarray(centre)
+                # Convert numeric types (e.g., uint8 from list_contains_any) to boolean
+                if numpy.issubdtype(centre.dtype, numpy.integer):
+                    centre = centre.astype(numpy.bool_)
             centre = pyarrow.array(centre, type=pyarrow.bool_())
             return pyarrow.compute.invert(centre)
 

tests/integration/sql_battery/test_shapes_aliases_distinct.py

@@ -229,6 +229,7 @@
         ("SELECT missions FROM $astronauts WHERE ARRAY_CONTAINS_ALL(missions, ('Apollo 8', 'Gemini 7'))", 2, 1, None),
         ("SELECT missions FROM $astronauts WHERE ARRAY_CONTAINS_ALL(missions, ('Gemini 7', 'Apollo 8'))", 2, 1, None),
         ("SELECT missions FROM $astronauts WHERE missions @> ('Apollo 8', 'Apollo 13')", 5, 1, None),
+        ("SELECT * FROM (SELECT name, IFNULL(missions, []) AS missions FROM $astronauts) WHERE NOT missions @> ['Apollo 11']", 354, 2, None),
         ("SELECT * FROM $astronauts WHERE missions @>> ('Apollo 11', 'Gemini 12')", 1, 19, None),
         ("SELECT * FROM $astronauts WHERE missions @>> ('Gemini 7', 'Apollo 8')", 2, 19, None),