# *************************************************************************** # * Copyright (c) 2018 Bernd Hahnebach * # * * # * This file is part of the FreeCAD CAx development system. * # * * # * This program is free software; you can redistribute it and/or modify * # * it under the terms of the GNU Lesser General Public License (LGPL) * # * as published by the Free Software Foundation; either version 2 of * # * the License, or (at your option) any later version. * # * for detail see the LICENCE text file. * # * * # * This program is distributed in the hope that it will be useful, * # * but WITHOUT ANY WARRANTY; without even the implied warranty of * # * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * # * GNU Library General Public License for more details. * # * * # * You should have received a copy of the GNU Library General Public * # * License along with this program; if not, write to the Free Software * # * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * # * USA * # * * # *************************************************************************** __title__ = "Solver elmer FEM unit tests" __author__ = "Bernd Hahnebach" __url__ = "https://www.freecad.org" import unittest from os.path import join import FreeCAD import femsolver.run from . import support_utils as testtools from .support_utils import fcc_print from .support_utils import get_namefromdef class TestSolverElmer(unittest.TestCase): fcc_print("import TestSolverElmer") # ******************************************************************************************** def setUp(self): # setUp is executed before every test # new document self.document = FreeCAD.newDocument(self.__class__.__name__) # more inits self.pre_dir_name = "solver_elmer_" self.ending = ".sif" self.infilename = "case" self.test_file_dir = join(testtools.get_fem_test_home_dir(), "elmer") # set Units # since in Elmer writer the FreeCAD pref is used, here we need to set the FreeCAD pref # the use of FreeCAD.Units.setScheme would not take affect because the pref is not changed # see https://forum.freecad.org/viewtopic.php?t=48451 param = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units") self.saved_unit_schema = param.GetInt("UserSchema") # ******************************************************************************************** def tearDown(self): # set back unit unit schema param = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units") unit_schema = param.GetInt("UserSchema") if unit_schema != self.saved_unit_schema: fcc_print(f"Reset unit schema back to {self.saved_unit_schema}") param = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units") param.SetInt("UserSchema", self.saved_unit_schema) # tearDown is executed after every test FreeCAD.closeDocument(self.document.Name) fcc_print("") # ******************************************************************************************** def test_00print(self): # since method name starts with 00 this will be run first # this test just prints a line with stars fcc_print( "\n{0}\n{1} run FEM TestSolverElmer tests {2}\n{0}".format( 100 * "*", 10 * "*", 55 * "*" ) ) # ******************************************************************************************** def set_unit_schema(self, new_unit_schema=0): fcc_print( "\nSaved unit schema: {}. Set unit schema to {}.".format( self.saved_unit_schema, new_unit_schema ) ) param = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units") param.SetInt("UserSchema", new_unit_schema) # ******************************************************************************************** def test_box_static_0_mm(self): fcc_print("") self.set_unit_schema(0) # mm/kg/s # set up the Elmer static analysis example from femexamples.boxanalysis_static import setup setup(self.document, "elmer") # for information: # elmer needs gmsh mesho object # FIXME error message on Python solver run # the examples do use a gmsh mesh object thus ok base_name = get_namefromdef("test_") analysis_dir = testtools.get_fem_test_tmp_dir(self.pre_dir_name + base_name) # save the file save_fc_file = join(analysis_dir, base_name + ".FCStd") # fcc_print("Save FreeCAD file to {}...".format(save_fc_file)) self.document.saveAs(save_fc_file) # write input files # fcc_print("Checking FEM input file writing for Elmer solver framework solver ...") machine_elmer = self.document.SolverElmer.Proxy.createMachine( self.document.SolverElmer, analysis_dir, True ) machine_elmer.target = femsolver.run.PREPARE machine_elmer.start() machine_elmer.join() # wait for the machine to finish. fcc_print("Test writing STARTINFO file") startinfo_given = join(self.test_file_dir, "ELMERSOLVER_STARTINFO") startinfo_totest = join(analysis_dir, "ELMERSOLVER_STARTINFO") # fcc_print("Comparing {} to {}".format(startinfo_given, startinfo_totest)) ret = testtools.compare_files(startinfo_given, startinfo_totest) self.assertFalse(ret, f"STARTINFO write file test failed.\n{ret}") fcc_print("Test writing case file") casefile_given = join(self.test_file_dir, base_name + self.ending) casefile_totest = join(analysis_dir, self.infilename + self.ending) # fcc_print("Comparing {} to {}".format(casefile_given, casefile_totest)) ret = testtools.compare_files(casefile_given, casefile_totest) self.assertFalse(ret, f"case write file test failed.\n{ret}") fcc_print("Test writing GMSH geo file") gmshgeofile_given = join(self.test_file_dir, "group_mesh.geo") gmshgeofile_totest = join(analysis_dir, "group_mesh.geo") # fcc_print("Comparing {} to {}".format(gmshgeofile_given, gmshgeofile_totest)) ret = testtools.compare_files(gmshgeofile_given, gmshgeofile_totest) self.assertFalse(ret, f"GMSH geo write file test failed.\n{ret}") # ******************************************************************************************** def test_ccxcantilever_faceload_0_mm(self): fcc_print("") self.set_unit_schema(0) # mm/kg/s from femexamples.ccx_cantilever_faceload import setup setup(self.document, "elmer") self.input_file_writing_test(get_namefromdef("test_")) # ******************************************************************************************** def test_ccxcantilever_faceload_1_si(self): fcc_print("") self.set_unit_schema(1) # SI-units m/kg/s from femexamples.ccx_cantilever_faceload import setup setup(self.document, "elmer") self.input_file_writing_test(get_namefromdef("test_")) # ******************************************************************************************** def test_ccxcantilever_nodeload_0_mm(self): fcc_print("") self.set_unit_schema(0) # mm/kg/s from femexamples.ccx_cantilever_nodeload import setup setup(self.document, "elmer") self.input_file_writing_test(get_namefromdef("test_")) # ******************************************************************************************** def test_ccxcantilever_prescribeddisplacement_0_mm(self): fcc_print("") self.set_unit_schema(0) # mm/kg/s from femexamples.ccx_cantilever_prescribeddisplacement import setup setup(self.document, "elmer") self.input_file_writing_test(get_namefromdef("test_")) # ******************************************************************************************** def input_file_writing_test(self, base_name): self.document.recompute() # get analysis working directory and save FreeCAD file working_dir = testtools.get_fem_test_tmp_dir(self.pre_dir_name + base_name) save_fc_file = join(working_dir, base_name + ".FCStd") # fcc_print("Save FreeCAD file to {} ...".format(save_fc_file)) self.document.saveAs(save_fc_file) # write input file machine = self.document.SolverElmer.Proxy.createMachine( self.document.SolverElmer, working_dir, True # set testmode to True ) machine.target = femsolver.run.PREPARE machine.start() machine.join() # wait for the machine to finish # compare input file with the given one inpfile_given = join(self.test_file_dir, base_name + self.ending) inpfile_totest = join(working_dir, self.infilename + self.ending) # fcc_print("Comparing {} to {}".format(inpfile_given, inpfile_totest)) ret = testtools.compare_inp_files(inpfile_given, inpfile_totest) self.assertFalse(ret, f"Elmer write_inp_file for {base_name} test failed.\n{ret}")