def test_setBCData(self): "Test if bcdata is correctly set to data arrays" CFDSolver = ADFLOW(options=self.options) CFDSolver.addFamilyGroup('upstream',['inlet']) CFDSolver.addFamilyGroup('downstream',['outlet']) bc_data = CFDSolver.getBCData(groupNames=self.set_dvs.keys()) for patch in bc_data: if patch.fam in self.set_dvs: for arr in patch: if arr.name in self.set_dvs[patch.fam]: arr.data = self.set_dvs[patch.fam][arr.name] CFDSolver.setBCData(bc_data) # CFDSolver.setAeroProblem(self.ap) new_bc_data = CFDSolver.getBCData(groupNames=self.set_dvs.keys()) ## check that the mesh bc match for patch in new_bc_data: for arr in patch: np.testing.assert_array_equal(arr.data, np.array(self.set_dvs[patch.fam][arr.name]) , err_msg=arr.name)
def test_setBCData_array(self): gridFile = os.path.join(baseDir, '../input_files/conic_conv_nozzle_mb_L4_array.cgns') options = copy.deepcopy(self.options) options['gridfile'] = gridFile CFDSolver = ADFLOW(options=options) CFDSolver.addFamilyGroup('upstream',['inlet']) CFDSolver.addFamilyGroup('downstream',['outlet']) bc_data = CFDSolver.getBCData(groupNames=self.set_dvs.keys()) for patch in bc_data: if patch.fam in self.set_dvs: for arr in patch: if arr.name in self.set_dvs[patch.fam]: arr.data = self.set_dvs[patch.fam][arr.name]*np.ones(arr.size) CFDSolver.setBCData(bc_data) bc_data = CFDSolver.getBCData(groupNames=self.set_dvs.keys()) for patch in bc_data: for arr in patch: np.testing.assert_array_equal(arr.data, np.ones(arr.size)* self.set_dvs[patch.fam][arr.name])