From eca76c3781cb161fb216ff5caf6a289ebba7f85e Mon Sep 17 00:00:00 2001
From: Bernd Hahnebach <bernd@bimstatik.org>
Date: Mon, 14 Mar 2016 06:38:13 +0100
Subject: [PATCH] FEM: unit test, add some helper to create result files

---
 src/Mod/Fem/TestFem.py | 81 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)

diff --git a/src/Mod/Fem/TestFem.py b/src/Mod/Fem/TestFem.py
index 3d0368f86b..71cd03c179 100644
--- a/src/Mod/Fem/TestFem.py
+++ b/src/Mod/Fem/TestFem.py
@@ -306,3 +306,84 @@ class FemTest(unittest.TestCase):
     def tearDown(self):
         FreeCAD.closeDocument("FemTest")
         pass
+
+
+# helpers
+def open_cube_test():
+    cube_file = test_file_dir + '/cube.fcstd'
+    FreeCAD.open(cube_file)
+
+
+def create_cube_test_results():
+    import os
+    import shutil
+    cube_file = test_file_dir + '/cube.fcstd'
+
+    FreeCAD.open(cube_file)
+    import FemGui
+    FemGui.setActiveAnalysis(FreeCAD.ActiveDocument.MechanicalAnalysis)
+    import FemTools
+    fea = FemTools.FemTools()
+
+    # static
+    fea.reset_all()
+    fea.run()
+
+    fea.load_results()
+    stat_types = ["U1", "U2", "U3", "Uabs", "Sabs"]
+    stats_static = []  # we only have one result object so we are fine
+    for s in stat_types:
+        stats_static.append("{}: {}\n".format(s, fea.get_stats(s)))
+    static_expected_values_file = temp_dir + '/cube_static_expected_values'
+    f = open(static_expected_values_file, 'w')
+    for s in stats_static:
+        f.write(s)
+    f.close()
+
+    # could be added in FemTools to the self object as an Attribut
+    frd_result_file = os.path.splitext(fea.inp_file_name)[0] + '.frd'
+    dat_result_file = os.path.splitext(fea.inp_file_name)[0] + '.dat'
+
+    frd_static_test_result_file = temp_dir + '/cube_static.frd'
+    dat_static_test_result_file = temp_dir + '/cube_static.dat'
+    shutil.copyfile(frd_result_file, frd_static_test_result_file)
+    shutil.copyfile(dat_result_file, dat_static_test_result_file)
+
+    # frequency
+    fea.reset_all()
+    fea.set_analysis_type('frequency')
+    fea.set_eigenmode_parameters(1)  # we should only have one result object
+    fea.run()
+
+    fea.load_results()
+    stats_frequency = []  # since we set eigenmodeno. we only have one result object so we are fine
+    for s in stat_types:
+        stats_frequency.append("{}: {}\n".format(s, fea.get_stats(s)))
+    frequency_expected_values_file = temp_dir + '/cube_frequency_expected_values'
+    f = open(frequency_expected_values_file, 'w')
+    for s in stats_frequency:
+        f.write(s)
+    f.close()
+
+    frd_frequency_test_result_file = temp_dir + '/cube_frequency.frd'
+    dat_frequency_test_result_file = temp_dir + '/cube_frequency.dat'
+    shutil.copyfile(frd_result_file, frd_frequency_test_result_file)
+    shutil.copyfile(dat_result_file, dat_frequency_test_result_file)
+
+    print('Results copied to: ' + temp_dir)
+
+
+'''
+update the results in FEM untit tests:
+start FreeCAD
+
+import TestFem
+TestFem.create_cube_test_results()
+
+copy result files from /tmp into the src dirctory
+run make
+start FreeCAD and run FEM unit test
+if FEM unit test is fine --> commit new FEM unit test results
+
+TODO compare the inp file of the helper with the inp file of FEM unit tests
+'''