FEM: code formating, max line length < 100, fem test modules
This commit is contained in:
Bernd Hahnebach
@@ -36,7 +36,10 @@ from os.path import join
|
||||
class TestSolverFrameWork(unittest.TestCase):
|
||||
fcc_print('import TestSolverFrameWork')
|
||||
|
||||
def setUp(self):
|
||||
# ********************************************************************************************
|
||||
def setUp(
|
||||
self
|
||||
):
|
||||
# init, is executed before every test
|
||||
self.doc_name = "TestSolverFrameWork"
|
||||
try:
|
||||
@@ -50,15 +53,24 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
self.temp_dir = testtools.get_fem_test_tmp_dir()
|
||||
self.test_file_dir = join(testtools.get_fem_test_home_dir(), 'ccx')
|
||||
|
||||
def test_solver_framework(self):
|
||||
# ********************************************************************************************
|
||||
def test_solver_framework(
|
||||
self
|
||||
):
|
||||
fcc_print('\n--------------- Start of FEM tests solver frame work ---------------')
|
||||
box = self.active_doc.addObject("Part::Box", "Box")
|
||||
fcc_print('Checking FEM new analysis...')
|
||||
analysis = ObjectsFem.makeAnalysis(self.active_doc, 'Analysis')
|
||||
analysis = ObjectsFem.makeAnalysis(
|
||||
self.active_doc,
|
||||
'Analysis'
|
||||
)
|
||||
self.assertTrue(analysis, "FemTest of new analysis failed")
|
||||
|
||||
fcc_print('Checking FEM new material...')
|
||||
material_object = ObjectsFem.makeMaterialSolid(self.active_doc, 'MechanicalMaterial')
|
||||
material_object = ObjectsFem.makeMaterialSolid(
|
||||
self.active_doc,
|
||||
'MechanicalMaterial'
|
||||
)
|
||||
mat = material_object.Material
|
||||
mat['Name'] = "Steel-Generic"
|
||||
mat['YoungsModulus'] = "200000 MPa"
|
||||
@@ -69,13 +81,19 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
analysis.addObject(material_object)
|
||||
|
||||
fcc_print('Checking FEM new fixed constraint...')
|
||||
fixed_constraint = self.active_doc.addObject("Fem::ConstraintFixed", "FemConstraintFixed")
|
||||
fixed_constraint = self.active_doc.addObject(
|
||||
"Fem::ConstraintFixed",
|
||||
"FemConstraintFixed"
|
||||
)
|
||||
fixed_constraint.References = [(box, "Face1")]
|
||||
self.assertTrue(fixed_constraint, "FemTest of new fixed constraint failed")
|
||||
analysis.addObject(fixed_constraint)
|
||||
|
||||
fcc_print('Checking FEM new force constraint...')
|
||||
force_constraint = self.active_doc.addObject("Fem::ConstraintForce", "FemConstraintForce")
|
||||
force_constraint = self.active_doc.addObject(
|
||||
"Fem::ConstraintForce",
|
||||
"FemConstraintForce"
|
||||
)
|
||||
force_constraint.References = [(box, "Face6")]
|
||||
force_constraint.Force = 40000.0
|
||||
force_constraint.Direction = (box, ["Edge5"])
|
||||
@@ -86,7 +104,10 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
analysis.addObject(force_constraint)
|
||||
|
||||
fcc_print('Checking FEM new pressure constraint...')
|
||||
pressure_constraint = self.active_doc.addObject("Fem::ConstraintPressure", "FemConstraintPressure")
|
||||
pressure_constraint = self.active_doc.addObject(
|
||||
"Fem::ConstraintPressure",
|
||||
"FemConstraintPressure"
|
||||
)
|
||||
pressure_constraint.References = [(box, "Face2")]
|
||||
pressure_constraint.Pressure = 1000.0
|
||||
pressure_constraint.Reversed = False
|
||||
@@ -101,7 +122,10 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
self.assertTrue(ret, "Import of mesh nodes failed")
|
||||
ret = create_elements_cube(mesh)
|
||||
self.assertTrue(ret, "Import of mesh volumes failed")
|
||||
mesh_object = self.active_doc.addObject('Fem::FemMeshObject', self.mesh_name)
|
||||
mesh_object = self.active_doc.addObject(
|
||||
'Fem::FemMeshObject',
|
||||
self.mesh_name
|
||||
)
|
||||
mesh_object.FemMesh = mesh
|
||||
self.assertTrue(mesh, "FemTest of new mesh failed")
|
||||
analysis.addObject(mesh_object)
|
||||
@@ -111,7 +135,10 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
# solver frame work ccx solver
|
||||
# calculix solver object
|
||||
fcc_print('\nChecking FEM CalculiX solver for solver frame work...')
|
||||
solver_ccx_object = ObjectsFem.makeSolverCalculix(self.active_doc, 'SolverCalculiX')
|
||||
solver_ccx_object = ObjectsFem.makeSolverCalculix(
|
||||
self.active_doc,
|
||||
'SolverCalculiX'
|
||||
)
|
||||
solver_ccx_object.AnalysisType = 'static'
|
||||
solver_ccx_object.GeometricalNonlinearity = 'linear'
|
||||
solver_ccx_object.ThermoMechSteadyState = False
|
||||
@@ -124,17 +151,27 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
analysis.addObject(solver_ccx_object)
|
||||
|
||||
static_base_name = 'cube_static'
|
||||
solverframework_analysis_dir = testtools.get_unit_test_tmp_dir(testtools.get_fem_test_tmp_dir(), 'FEM_solverframework')
|
||||
solverframework_analysis_dir = testtools.get_unit_test_tmp_dir(
|
||||
testtools.get_fem_test_tmp_dir(),
|
||||
'FEM_solverframework'
|
||||
)
|
||||
|
||||
# write input file
|
||||
fcc_print('Checking FEM ccx solver for solver frame work......')
|
||||
fcc_print('machine_ccx')
|
||||
machine_ccx = solver_ccx_object.Proxy.createMachine(solver_ccx_object, solverframework_analysis_dir)
|
||||
machine_ccx = solver_ccx_object.Proxy.createMachine(
|
||||
solver_ccx_object,
|
||||
solverframework_analysis_dir
|
||||
)
|
||||
machine_ccx.target = femsolver.run.PREPARE
|
||||
machine_ccx.start()
|
||||
machine_ccx.join() # wait for the machine to finish.
|
||||
|
||||
infile_given = join(testtools.get_fem_test_home_dir(), 'ccx', (static_base_name + '.inp'))
|
||||
infile_given = join(
|
||||
testtools.get_fem_test_home_dir(),
|
||||
'ccx',
|
||||
(static_base_name + '.inp')
|
||||
)
|
||||
inpfile_totest = join(solverframework_analysis_dir, (self.mesh_name + '.inp'))
|
||||
fcc_print('Comparing {} to {}'.format(infile_given, inpfile_totest))
|
||||
ret = testtools.compare_inp_files(infile_given, inpfile_totest)
|
||||
@@ -142,20 +179,30 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
|
||||
# use solver frame work elmer solver
|
||||
# elmer solver object
|
||||
solver_elmer_object = ObjectsFem.makeSolverElmer(self.active_doc, 'SolverElmer')
|
||||
solver_elmer_object = ObjectsFem.makeSolverElmer(
|
||||
self.active_doc,
|
||||
'SolverElmer'
|
||||
)
|
||||
self.assertTrue(solver_elmer_object, "FemTest of elmer solver failed")
|
||||
analysis.addObject(solver_elmer_object)
|
||||
solver_elmer_eqobj = ObjectsFem.makeEquationElasticity(self.active_doc, solver_elmer_object)
|
||||
solver_elmer_eqobj = ObjectsFem.makeEquationElasticity(
|
||||
self.active_doc,
|
||||
solver_elmer_object
|
||||
)
|
||||
self.assertTrue(solver_elmer_eqobj, "FemTest of elmer elasticity equation failed")
|
||||
|
||||
# set ThermalExpansionCoefficient, current elmer seems to need it even on simple elasticity analysis
|
||||
# set ThermalExpansionCoefficient
|
||||
# current elmer seems to need it even on simple elasticity analysis
|
||||
mat = material_object.Material
|
||||
mat['ThermalExpansionCoefficient'] = "0 um/m/K" # FIXME elmer elasticity needs the dictionary key, otherwise it fails
|
||||
# FIXME elmer elasticity needs the dictionary key, otherwise it fails
|
||||
mat['ThermalExpansionCoefficient'] = "0 um/m/K"
|
||||
material_object.Material = mat
|
||||
|
||||
mesh_gmsh = ObjectsFem.makeMeshGmsh(self.active_doc)
|
||||
mesh_gmsh.CharacteristicLengthMin = "9 mm"
|
||||
mesh_gmsh.FemMesh = mesh_object.FemMesh # elmer needs a GMHS mesh object, FIXME error message on Python solver run
|
||||
# elmer needs a GMHS mesh object
|
||||
# FIXME error message on Python solver run
|
||||
mesh_gmsh.FemMesh = mesh_object.FemMesh
|
||||
mesh_gmsh.Part = box
|
||||
analysis.addObject(mesh_gmsh)
|
||||
self.active_doc.removeObject(mesh_object.Name)
|
||||
@@ -163,7 +210,11 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
# solver frame work Elmer solver
|
||||
# write input files
|
||||
fcc_print('\nChecking FEM Elmer solver for solver frame work...')
|
||||
machine_elmer = solver_elmer_object.Proxy.createMachine(solver_elmer_object, solverframework_analysis_dir, True)
|
||||
machine_elmer = solver_elmer_object.Proxy.createMachine(
|
||||
solver_elmer_object,
|
||||
solverframework_analysis_dir,
|
||||
True
|
||||
)
|
||||
machine_elmer.target = femsolver.run.PREPARE
|
||||
machine_elmer.start()
|
||||
machine_elmer.join() # wait for the machine to finish.
|
||||
@@ -197,7 +248,10 @@ class TestSolverFrameWork(unittest.TestCase):
|
||||
self.active_doc.saveAs(save_fc_file)
|
||||
fcc_print('--------------- End of FEM tests solver frame work ---------------')
|
||||
|
||||
def tearDown(self):
|
||||
# ********************************************************************************************
|
||||
def tearDown(
|
||||
self
|
||||
):
|
||||
# clearance, is executed after every test
|
||||
FreeCAD.closeDocument(self.doc_name)
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user