tests: Add initial test for specialization at operator apply

Author: EdCaunt

devito/operator/operator.py

@@ -1014,6 +1014,8 @@ def apply(self, **kwargs):
                                   if k not in specialize}
 
             # TODO: Does this cause problems for profilers?
+            # FIXME: Need some way to inspect this Operator for testing
+            # FIXME: Perhaps this should use some separate method
             return op.apply(**specialized_kwargs)
 
         # Invoke kernel function with args

tests/test_specialization.py

@@ -203,14 +203,23 @@ def test_apply_basic(self):
         assert np.all(check == f.data)
 
 
-# class TestApply:
-#     """Tests for specialization of operators at apply time"""
+class TestApply:
+    """Tests for specialization of operators at apply time"""
 
-#     def test_basic(self):
-#         grid = Grid(shape=(11, 11))
+    def test_basic(self):
+        grid = Grid(shape=(11, 11))
+        f = Function(name='f', grid=grid, dtype=np.int32)
+        op = Operator(Eq(f, f+1))
+
+        # TODO: Need to verify that specialized operator is actually the one
+        # being run. How can I achieve this?
+        op.apply(specialize=('x_m', 'x_M'))
 
-#         f = TimeFunction(name='f', grid=grid, space_order=2)
+        check = np.array(f.data[:])
+        f.data[:] = 0
+        op.apply()
 
-#         op = Operator(Eq(f.forward, f + 1))
+        assert np.all(check == f.data)
 
-#         op.apply(time_M=10, specialize=('x_m', 'x_M'))
+# Need to test combining specialization and overrides (a range of them)
+# Need to test specialization with MPI (both at)