tests: Add trigonometric

devito/ir/xdsl_iet/cluster_to_ssa.py

@@ -290,7 +290,6 @@ def _visit_math_nodes(self, dim: SteppingDimension, node: Expr,
 
         elif isinstance(node, sin):
             assert len(node.args) == 1, "Expected single argument for sin."
-            #import pdb; pdb.set_trace()
             return math.SinOp(self._visit_math_nodes(dim, node.args[0],
                               output_indexed)).result
 

tests/test_xdsl_base.py

@@ -1,8 +1,9 @@
 import numpy as np
 import pytest
+from sympy import S
 
 from devito import (Grid, TensorTimeFunction, VectorTimeFunction, div, grad, diag, solve,
-                    Operator, Eq, Constant, norm, SpaceDimension, switchconfig)
+                    Operator, Eq, Constant, norm, SpaceDimension, switchconfig, sin, cos)
 from devito.types import Array, Function, TimeFunction
 from devito.tools import as_tuple
 
@@ -970,6 +971,37 @@ def test_function_IV():
     assert np.isclose(norm(u), devito_norm_u)
 
 
+class TestTrigonometric(object):
+
+    @pytest.mark.parametrize('deg, exp', ([90.0, 3.5759869], [30.0, 3.9521265],
+                             [45.0, 3.403614]))
+    def test_sine(self, deg, exp):
+        grid = Grid(shape=(4, 4))
+
+        u = Function(name="u", grid=grid)
+        u.data[:, :] = 0
+
+        eq0 = Eq(u, sin(deg))
+
+        op = Operator([eq0], opt='xdsl')
+        op.apply()
+        assert np.isclose(norm(u), exp, rtol=1e-4)
+
+    @pytest.mark.parametrize('deg, exp', ([90.0, 1.7922944], [30.0, 0.6170056],
+                             [45.0, 2.101288]))
+    def test_cosine(self, deg, exp):
+        grid = Grid(shape=(4, 4))
+
+        u = Function(name="u", grid=grid)
+        u.data[:, :] = 0
+
+        eq0 = Eq(u, cos(deg))
+
+        op = Operator([eq0], opt='xdsl')
+        op.apply()
+        assert np.isclose(norm(u), exp, rtol=1e-4)
+
+
 class TestOperatorUnsupported(object):
 
     @pytest.mark.xfail(reason="stencil.return operation does not verify for i64")