src/tests/Test/test_2d_1d_mesh.py

   1 #!/usr/bin/env python
   2
   3 ###
   4 ### This file is generated automatically by SALOME v9.11.0 with dump python functionality
   5 ###
   6
   7 import sys
   8 import salome
   9 from os import path
  10 import shutil
  11 from MEDLoader import *
  12
  13 salome.salome_init()
  14 import salome_notebook
  15 notebook = salome_notebook.NoteBook()
  16
  17 import tempfile
  18 tmp_folder = tempfile.mkdtemp()
  19 print("Temporary folder", tmp_folder)
  20
  21 # Create mesh
  22 mesh_name = "mesh"
  23
  24 arr = DataArrayDouble(list(range(10)))
  25 mesh = MEDCouplingUMesh(mesh_name, 1)
  26
  27
  28 coords = [0.,0., 1.,0., 2.,0., 3.,0., 4.,0., 5.,0., 6.,0., 7.,0.]
  29
  30 mesh.allocateCells(7)
  31
  32 for i in range(7):
  33     mesh.insertNextCell(NORM_SEG2, [i,i+1])
  34
  35 mesh.finishInsertingCells()
  36
  37 coordsArr=DataArrayDouble(coords, len(coords)//2, 2)
  38
  39 mesh.setCoords(coordsArr)
  40 mesh.setMeshDimension(2)
  41
  42 mesh_file = path.join(tmp_folder, "mesh_2d_1d.med")
  43 #mesh_file = "mesh_2d_1d.med"
  44
  45 WriteMesh(mesh_file, mesh, True)
  46 print("mesh_file ", mesh_file)
  47
  48 ###
  49 ### HOMARD component
  50 ###
  51
  52 import HOMARD
  53 homard = salome.lcc.FindOrLoadComponent('FactoryServer','HOMARD')
  54 homard.UpdateStudy()
  55 #
  56 # Creation of the hypotheses
  57 # ==========================
  58 # Creation of the hypothesis Hypo_1
  59 Hypo_1 = homard.CreateHypothesis("Hypo_1")
  60 Hypo_1.SetUnifRefinUnRef(1)
  61 #
  62 # Creation of the cases
  63 # =====================
  64 # Creation of the case Case_1
  65 Case_1 = homard.CreateCase("Case_1", mesh_name, mesh_file)
  66 Case_1.SetDirName(tmp_folder)
  67 Case_1.SetConfType(0)
  68 Case_1.SetExtType(0)
  69 #
  70 # Creation of the iterations
  71 # ==========================
  72 # Creation of the iteration Iter_1
  73 Iter_1 = Case_1.NextIteration("Iter_1")
  74 Iter_1.AssociateHypo("Hypo_1")
  75 Iter_1.SetMeshName(mesh_name)
  76 refined_mesh = path.join(tmp_folder, "maill.01.med")
  77 Iter_1.SetMeshFile(refined_mesh)
  78 print("refined_mesh ", refined_mesh)
  79 codret = Iter_1.Compute(1, 1)
  80
  81 ref_mesh = ReadUMeshFromFile(refined_mesh)
  82 coords = ref_mesh.getCoords()
  83 y_coords = coords.getValues()[1::2]
  84
  85 ## Checking that Y values are still zeros
  86 assert all([val<1e-8 for val in y_coords]), y_coords
  87
  88 shutil.rmtree(tmp_folder)