class TestCmplxStep(reg_test_classes.CmplxRegTest):
"""
Tests that sensitives calculated from solving an adjoint are correct.
and jacobian vector products are accurate.
based on old regression tests_cs 12, and 14
"""
N_PROCS = 2
h = 1e-40
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 training, 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)
self.ffdFile = os.path.join(baseDir, "../../input_files/mdo_tutorial_ffd.fmt")
mesh_options = copy.copy(IDWarpDefOpts)
mesh_options.update({"gridFile": options["gridfile"]})
self.ap = copy.deepcopy(self.aero_prob)
# Setup aeroproblem
self.ap.evalFuncs = self.evalFuncs
# add the default dvs to the problem
for dv in defaultAeroDVs:
self.ap.addDV(dv)
self.CFDSolver = ADFLOW_C(options=options, debug=True)
self.CFDSolver.setMesh(USMesh_C(options=mesh_options))
self.CFDSolver.setDVGeo(setDVGeo(self.ffdFile, cmplx=True))
# propagates the values from the restart file throughout the code
self.CFDSolver.getResidual(self.ap)
def cmplx_test_aero_dvs(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 training, but will hopefully be fixed down the line
return
for dv in ["alpha", "mach"]: # defaultAeroDVs:
funcsSens = defaultdict(lambda: {})
setattr(self.ap, dv, getattr(self.ap, dv) + self.h * 1j)
self.CFDSolver.resetFlow(self.ap)
self.CFDSolver(self.ap, writeSolution=False)
self.assert_solution_failure()
funcs = {}
self.CFDSolver.evalFunctions(self.ap, funcs)
setattr(self.ap, dv, getattr(self.ap, dv) - self.h * 1j)
for f in self.ap.evalFuncs:
key = self.ap.name + "_" + f
dv_key = dv + "_" + self.ap.name
funcsSens[key][dv_key] = numpy.imag(funcs[key]) / self.h
if MPI.COMM_WORLD.rank == 0:
print("====================================")
print(self.ap.alpha)
print(self.ap.mach)
print(self.name, funcsSens)
print("====================================")
self.handler.root_add_dict("Eval Functions Sens:", funcsSens, rtol=1e-8, atol=5e-10)
def cmplx_test_geom_dvs(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 training, but will hopefully be fixed down the line
return
# redo the setup for a cmplx test
funcsSens = defaultdict(lambda: {})
xRef = {"twist": [0.0] * 6, "span": [0.0], "shape": numpy.zeros(72, dtype="D")}
for dv in ["span", "twist", "shape"]:
xRef[dv][0] += self.h * 1j
self.CFDSolver.resetFlow(self.ap)
self.CFDSolver.DVGeo.setDesignVars(xRef)
self.CFDSolver(self.ap, writeSolution=False)
self.assert_solution_failure()
funcs = {}
self.CFDSolver.evalFunctions(self.ap, funcs)
xRef[dv][0] -= self.h * 1j
for f in self.ap.evalFuncs:
key = self.ap.name + "_" + f
dv_key = dv
funcsSens[key][dv_key] = numpy.imag(funcs[key]) / self.h
err_msg = "Failed value for: {}".format(key + " " + dv_key)
ref_val = self.handler.db["Eval Functions Sens:"][key][dv_key]
ref_val = ref_val.flatten()[0]
numpy.testing.assert_allclose(funcsSens[key][dv_key], ref_val, atol=5e-9, rtol=5e-9, err_msg=err_msg)
if MPI.COMM_WORLD.rank == 0:
print("====================================")
print(self.name, funcsSens)
print("====================================")
for i in range(nTwist):
geo.rot_z['wing'].coef[i] = val[i]
def span(val, geo):
# Span
C = geo.extractCoef('wing')
s = geo.extractS('wing')
for i in range(len(C)-1):
C[-1, 2] = C[-1, 2] + val[0]
geo.restoreCoef(C, 'wing')
DVGeo.addGeoDVGlobal('twist', [0]*nTwist, twist, lower=-10, upper=10, scale=1.0)
DVGeo.addGeoDVGlobal('span', [0], span, lower=-10, upper=10, scale=1.0)
DVGeo.addGeoDVLocal('shape', lower=-0.5, upper=0.5, axis='y', scale=10.0)
mesh = MBMesh(options={'gridFile':'../inputFiles/mdo_tutorial_euler.cgns'})
CFDSolver.setMesh(mesh)
CFDSolver.setDVGeo(DVGeo)
#Aeroproblem must be set before we can call DVGeo.setDesignVars
CFDSolver.setAeroProblem(ap)
if not 'complex' in sys.argv:
# Solve system
CFDSolver(ap, writeSolution=False)
funcs = {}
CFDSolver.evalFunctions(ap, funcs)
# Solve sensitivities
funcsSens = {}
CFDSolver.evalFunctionsSens(ap, funcsSens)
# Write values and derivatives out:
if MPI.COMM_WORLD.rank == 0:
for key in ['cl','cmz','drag']: