compiler: Add unevaluable Mod

Author: FabioLuporini

devito/symbolics/unevaluation.py

@@ -1,6 +1,6 @@
 import sympy
 
-__all__ = ['Add', 'Mul', 'Pow']
+__all__ = ['Add', 'Mod', 'Mul', 'Pow']
 
 
 class UnevaluableMixin:
@@ -13,6 +13,10 @@ class Add(sympy.Add, UnevaluableMixin):
     __new__ = UnevaluableMixin.__new__
 
 
+class Mod(sympy.Mod, UnevaluableMixin):
+    __new__ = UnevaluableMixin.__new__
+
+
 class Mul(sympy.Mul, UnevaluableMixin):
     __new__ = UnevaluableMixin.__new__
 

tests/test_pickle.py

@@ -22,6 +22,7 @@
     CallFromPointer, Cast, DefFunction, FieldFromPointer, IntDiv, ListInitializer, SizeOf,
     indexify, pow_to_mul
 )
+from devito.symbolics.unevaluation import Mod
 from devito.tools import EnrichedTuple
 from devito.types import (
     Array, ComponentAccess, CustomDimension, DefaultDimension, DeviceID, FIndexed,
@@ -720,6 +721,15 @@ def test_local_sum(self, pickle):
             tuple(map(str, reduction.conditionals.values()))
         assert rebuilt.free_symbols.isdisjoint(rebuilt.bound_symbols)
 
+    @pytest.mark.parametrize('args', [(0, 3), (1, 3), (3, 3), (Symbol('i'), 3)])
+    def test_unevaluated_mod(self, pickle, args):
+        expr = Mod(*args)
+
+        rebuilt = pickle.loads(pickle.dumps(expr))
+
+        assert rebuilt == expr
+        assert rebuilt.args == args
+
 
 class TestAdvanced: