Implement invert correctly

src/gt4py/next/embedded/function_field.py

@@ -82,6 +82,10 @@ def __post_init__(self):
                     f"Invariant violation: len(self.domain) ({num_params}) does not match the number of parameters of the provided function ({params})",
                 )
 
+    @property
+    def dtype(self) -> core_defs.DType[core_defs.ScalarT]:
+        return core_defs.dtype(self.ndarray.dtype.type)
+
     def restrict(self, index: common.AnyIndexSpec) -> FunctionField:
         new_domain = embedded_common.sub_domain(self.domain, index)
         return self.__class__(self.func, new_domain)
@@ -206,12 +210,14 @@ def __neg__(self) -> common.Field:
     def __abs__(self) -> common.Field:
         return self._unary_op(abs)
 
+    def __invert__(self) -> common.Field:
+        if self.dtype == core_defs.BoolDType():
+            return self._unary_op(operator.invert)
+        raise NotImplementedError("`__invert__` not implemented for non-`bool` fields.")
+
     def __call__(self, *args, **kwargs) -> common.Field:
         return self.func(*args, **kwargs)
 
-    def __invert__(self) -> common.Field:
-        raise NotImplementedError("Method invert not implemented")
-
     def remap(self, *args, **kwargs) -> common.Field:
         raise NotImplementedError("Method remap not implemented")
 

tests/next_tests/unit_tests/embedded_tests/test_function_field.py

@@ -13,7 +13,6 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 import math
 import operator
-import array
 
 import numpy as np
 import pytest
@@ -310,3 +309,16 @@ def test_function_field_builtins(function_field, builtin_name):
         assert math.isnan(np.__getattribute__(builtin_name)(3))
     else:
         assert result == np.__getattribute__(builtin_name)(3)
+
+
+def test_unary_logical_op_boolean():
+    boolean_func = lambda x: x % 2 != 0
+    field = funcf.FunctionField(boolean_func, common.Domain((I, UnitRange(1, 10))))
+    assert np.allclose(~field.ndarray, np.invert(np.fromfunction(boolean_func, (9,))))
+
+
+def test_unary_logical_op_scalar():
+    scalar_func = lambda x: x % 2
+    field = funcf.FunctionField(scalar_func, common.Domain((I, UnitRange(1, 10))))
+    with pytest.raises(NotImplementedError):
+        ~field