class TestJacVecBWDFast(reg_test_classes.RegTest):
"""
Tests that given a flow state the state jacobian vector products are accurate.
"""
N_PROCS = 2
def setUp(self):
if not hasattr(self, "name"):
# return immediately when the setup method is being called on the based class and NOT the
# classes created using parametrized
# this will happen when testing, but will hopefully be fixed down the line
return
super().setUp()
options = copy.copy(adflowDefOpts)
options["outputdirectory"] = os.path.join(baseDir,
options["outputdirectory"])
options.update(self.options)
# Create the solver
self.CFDSolver = ADFLOW(options=copy.deepcopy(options), debug=True)
self.ap = copy.deepcopy(self.aero_prob)
# add the default dvs to the problem
for dv in defaultAeroDVs:
self.ap.addDV(dv)
# propagates the values from the restart file throughout the code
self.CFDSolver.getResidual(self.ap)
# ------------------- Derivative routine checks ----------------------------
def test_BWD(self):
#
dwBar = self.CFDSolver.getStatePerturbation(314)
wBar = self.CFDSolver.computeJacobianVectorProductBwd(
resBar=dwBar,
wDeriv=True,
)
wBarfast = self.CFDSolver.computeJacobianVectorProductBwdFast(
resBar=dwBar)
np.testing.assert_allclose(wBar,
wBarfast,
atol=1e-16,
err_msg="w wrt res")
def test_repeated_calls(self):
dwBar = self.CFDSolver.getStatePerturbation(314)
wBarfast1 = self.CFDSolver.computeJacobianVectorProductBwdFast(
resBar=dwBar)
wBarfast2 = self.CFDSolver.computeJacobianVectorProductBwdFast(
resBar=dwBar)
np.testing.assert_allclose(wBarfast1,
wBarfast2,
atol=1e-16,
err_msg="w wrt res double call")
class TestJacVecFwdFD(reg_test_classes.RegTest):
"""
Tests that given a flow state the FWD jacobian vector products are agree with FD.
"""
N_PROCS = 2
def setUp(self):
if not hasattr(self, "name"):
# return immediately when the setup method is being called on the based class and NOT the
# classes created using parametrized
# this will happen when testing, but will hopefully be fixed down the line
return
super().setUp()
options = copy.copy(adflowDefOpts)
options["outputdirectory"] = os.path.join(baseDir,
options["outputdirectory"])
options.update(self.options)
# Create the solver
self.CFDSolver = ADFLOW(options=copy.deepcopy(options), debug=True)
self.ap = copy.deepcopy(self.aero_prob)
# add the default dvs to the problem
for dv in defaultAeroDVs:
self.ap.addDV(dv)
# propagates the values from the restart file throughout the code
self.CFDSolver.getResidual(self.ap)
# ------------------- Derivative routine checks ----------------------------
def test_wDot(self):
# perturb each input and check that the outputs match the FD to with in reason
wDot = self.CFDSolver.getStatePerturbation(321)
resDot, funcsDot, fDot = self.CFDSolver.computeJacobianVectorProductFwd(
wDot=wDot, residualDeriv=True, funcDeriv=True, fDeriv=True)
resDot_FD, funcsDot_FD, fDot_FD = self.CFDSolver.computeJacobianVectorProductFwd(
wDot=wDot,
residualDeriv=True,
funcDeriv=True,
fDeriv=True,
mode="FD",
h=1e-8)
np.testing.assert_allclose(resDot_FD,
resDot,
rtol=8e-4,
err_msg="residual")
for func in funcsDot:
np.testing.assert_allclose(funcsDot_FD[func],
funcsDot[func],
rtol=1e-5,
err_msg=func)
np.testing.assert_allclose(fDot_FD, fDot, rtol=5e-4, err_msg="forces")
def test_xVDot(self):
# perturb each input and check that the outputs match the FD to with in reason
xVDot = self.CFDSolver.getSpatialPerturbation(314)
resDot, funcsDot, fDot = self.CFDSolver.computeJacobianVectorProductFwd(
xVDot=xVDot, residualDeriv=True, funcDeriv=True, fDeriv=True)
resDot_FD, funcsDot_FD, fDot_FD = self.CFDSolver.computeJacobianVectorProductFwd(
xVDot=xVDot,
residualDeriv=True,
funcDeriv=True,
fDeriv=True,
mode="FD",
h=1e-8)
idx_max = np.argmax((resDot_FD - resDot) / resDot)
print(resDot[idx_max], resDot_FD[idx_max])
np.testing.assert_allclose(resDot_FD,
resDot,
atol=5e-4,
err_msg="residual")
for func in funcsDot:
np.testing.assert_allclose(funcsDot_FD[func],
funcsDot[func],
rtol=5e-6,
err_msg=func)
np.testing.assert_allclose(fDot_FD, fDot, rtol=5e-4, err_msg="forces")
def test_xDvDot(self):
# perturb each input and check that the outputs match the FD to with in reason
step_size = {
"alpha": 1e-4,
"beta": 1e-5,
"mach": 1e-5,
"P": 1e-1,
"T": 1e-4,
"xRef": 1e-5,
"yRef": 1e-5,
"zRef": 1e-5,
}
for aeroDV in self.ap.DVs.values():
key = aeroDV.key
xDvDot = {key: 1.0}
resDot, funcsDot, fDot = self.CFDSolver.computeJacobianVectorProductFwd(
xDvDot=xDvDot, residualDeriv=True, funcDeriv=True, fDeriv=True)
resDot_FD, funcsDot_FD, fDot_FD = self.CFDSolver.computeJacobianVectorProductFwd(
xDvDot=xDvDot,
residualDeriv=True,
funcDeriv=True,
fDeriv=True,
mode="FD",
h=step_size[key])
# the tolerances here are loose becuase different ouputs have different optimal steps
np.testing.assert_allclose(resDot_FD,
resDot,
atol=5e-5,
err_msg=f"residual wrt {key}")
for func in funcsDot:
if np.abs(funcsDot[func]) <= 1e-16:
np.testing.assert_allclose(funcsDot_FD[func],
funcsDot[func],
atol=5e-5,
err_msg=f"{func} wrt {key}")
else:
np.testing.assert_allclose(funcsDot_FD[func],
funcsDot[func],
rtol=1e-3,
err_msg=f"{func} wrt {key}")
np.testing.assert_allclose(fDot_FD,
fDot,
atol=5e-7,
err_msg=f"forces wrt {key}")
