add dJdm to taylor_test calls

firedrakeproject · Jul 12, 2024 · 8645c6d · 8645c6d
1 parent e011ac7
commit 8645c6d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/ensemble_reduced_functional/test_reduced_functional.py b/tests/ensemble_reduced_functional/test_reduced_functional.py
@@ -38,7 +38,6 @@ def test_verification():
     dJdm = rf.derivative()
     assert_allclose(ensemble_J, size, rtol=1e-12)
     assert_allclose(dJdm.dat.data_ro, 2.0 * size, rtol=1e-12)
-    dJdm = ensemble.ensemble_comm.allreduce(sendobj=dJdm, op=MPI.SUM)
     assert taylor_test(rf, x, Function(R, val=0.1)) > 1.9
 
 
@@ -63,7 +62,7 @@ def test_verification_gather_functional_adjfloat():
     assert_allclose(ensemble_J, 1.0**4+2.0**4, rtol=1e-12)
     assert_allclose(dJdm.dat.data_ro, 4*(rank+1)**3, rtol=1e-12)
     dJdm = ensemble.ensemble_comm.allreduce(sendobj=dJdm, op=MPI.SUM)
-    assert taylor_test(rf, x, Function(R, val=0.1)) > 1.9
+    assert taylor_test(rf, x, Function(R, val=0.1), dJdm=dJdm) > 1.9
 
 
 @pytest.mark.parallel(nprocs=4)
@@ -88,7 +87,7 @@ def test_verification_gather_functional_Function():
     assert_allclose(ensemble_J, 1.0**4+2.0**4, rtol=1e-12)
     assert_allclose(dJdm.dat.data_ro, 4*(rank+1)**3, rtol=1e-12)
     dJdm = ensemble.ensemble_comm.allreduce(sendobj=dJdm, op=MPI.SUM)
-    assert taylor_test(rf, x, Function(R, val=0.1)) > 1.9
+    assert taylor_test(rf, x, Function(R, val=0.1), dJdm=dJdm) > 1.9
 
 
 @pytest.mark.parallel(nprocs=6)