tests/test_1.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_1
  24 """
  25 __revision__ = "V1.9"
  26
  27 #========================================================================
  28 Test_Name = "test_1"
  29 n_iter_test_file = 3
  30 #========================================================================
  31 import os
  32 import sys
  33 import tempfile
  34 import HOMARD
  35 import salome
  36 #
  37 pathHomard=os.getenv('HOMARD_ROOT_DIR')
  38 Rep_Test = os.path.join(pathHomard,"share/salome/resources/homard")
  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 zones
  79   # =====================
  80   # Creation of the box Zone_1
  81     Zone_1 = homard.CreateZoneBox('Zone_1', -0.01, 1.01, -0.01, 0.4, -0.01, 0.6)
  82
  83   # Creation of the sphere Zone_2
  84     Zone_2 = homard.CreateZoneSphere('Zone_2', 0.5, 0.6, 0.7, 0.75)
  85   #
  86   # Creation of the hypotheses
  87   # ==========================
  88   # Creation of the hypothesis a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM
  89     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM = homard.CreateHypothesis('a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM')
  90     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.SetAdapRefinUnRef(1, 1, 0)
  91     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.SetField('RESU____ERRE_ELEM_SIGM__________')
  92     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.SetUseComp(0)
  93     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.AddComp('ERREST')
  94     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.SetRefinThr(3, 10.1)
  95     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.SetTypeFieldInterp(2)
  96     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.AddFieldInterp('RESU____DEPL____________________')
  97     a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM.AddFieldInterp('RESU____ERRE_ELEM_SIGM__________')
  98   # Creation of the hypothesis Zones_1_et_2
  99     Zones_1_et_2 = homard.CreateHypothesis('Zones_1_et_2')
 100     Zones_1_et_2.SetAdapRefinUnRef(0, 1, 0)
 101     Zones_1_et_2.AddZone('Zone_1', 1)
 102     Zones_1_et_2.AddZone('Zone_2', 1)
 103   #
 104   # Creation of the cases
 105   # =====================
 106     # Creation of the case zzzz121b
 107     Case_1 = homard.CreateCase('zzzz121b', 'MAILL', os.path.join(Rep_Test, Test_Name + '.00.med'))
 108     Case_1.SetDirName(Rep_Test_Resu)
 109     Case_1.SetConfType(1)
 110   #
 111   # Creation of the iterations
 112   # ==========================
 113   # Creation of the iteration I1
 114     I1 = Case_1.NextIteration('I1')
 115     I1.SetMeshName('M1')
 116     I1.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.01.med'))
 117     I1.SetFieldFile(os.path.join(Rep_Test, Test_Name + '.00.med'))
 118     I1.SetTimeStepRank(1, 1)
 119     I1.AssociateHypo('a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM')
 120     error = I1.Compute(1, 1)
 121     if error :
 122       error = 1
 123       break
 124
 125   # Creation of the iteration I2
 126     I2 = I1.NextIteration('I2')
 127     I2.SetMeshName('M2')
 128     I2.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.02.med'))
 129     I2.SetFieldFile(os.path.join(Rep_Test, Test_Name + '.01.med'))
 130     I2.SetTimeStepRank(1, 1)
 131     I2.AssociateHypo('a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM')
 132     error = I2.Compute(1, 1)
 133     if error :
 134       error = 2
 135       break
 136
 137   # Creation of the iteration I3
 138     I3 = I2.NextIteration('I3')
 139     I3.SetMeshName('M3')
 140     I3.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.03.med'))
 141     I3.AssociateHypo('Zones_1_et_2')
 142     error = I3.Compute(1, 1)
 143     if error :
 144       error = 3
 145       break
 146   #
 147     break
 148   #
 149     return error
 150
 151 #========================================================================
 152
 153 homard = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 154 assert homard is not None, "Impossible to load homard engine"
 155
 156 #
 157 # Exec of HOMARD-SALOME
 158 #
 159 try :
 160   error_main = homard_exec(salome.myStudy)
 161   if error_main :
 162     raise Exception('Pb in homard_exec at iteration %d' %error_main )
 163 except Exception, e:
 164   raise Exception('Pb in homard_exec: '+e.message)
 165
 166 #
 167 # Test of the result
 168 #
 169 test_file_suff = "apad.%02d.bilan" % n_iter_test_file
 170 rep_test_file = "I%02d" % n_iter_test_file
 171 #
 172 test_file = os.path.join(Rep_Test, Test_Name + "." + test_file_suff)
 173 mess_error_ref = "\nReference file: " + test_file
 174 try :
 175   file = open (test_file, "r")
 176   mess_ref = file.readlines()
 177   file.close()
 178 except :
 179   mess_error = mess_error_ref + "\nThis file does not exist.\n"
 180   raise Exception(mess_error)
 181   sys.exit(2)
 182 #
 183 test_file = os.path.join(Rep_Test_Resu, rep_test_file, test_file_suff)
 184 if os.path.isfile (test_file) :
 185    file = open (test_file, "r")
 186    mess = file.readlines()
 187    file.close()
 188 else :
 189   mess_error  = "\nResult file: " + test_file
 190   mess_error += "\nThis file does not exist.\n"
 191   raise Exception(mess_error)
 192   sys.exit(2)
 193
 194 nblign = len(mess_ref)
 195 if ( len(mess) != nblign ):
 196   mess_error = mess_error_ref +  "\nResult file: " + test_file
 197   mess_error += "\nThe number of lines of the files are not the same.\n"
 198   raise Exception(mess_error)
 199   sys.exit(2)
 200
 201 for num in range(nblign) :
 202    if (( "creation" not in mess_ref[num] ) and ( mess_ref[num] != mess[num])) :
 203        message_erreur = "\nRefe : " + mess_ref[num]
 204        message_erreur += "Test : " + mess[num][:-1]
 205        message_erreur += "\nThe test is different from the reference."
 206        raise Exception(message_erreur)
 207        sys.exit(10)
 208 #
 209 remove_dir(Rep_Test_Resu)
 210 #
 211 if salome.sg.hasDesktop():
 212   salome.sg.updateObjBrowser(1)
 213   iparameters.getSession().restoreVisualState(1)
 214