src/Plugins/MEDReader/Test/testMEDReader17.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2019  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, or (at your option) any later version.
   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 # Author : Anthony Geay (EDF R&D)
  21
  22 from MEDLoader import *
  23
  24 """ This test is a non regression test of EDF9622."""
  25
  26 fname="testMEDReader17.med"
  27
  28 arr=DataArrayDouble([0,1,2])
  29 m=MEDCouplingCMesh() ; m.setCoords(arr,arr) ; m=m.buildUnstructured() ; m.setName("Mesh")
  30 mm=MEDFileUMesh() ; mm.setMeshAtLevel(0,m)
  31 grp0=DataArrayInt([0,1]) ; grp0.setName("grp0")
  32 grp1=DataArrayInt([2,3]) ; grp1.setName("grp1")
  33 grp2=DataArrayInt([1,2]) ; grp2.setName("grp2")
  34 grp3=DataArrayInt([0,1,2,3]) ; grp3.setName("grp3")
  35 mm.setGroupsAtLevel(0,[grp0,grp1,grp2,grp3])
  36 f=MEDCouplingFieldDouble(ON_GAUSS_NE) ; f.setMesh(m) ; f.setName("MyField") ; f.setTime(0.,0,0)
  37 arr2=DataArrayDouble(4*4*2) ; arr2.iota() ; arr2.rearrange(2) ; arr2.setInfoOnComponents(["aa","bbb"])
  38 f.setArray(arr2) ; arr2+=0.1 ; f.checkConsistencyLight()
  39 mm.write(fname,2)
  40 WriteFieldUsingAlreadyWrittenMesh(fname,f)
  41 #
  42 from paraview.simple import *
  43 from paraview import servermanager
  44
  45 paraview.simple._DisableFirstRenderCameraReset()
  46 reader=MEDReader(FileName=fname)
  47 ExpectedEntries=['TS0/Mesh/ComSup0/MyField@@][@@GSSNE','TS1/Mesh/ComSup0/Mesh@@][@@P0']
  48 assert(reader.GetProperty("FieldsTreeInfo")[::2]==ExpectedEntries)
  49 reader.AllArrays=['TS0/Mesh/ComSup0/MyField@@][@@GSSNE']
  50 ExtractGroup1 = ExtractGroup(Input=reader)
  51 #ExtractGroup1.UpdatePipelineInformation()
  52 ExtractGroup1.AllGroups=["GRP_grp1"]
  53 ELNOfieldToPointGaussian1=ELNOfieldToPointGaussian(Input=ExtractGroup1)
  54 ELNOfieldToPointGaussian1.SelectSourceArray=['ELNO@MyField']
  55 #
  56 ELNOfieldToPointGaussian1=ELNOfieldToPointGaussian(Input=reader)
  57 ELNOfieldToPointGaussian1.SelectSourceArray=['ELNO@MyField']
  58 ExtractGroup1 = ExtractGroup(Input=ELNOfieldToPointGaussian1)
  59 #ExtractGroup1.UpdatePipelineInformation()
  60 ExtractGroup1.AllGroups=["GRP_grp1"]
  61 #ExtractGroup1.UpdatePipeline()
  62 res=servermanager.Fetch(ExtractGroup1,0)
  63 assert(res.GetBlock(0).GetNumberOfCells()==8)
  64 vtkArrToTest=res.GetBlock(0).GetPointData().GetArray("MyField")
  65 assert(vtkArrToTest.GetNumberOfComponents()==2)
  66 assert(vtkArrToTest.GetNumberOfTuples()==8)
  67 vals=[vtkArrToTest.GetValue(i) for i in range(16)]
  68 assert(DataArrayDouble([(16.1,17.1),(18.1,19.1),(20.1,21.1),(22.1,23.1),(24.1,25.1),(26.1,27.1),(28.1,29.1),(30.1,31.1)]).isEqual(DataArrayDouble(vals,8,2),1e-12))