noodle number 4

Author: JDBetteridge

tests/test_dimension.py

@@ -1492,8 +1492,22 @@ def test_stepping_dim_in_condition_lowering(self):
         op.apply(time_M=threshold+3)
         assert np.all(g.data[0, :, :] == threshold)
         assert np.all(g.data[1, :, :] == threshold + 1)
-        assert 'if (g[t0][x + 1][y + 1] <= 10)\n        {' + \
-            '\n          g[t1][x + 1][y + 1] = g[t0][x + 1][y + 1] + 1' in str(op.ccode)
+
+        # We want to assert that the following snippet:
+        # ```
+        #   if (g[t0][x + 1][y + 1] <= 10)
+        #   {
+        #     g[t1][x + 1][y + 1] = g[t0][x + 1][y + 1] + 1
+        # ```
+        # is in the generated code, but indentation etc. seems to vary.
+        part1 = 'if (g[t0][x + 1][y + 1] <= 10)\n'
+        part2 = 'g[t1][x + 1][y + 1] = g[t0][x + 1][y + 1] + 1'
+        whole_code = str(op.ccode)
+
+        loc = whole_code.find(part1)
+        assert loc != -1
+        loc = whole_code.find(part2, loc + len(part1))
+        assert loc != -1
 
     def test_expr_like_lowering(self):
         """