tests/test_2.py

   1 # -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2011-2013  CEA/DEN, EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 """
  21 Python script for HOMARD
  22 Copyright EDF-R&D 2010, 2013
  23 Test test_2
  24 """
  25 __revision__ = "V1.11"
  26
  27 #========================================================================
  28 Test_Name = "test_2"
  29 n_iter_test_file = 3
  30 #========================================================================
  31 import os
  32 import tempfile
  33 import HOMARD
  34 import salome
  35 #
  36 pathHomard = os.getenv('HOMARD_ROOT_DIR')
  37 Rep_Test = os.path.join(pathHomard, "share", "salome", "resources", "homard")
  38 Rep_Test = os.path.normpath(Rep_Test)
  39 Rep_Test_Resu = tempfile.mktemp()
  40 os.mkdir(Rep_Test_Resu)
  41
  42 salome.salome_init()
  43 import iparameters
  44 ipar = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
  45 ipar.append("AP_MODULES_LIST", "Homard")
  46 #========================================================================
  47 #========================================================================
  48 def remove_dir(directory) :
  49   """
  50 Empties, then removes a directory.
  51 Copyright EDF-R&D 2013
  52   """
  53 #
  54   l_aux = os.listdir(directory)
  55   for fic in l_aux :
  56     fic_a = os.path.join(directory, fic)
  57     if os.path.isdir(fic_a) :
  58       remove_dir(fic_a)
  59     else :
  60       os.remove(fic_a)
  61   os.rmdir(directory)
  62 #
  63   return
  64 #
  65 #========================================================================
  66 #========================================================================
  67 def homard_exec(theStudy):
  68   """
  69 Python script for HOMARD
  70 Copyright EDF-R&D 2010, 2013
  71   """
  72   error = 0
  73 #
  74   while not error :
  75   #
  76     homard.SetCurrentStudy(theStudy)
  77   #
  78   # Creation of the boundaries
  79   # ==========================
  80   # Creation of the discrete boundary Boundary_1
  81     Boundary_1 = homard.CreateBoundaryDi('internal_boundary', 'plaque', os.path.join(Rep_Test, Test_Name + '.fr.med'))
  82   #
  83   # Creation of the hypotheses
  84   # ==========================
  85   # Creation of the hypothesis Hypo_2_1
  86     HypoName_1 = "Hypo_2_1"
  87     print "-------- Creation of the hypothesis", HypoName_1
  88     Hypo_2_1 = homard.CreateHypothesis(HypoName_1)
  89     Hypo_2_1.SetAdapRefinUnRef(-1, 1, 0)
  90     Hypo_2_1.AddGroup('EG')
  91     Hypo_2_1.AddGroup('BANDE')
  92     print HypoName_1, " : zones utilisées :", Hypo_2_1.GetZones()
  93     print HypoName_1, " : champ utilisé :", Hypo_2_1.GetFieldName()
  94     print HypoName_1, " : composantes utilisées :", Hypo_2_1.GetComps()
  95     if ( len (Hypo_2_1.GetFieldName()) > 0 ) :
  96       print ".. caractéristiques de l'adaptation :", Hypo_2_1.GetField()
  97
  98   # Creation of the hypothesis Hypo_2_2
  99     HypoName_2 = "Hypo_2_2"
 100     print "-------- Creation of the hypothesis", HypoName_2
 101     Hypo_2_2 = homard.CreateHypothesis(HypoName_2)
 102     Hypo_2_2.SetAdapRefinUnRef(-1, 1, 0)
 103     Hypo_2_2.AddGroup('M_D')
 104     print HypoName_2, " : zones utilisées :", Hypo_2_2.GetZones()
 105     print HypoName_2, " : champ utilisé :", Hypo_2_2.GetFieldName()
 106     print HypoName_2, " : composantes utilisées :", Hypo_2_2.GetComps()
 107     if ( len (Hypo_2_2.GetFieldName()) > 0 ) :
 108       print ".. caractéristiques de l'adaptation :", Hypo_2_2.GetField()
 109   #
 110   # Creation of the cases
 111   # =====================
 112     # Creation of the case Case_2
 113     MeshFile = os.path.join(Rep_Test, Test_Name + '.00.med')
 114     Case_2 = homard.CreateCase('Case_2', 'PLAQUE_0', MeshFile)
 115     Case_2.SetDirName(Rep_Test_Resu)
 116     Case_2.SetConfType(1)
 117     Case_2.AddBoundaryGroup('internal_boundary', '')
 118   #
 119   # Creation of the iterations
 120   # ==========================
 121   # Creation of the iteration Iter_2_1
 122     Iter_2_1 = Case_2.NextIteration('Iter_2_1')
 123     Iter_2_1.SetMeshName('PLAQUE_1')
 124     Iter_2_1.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.01.med'))
 125     Iter_2_1.AssociateHypo(HypoName_1)
 126     error = Iter_2_1.Compute(1, 1)
 127     if error :
 128       error = 1
 129       break
 130
 131   # Creation of the iteration Iter_2_2
 132     Iter_2_2 = Iter_2_1.NextIteration('Iter_2_2')
 133     Iter_2_2.SetMeshName('PLAQUE_2')
 134     Iter_2_2.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.02.med'))
 135     Iter_2_2.AssociateHypo(HypoName_1)
 136     error = Iter_2_2.Compute(1, 1)
 137     if error :
 138       error = 2
 139       break
 140
 141   # Creation of the iteration Iter_2_3
 142     Iter_2_3 = Iter_2_2.NextIteration('Iter_2_3')
 143     Iter_2_3.SetMeshName('PLAQUE_3')
 144     Iter_2_3.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.03.med'))
 145     Iter_2_3.AssociateHypo(HypoName_2)
 146     error = Iter_2_3.Compute(1, 1)
 147     if error :
 148       error = 3
 149       break
 150   #
 151   # Creation of the schema YACS
 152   # ===========================
 153     ScriptFile = os.path.join(pathHomard, "share", "doc", "salome", "gui", "HOMARD", "en", "_downloads", "yacs_script.py")
 154     ScriptFile = os.path.normpath(ScriptFile)
 155     DirName = Rep_Test_Resu
 156     YACS_2 = Case_2.CreateYACSSchema("YACS_2", ScriptFile, DirName, MeshFile)
 157     YACS_2.SetType(1)
 158     filexml = os.path.join(Rep_Test_Resu, 'YACS_2.xml')
 159     error = YACS_2.WriteOnFile(filexml)
 160     if error :
 161       error = 4
 162       break
 163   #
 164     break
 165   #
 166   return error
 167
 168 #========================================================================
 169
 170 homard = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 171 assert homard is not None, "Impossible to load homard engine"
 172 #
 173 # Exec of HOMARD-SALOME
 174 #
 175 try :
 176   error_main = homard_exec(salome.myStudy)
 177   if error_main :
 178     raise Exception('Pb in homard_exec at iteration %d' %error_main )
 179 except Exception, e:
 180   raise Exception('Pb in homard_exec: '+e.message)
 181
 182 #
 183 # Test of the result
 184 #
 185 test_file_suff = "apad.%02d.bilan" % n_iter_test_file
 186 rep_test_file = "I%02d" % n_iter_test_file
 187 #
 188 test_file = os.path.join(Rep_Test, Test_Name + "." + test_file_suff)
 189 mess_error_ref = "\nReference file: " + test_file
 190 try :
 191   file = open (test_file, "r")
 192   mess_ref = file.readlines()
 193   file.close()
 194 except :
 195   mess_error = mess_error_ref + "\nThis file does not exist.\n"
 196   raise Exception(mess_error)
 197 #
 198 test_file = os.path.join(Rep_Test_Resu, rep_test_file, test_file_suff)
 199 if os.path.isfile (test_file) :
 200   file = open (test_file, "r")
 201   mess = file.readlines()
 202   file.close()
 203 else :
 204   mess_error  = "\nResult file: " + test_file
 205   mess_error += "\nThis file does not exist.\n"
 206   raise Exception(mess_error)
 207
 208 nblign = len(mess_ref)
 209 if ( len(mess) != nblign ):
 210   mess_error = mess_error_ref +  "\nResult file: " + test_file
 211   mess_error += "\nThe number of lines of the files are not the same.\n"
 212   raise Exception(mess_error)
 213
 214 for num in range(nblign) :
 215   if (( "creation" not in mess_ref[num] ) and ( mess_ref[num] != mess[num])) :
 216     message_erreur = "\nRefe : " + mess_ref[num]
 217     message_erreur += "Test : " + mess[num][:-1]
 218     message_erreur += "\nThe test is different from the reference."
 219     raise Exception(message_erreur)
 220 #
 221 remove_dir(Rep_Test_Resu)
 222 #
 223 if salome.sg.hasDesktop():
 224   salome.sg.updateObjBrowser(1)
 225   iparameters.getSession().restoreVisualState(1)
 226