tests: Add test for python-level printing to improve coverage

Author: EdCaunt

devito/ir/equations/equation.py

@@ -49,11 +49,9 @@ def directions(self):
     @property
     def dtype(self):
         try:
-            rhs_dtype = self.rhs.dtype
+            return infer_dtype({self.lhs.dtype, self.rhs.dtype} - {None})
         except AttributeError:
-            rhs_dtype = None
-
-        return infer_dtype({self.lhs.dtype, rhs_dtype} - {None})
+            return self.lhs.dtype
 
     @property
     def state(self):

devito/passes/iet/languages/C.py

@@ -57,11 +57,9 @@ def _print_ImaginaryUnit(self, expr):
         return '_Complex_I'
 
     def _print_Re(self, expr):
-        """Print an Re as an access into the second entry of a float array."""
         return (f'{self.func_prefix(expr)}real{self.func_literal(expr).lower()}'
                 f'({self._print(expr.args[0])})')
 
     def _print_Im(self, expr):
-        """Print an Im as an access into the second entry of a float array."""
         return (f'{self.func_prefix(expr)}imag{self.func_literal(expr).lower()}'
                 f'({self._print(expr.args[0])})')

tests/test_symbolics.py

@@ -890,6 +890,12 @@ def run_operator(self, eqs, language):
         with switchconfig(language=language):
             Operator(eqs)()
 
+    def test_devito_print(self):
+        f, _, _ = self.setup_basic(np.complex64)
+
+        assert str(Re(f)) == 'Re(f(x))'
+        assert str(Im(f)) == 'Im(f(x))'
+
     @pytest.mark.parametrize('language', ['C', 'CXX', 'CXXopenmp'])
     def test_printing(self, language):
         f, f_real, f_imag = self.setup_basic(np.complex64)