Added new tests for two consecutive adjoint runs (mdolab#362)

* Added tests for running adjoints back to back * clean-up formatting * Double adjoint test now tests against single adjoint results. Black and Flake8 fixes. * Reran tests on docker * Remove extra files * Black formatting fixes * Fixed reference for 1 test * Update remaining references * Black formatting fixes --------- Co-authored-by: Andrew Lamkin <[email protected]>
DavidAnderegg · Dec 20, 2024 · 881ac51 · 881ac51
1 parent 3997bfa
commit 881ac51
Show file tree

Hide file tree

Showing 6 changed files with 33 additions and 0 deletions.
diff --git a/tests/reg_tests/refs/adjoint_euler_scalar_jst_tut_wing.json b/tests/reg_tests/refs/adjoint_euler_scalar_jst_tut_wing.json
@@ -1,4 +1,5 @@
 {
+    "Adjoint States": 0.0,
     "Eval Functions Sens:": {
         "mdo_tutorial_cd": {
             "P_mdo_tutorial": 6.776263578034403e-21,

diff --git a/tests/reg_tests/refs/adjoint_laminar_tut_wing.json b/tests/reg_tests/refs/adjoint_laminar_tut_wing.json
@@ -1,4 +1,5 @@
 {
+    "Adjoint States": 0.0,
     "Eval Functions Sens:": {
         "mdo_tutorial_cd": {
             "P_mdo_tutorial": 1.5227077123299861e-05,

diff --git a/tests/reg_tests/refs/adjoint_rans_rotating.json b/tests/reg_tests/refs/adjoint_rans_rotating.json
@@ -1,4 +1,5 @@
 {
+    "Adjoint States": 0.0,
     "Eval Functions Sens:": {
         "mdo_tutorial_fy": {
             "T_mdo_tutorial": 0.0009508864923091359,

diff --git a/tests/reg_tests/refs/adjoint_rans_tut_wing.json b/tests/reg_tests/refs/adjoint_rans_tut_wing.json
@@ -1,4 +1,5 @@
 {
+    "Adjoint States": 0.0,
     "Eval Functions Sens:": {
         "mdo_tutorial_cavitation": {
             "P_mdo_tutorial": -3.3530531445605923e-10,

diff --git a/tests/reg_tests/reg_test_utils.py b/tests/reg_tests/reg_test_utils.py
@@ -14,6 +14,27 @@ def assert_adjoint_sens_allclose(handler, CFDSolver, ap, evalFuncs=None, **kwarg
     handler.root_add_dict("Eval Functions Sens:", funcsSens, rtol=rtol, atol=atol)
 
 
+def assert_adjoint2_sens_allclose(handler, CFDSolver, ap, evalFuncs=None, **kwargs):
+    rtol, atol = getTol(**kwargs)
+    funcsSens = {}
+    CFDSolver.evalFunctionsSens(ap, funcsSens, evalFuncs=None)
+    CFDSolver.evalFunctionsSens(ap, funcsSens, evalFuncs=None)
+    handler.root_print("Eval Functions Sens:")
+    handler.root_add_dict("Eval Functions Sens:", funcsSens, rtol=rtol, atol=atol)
+
+
+def assert_adjoint_states_allclose(handler, CFDSolver, ap, evalFuncs=None, **kwargs):
+    rtol, atol = getTol(**kwargs)
+    funcsSens = {}
+    CFDSolver.evalFunctionsSens(ap, funcsSens, evalFuncs=None)
+    states1 = CFDSolver.getStates()
+    CFDSolver.evalFunctionsSens(ap, funcsSens, evalFuncs=None)
+    states2 = CFDSolver.getStates()
+    resNorm = numpy.sqrt((1 / CFDSolver.getStateSize()) * numpy.sum((states1 - states2) ** 2))
+    handler.root_print("Adjoint States")
+    handler.par_add_norm("Adjoint States", resNorm, rtol=rtol, atol=atol)
+
+
 def assert_problem_size_equal(handler, CFDSolver, **kwargs):
     rtol, atol = getTol(**kwargs)
     # Now a few simple checks

diff --git a/tests/reg_tests/test_adjoint.py b/tests/reg_tests/test_adjoint.py
@@ -276,6 +276,14 @@ def test_adjoint(self):
         utils.assert_adjoint_sens_allclose(self.handler, self.CFDSolver, self.ap, tol=1e-10)
         self.assert_adjoint_failure()
 
+    def test_adjoint2(self):
+        utils.assert_adjoint2_sens_allclose(self.handler, self.CFDSolver, self.ap, tol=1e-10)
+        self.assert_adjoint_failure()
+
+    def test_adjoint_states(self):
+        utils.assert_adjoint_states_allclose(self.handler, self.CFDSolver, self.ap, tol=1e-10)
+        self.assert_adjoint_failure()
+
 
 @parameterized_class(test_params)
 class TestCmplxStep(reg_test_classes.CmplxRegTest):