tests: Fix localsize name in test_petsc

ZoeLeibowitz · Sep 30, 2024 · 62fd454 · 62fd454
1 parent ff8d580
commit 62fd454
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_petsc.py b/tests/test_petsc.py
@@ -572,7 +572,7 @@ def test_start_ptr():
         op1 = Operator(petsc1)
 
     # Verify the case with modulo time stepping
-    assert 'float * start_ptr_0 = t1*localsize + (float *)(u1_vec->data);' in str(op1)
+    assert 'float * start_ptr_0 = t1*localsize_0 + (float *)(u1_vec->data);' in str(op1)
 
     # Verify the case with no modulo time stepping
     u2 = TimeFunction(name='u2', grid=grid, space_order=2, dtype=np.float32, save=5)
@@ -582,7 +582,7 @@ def test_start_ptr():
     with switchconfig(openmp=False):
         op2 = Operator(petsc2)
 
-    assert 'float * start_ptr_0 = (time + 1)*localsize + ' + \
+    assert 'float * start_ptr_0 = (time + 1)*localsize_0 + ' + \
         '(float *)(u2_vec->data);' in str(op2)