1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2007-2022 CEA/DEN, EDF R&D
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.
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.
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
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 from medcoupling import *
24 from math import pi,e,sqrt,cos,sin
25 from datetime import datetime
26 from MEDCouplingDataForTest import MEDCouplingDataForTest
27 import rlcompleter,readline # this line has to be here, to ensure a usability of MEDCoupling/MEDLoader. B4 removing it please notify to anthony.geay@edf.fr
29 class MEDCouplingBasicsTest3(unittest.TestCase):
30 def testSwigGetItem1(self):
36 da.setInfoOnComponent(0,"X [m]")
37 da.setInfoOnComponent(1,"Y [m]")
38 da.setInfoOnComponent(2,"Z [km]")
40 self.assertEqual([22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51],da2.getValues())
42 self.assertEqual([19, 20, 21],da2.getValues())
45 except InterpKernelException as e:
48 self.assertTrue(False)
51 self.assertEqual([24, 27, 30, 33, 36, 39, 42, 45, 48],da2.getValues())
53 self.assertEqual([22, 23, 24, 25, 26, 27, 28, 29, 30],da2.getValues())
55 self.assertTrue(da2.isEqual(da))
57 self.assertTrue(da2.isEqual(da))
60 except InterpKernelException as e:
63 self.assertTrue(False)
65 self.assertTrue(da[5:8,-2].isEqualWithoutConsideringStr(DataArrayInt([23,26,29])))
67 self.assertEqual([22, 25, 28],da2.getValues())
70 except InterpKernelException as e:
73 self.assertTrue(False)
76 self.assertEqual([],da2.getValues())
79 def testSwigGetItem2(self):
80 da=DataArrayDouble.New()
85 da.setInfoOnComponent(0,"X [m]")
86 da.setInfoOnComponent(1,"Y [m]")
87 da.setInfoOnComponent(2,"Z [km]")
89 self.assertEqual([22., 23., 24., 25., 26., 27., 28., 29., 30., 31., 32., 33., 34., 35., 36., 37., 38., 39., 40., 41., 42., 43., 44., 45., 46., 47., 48., 49., 50., 51.],da2.getValues())
91 self.assertEqual([19., 20., 21],da2.getValues())
94 except InterpKernelException as e:
97 self.assertTrue(False)
100 self.assertEqual([24., 27., 30., 33., 36., 39., 42., 45., 48.],da2.getValues())
102 self.assertEqual([22., 23., 24., 25., 26., 27., 28., 29., 30.],da2.getValues())
104 self.assertTrue(da2.isEqual(da,1e-12))
106 self.assertTrue(da2.isEqual(da,1e-12))
109 except InterpKernelException as e:
110 self.assertTrue(True)
112 self.assertTrue(False)
114 self.assertTrue(da[5:8,-2].isEqualWithoutConsideringStr(DataArrayDouble([23.,26.,29.]),1e-12))
116 self.assertEqual([22., 25., 28.],da2.getValues())
119 except InterpKernelException as e:
120 self.assertTrue(True)
122 self.assertTrue(False)
125 self.assertEqual([],da2.getValues())
128 def testSwigSetItem1(self):
129 da=DataArrayInt.New()
133 da.setInfoOnComponent(0,"X [m]") ; da.setInfoOnComponent(1,"Y [km]") ; da.setInfoOnComponent(2,"Y [m]")
134 da.setInfoOnComponent(3,"Z [W]") ; da.setInfoOnComponent(4,"ZZ [km]") ;
136 self.assertEqual([7, 8, 3, 10, 11, 12, 13, 3, 15, 16, 17, 18, 3, 20, 21, 22, 23, 3, 25, 26],da.getValues())
137 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
139 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 3, 3, 3, 3, 3, 22, 23, 24, 25, 26],da.getValues())
140 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
142 self.assertEqual([-1, -1, -1, -1, -1, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, -1, -1, -1, -1, -1],da.getValues())
143 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
145 self.assertEqual([7, -3, 9, -3, -3, 12, -3, 14, -3, -3, 17, -3, 19, -3, -3, 22, -3, 24, -3, -3],da.getValues())
146 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
147 da2=DataArrayInt.New() ; da2.setValues([0,2,3],3,1)
149 self.assertEqual([-7, -7, -7, -7, -7, 12, 13, 14, 15, 16, -7, -7, -7, -7, -7, -7, -7, -7, -7, -7],da.getValues())
150 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
152 self.assertEqual([7, 8, 9, -7, -7, 12, 13, 14, 15, 16, 17, 18, 19, -7, -7, 22, 23, 24, -7, -7],da.getValues())
153 # Let's test with DAI right hand side
154 da1=DataArrayInt.New()
155 da1.setValues([25,26,27,125,126,127],2,3)
157 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
159 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 25, 26, 27, 21, 22, 125, 126, 127, 26],da.getValues())
160 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
161 da[1:,3]=[225,226,227]
162 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 225, 16, 17, 18, 19, 226, 21, 22, 23, 24, 227, 26],da.getValues())
163 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
164 da[1,2:]=[225,226,227]
165 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 225, 226, 227, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26],da.getValues())
166 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
167 da[da2,-2:]=[88,99,1010,1111,1212,1313]
168 self.assertEqual([7, 8, 9, 88, 99, 12, 13, 14, 15, 16, 17, 18, 19, 1010, 1111, 22, 23, 24, 1212, 1313],da.getValues())
169 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
170 da3=DataArrayInt.New(); da3.setValues([88,99,1010,1111,1212,1313],3,2)
172 self.assertEqual([7, 8, 9, 88, 99, 12, 13, 14, 15, 16, 17, 18, 19, 1010, 1111, 22, 23, 24, 1212, 1313],da.getValues())
173 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
175 self.assertEqual([88, 8, 99, 10, 11, 12, 13, 14, 15, 16, 1010, 18, 1111, 20, 21, 1212, 23, 1313, 25, 26],da.getValues())
176 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
178 self.assertEqual([88, 8, 99, 10, 11, 12, 13, 14, 15, 16, 1010, 18, 1111, 20, 21, 1212, 23, 1313, 25, 26],da.getValues())
179 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
181 self.assertEqual([-8, 8, -8, 10, 11, 12, 13, 14, 15, 16, -8, 18, -8, 20, 21, -8, 23, -8, 25, 26],da.getValues())
184 def testSwigSetItem2(self):
185 da=DataArrayDouble.New()
189 da.setInfoOnComponent(0,"X [m]") ; da.setInfoOnComponent(1,"Y [km]") ; da.setInfoOnComponent(2,"Y [m]")
190 da.setInfoOnComponent(3,"Z [W]") ; da.setInfoOnComponent(4,"ZZ [km]") ;
192 self.assertEqual([7., 8., 3., 10., 11., 12., 13., 3., 15., 16., 17., 18., 3., 20., 21., 22., 23., 3., 25., 26.],da.getValues())
193 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
195 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 14., 15., 16., 3., 3., 3., 3., 3., 22., 23., 24., 25., 26.],da.getValues())
196 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
198 self.assertEqual([-1., -1., -1., -1., -1., 12., 13., 14., 15., 16., 17., 18., 19., 20., 21., -1., -1., -1., -1., -1.],da.getValues())
199 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
201 self.assertEqual([7., -3., 9., -3., -3., 12., -3., 14., -3., -3., 17., -3., 19., -3., -3., 22., -3., 24., -3., -3.],da.getValues())
202 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
203 da2=DataArrayInt.New() ; da2.setValues([0,2,3],3,1)
205 self.assertEqual([-7., -7., -7., -7., -7., 12., 13., 14., 15., 16., -7., -7., -7., -7., -7., -7., -7., -7., -7., -7.],da.getValues())
206 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
208 self.assertEqual([7., 8., 9., -7., -7., 12., 13., 14., 15., 16., 17., 18., 19., -7., -7., 22., 23., 24., -7., -7.],da.getValues())
209 # Let's test with DAI right hand side
210 da1=DataArrayDouble.New()
211 da1.setValues([25,26,27,125,126,127],2,3)
213 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
215 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 14., 15., 16., 17., 25., 26., 27., 21., 22., 125., 126., 127., 26.],da.getValues())
216 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
217 da[1:,3]=[225.,226.,227.]
218 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 14., 225., 16., 17., 18., 19., 226., 21., 22., 23., 24., 227., 26.],da.getValues())
219 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
220 da[1,2:]=[225,226,227]
221 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 225., 226., 227., 17., 18., 19., 20., 21., 22., 23., 24., 25., 26.],da.getValues())
222 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
223 da[da2,-2:]=[88,99,1010,1111,1212,1313]
224 self.assertEqual([7., 8., 9., 88., 99., 12., 13., 14., 15., 16., 17., 18., 19., 1010., 1111., 22., 23., 24., 1212., 1313.],da.getValues())
225 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
226 da3=DataArrayDouble.New(); da3.setValues([88,99,1010,1111,1212,1313],3,2)
228 self.assertEqual([7., 8., 9., 88., 99., 12., 13., 14., 15., 16., 17., 18., 19., 1010., 1111., 22., 23., 24., 1212., 1313.],da.getValues())
229 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
231 self.assertEqual([88., 8., 99., 10., 11., 12., 13., 14., 15., 16., 1010., 18., 1111., 20., 21., 1212., 23., 1313., 25., 26.],da.getValues())
232 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
234 self.assertEqual([88., 8., 99., 10., 11., 12., 13., 14., 15., 16., 1010., 18., 1111., 20., 21., 1212., 23., 1313., 25., 26.],da.getValues())
235 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
237 self.assertEqual([-8., 8., -8., 10., 11., 12., 13., 14., 15., 16., -8., 18., -8., 20., 21., -8., 23., -8., 25., 26.],da.getValues())
240 def testSwigDADOp(self):
241 da=DataArrayDouble.New()
244 da1=DataArrayDouble.New()
248 self.assertEqual([15., 17., 19., 21., 23., 25., 27., 29., 31., 33., 35., 37.],da2.getValues())
251 self.assertTrue(da2.isEqual(da3,1e-12))
253 self.assertEqual([6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0],da2.getValues())
255 self.assertEqual([-6.0, -7.0, -8.0, -9.0, -10.0, -11.0, -12.0, -13.0, -14.0, -15.0, -16.0, -17.0],da2.getValues())
257 self.assertEqual([21.0, 24.0, 27.0, 30.0, 33.0, 36.0, 39.0, 42.0, 45.0, 48.0, 51.0, 54.0],da2.getValues())
259 self.assertEqual([21.0, 24.0, 27.0, 30.0, 33.0, 36.0, 39.0, 42.0, 45.0, 48.0, 51.0, 54.0],da2.getValues())
261 self.assertEqual([56.0, 72.0, 90.0, 110.0, 132.0, 156.0, 182.0, 210.0, 240.0, 272.0, 306.0, 342.0],da2.getValues())
263 self.assertEqual([1.75, 2.0, 2.25, 2.5, 2.75, 3.0, 3.25, 3.5, 3.75, 4.0, 4.25, 4.5],da2.getValues())
266 self.assertTrue(da4.isUniform(1.,1e-12))
267 st1=da.getHiddenCppPointer()
269 st2=da.getHiddenCppPointer()
270 self.assertEqual(st1,st2)
271 self.assertTrue(da.isEqual(da1,1e-12))
273 st2=da.getHiddenCppPointer()
274 self.assertEqual(st1,st2)
275 self.assertEqual(list(range(12)), da.getValues())
277 st2=da.getHiddenCppPointer()
278 self.assertEqual(st1,st2)
279 self.assertEqual([8.0, 10.0, 12.0, 14.0, 16.0, 18.0, 20.0, 22.0, 24.0, 26.0, 28.0, 30.0],da.getValues())
281 st2=da.getHiddenCppPointer()
282 self.assertEqual(st1,st2)
283 self.assertEqual([4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0],da.getValues())
285 st2=da.getHiddenCppPointer()
286 self.assertEqual(st1,st2)
287 self.assertEqual([32.0, 45.0, 60.0, 77.0, 96.0, 117.0, 140.0, 165.0, 192.0, 221.0, 252.0, 285.0],da.getValues())
289 self.assertEqual(st1,st2)
290 self.assertEqual([4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0],da.getValues())
292 st2=da.getHiddenCppPointer()
293 self.assertEqual(st1,st2)
294 self.assertEqual([2.0, 2.5, 3.0, 3.5, 4.0, 4.5, 5.0, 5.5, 6.0, 6.5, 7.0, 7.5],da.getValues())
296 da5=DataArrayDouble.New()
297 da5.setValues([5.,4.,3.,2.],4,1)
298 da*=da5 # it works with unmathing number of compo
299 st2=da.getHiddenCppPointer()
300 self.assertEqual(st1,st2)
301 self.assertEqual([10.0, 12.5, 15.0, 14.0, 16.0, 18.0, 15.0, 16.5, 18.0, 13.0, 14.0, 15.0],da.getValues())
306 ids=DataArrayInt.New()
307 ids.setValues([3,4,7],3,1)
309 self.assertEqual([7.,8.,9.,10.,11.,12.,13.,14.,15.,5.,8.,9.,5.,8.,9.,22.,23.,24.,25.,26.,27.,5.,8.,9.,31.,32.,33.,34.,35.,36.0],da.getValues())
311 da.rearrange(1) ; da.iota(7) ; da.rearrange(3)
313 self.assertEqual([7.,8.,9.,10.,11.,12.,13.,14.,15.,16.,5.,8.,19.,5.,8.,22.,23.,24.,25.,26.,27.,28.,5.,8.,31.,32.,33.,34.,35.,36.],da.getValues())
316 def testSwigDAIOp(self):
317 da=DataArrayInt.New()
320 da1=DataArrayInt.New()
324 self.assertEqual([15,17,19,21,23,25,27,29,31,33,35,37],da2.getValues())
327 self.assertTrue(da2.isEqual(da3))
329 self.assertEqual([6,7,8,9,10,11,12,13,14,15,16,17],da2.getValues())
331 self.assertEqual([-6,-7,-8,-9,-10,-11,-12,-13,-14,-15,-16,-17],da2.getValues())
333 self.assertEqual([21,24,27,30,33,36,39,42,45,48,51,54.0],da2.getValues())
335 self.assertEqual([21,24,27,30,33,36,39,42,45,48,51,54.0],da2.getValues())
337 self.assertEqual([56,72,90,110,132,156,182,210,240,272,306,342.0],da2.getValues())
339 self.assertEqual([1,2,2,2,2,3,3,3,3,4,4,4],da2.getValues())
342 self.assertTrue(da4.isUniform(0))
343 st1=da.getHiddenCppPointer()
345 st2=da.getHiddenCppPointer()
346 self.assertEqual(st1,st2)
347 self.assertTrue(da.isEqual(da1))
349 st2=da.getHiddenCppPointer()
350 self.assertEqual(st1,st2)
351 self.assertEqual(list(range(12)), da.getValues())
353 st2=da.getHiddenCppPointer()
354 self.assertEqual(st1,st2)
355 self.assertEqual([8,10,12,14,16,18,20,22,24,26,28,30],da.getValues())
357 st2=da.getHiddenCppPointer()
358 self.assertEqual(st1,st2)
359 self.assertEqual([4,5,6,7,8,9,10,11,12,13,14,15],da.getValues())
361 st2=da.getHiddenCppPointer()
362 self.assertEqual(st1,st2)
363 self.assertEqual([32,45,60,77,96,117,140,165,192,221,252,285],da.getValues())
365 self.assertEqual(st1,st2)
366 self.assertEqual([4,5,6,7,8,9,10,11,12,13,14,15],da.getValues())
368 st2=da.getHiddenCppPointer()
369 self.assertEqual(st1,st2)
370 self.assertEqual([2,2, 3,3, 4,4, 5,5, 6,6, 7,7],da.getValues())
372 da5=DataArrayInt.New()
373 da5.setValues([5,4,3,2],4,1)
374 da*=da5 # it works with unmathing number of compo
375 st2=da.getHiddenCppPointer()
376 self.assertEqual(st1,st2)
377 self.assertEqual([10,10, 15,12,16,16,15,15, 18,12,14,14],da.getValues())
379 st2=da.getHiddenCppPointer()
380 self.assertEqual(st1,st2)
381 self.assertEqual([4,4,3,0,4,4,3,3,0,0,2,2],da.getValues())
386 ids=DataArrayInt.New()
387 ids.setValues([3,4,7],3,1)
389 self.assertEqual([7,8,9,10,11,12,13,14,15,5,8,9,5,8,9,22,23,24,25,26,27,5,8,9,31,32,33,34,35,36],da.getValues())
391 da.rearrange(1) ; da.iota(7) ; da.rearrange(3)
393 self.assertEqual([7,8,9,10,11,12,13,14,15,16,5,8,19,5,8,22,23,24,25,26,27,28,5,8,31,32,33,34,35,36],da.getValues())
396 def testSwigDAIOp2(self):
397 da=DataArrayInt.New()
398 st=da.getHiddenCppPointer()
408 self.assertEqual(st,da.getHiddenCppPointer())
409 self.assertEqual(da.getValues(),[7,7,7,3,8,8,7,7,7,9,14,20,12,17,26,7,7,7,18,23,38,21,26,44,24,29,70,27,32,76])
412 def testSwigDAIOp3(self):
413 da=DataArrayInt.New()
414 self.assertRaises(InterpKernelException,da.__len__)
415 self.assertRaises(InterpKernelException,da.__int__)
417 self.assertTrue(False)
420 da.rearrange(1) ; da.fillWithZero()
422 self.assertEqual(36,len(da));
425 self.assertRaises(InterpKernelException,tmp.__int__)
426 self.assertEqual(12,len(da));
428 for elt in enumerate(l):
431 ref=[0,0,0,0,0,1,0,0,2,0,0,3,0,0,4,0,0,5,0,0,6,0,0,7,0,0,8,0,0,9,0,0,10,0,0,11]
432 self.assertEqual(ref,da.getValues());
434 l=[int(elt) for elt in l1]
435 self.assertEqual(ref,da.getValues());
436 self.assertEqual(11,int(da[-1:]))
439 def testSwigDADOp3(self):
440 da=DataArrayDouble.New()
441 self.assertRaises(InterpKernelException,da.__len__)
442 self.assertRaises(InterpKernelException,da.__float__)
444 self.assertTrue(False)
447 da.rearrange(1) ; da.fillWithZero()
449 self.assertEqual(36,len(da));
452 self.assertRaises(InterpKernelException,tmp.__float__)
453 self.assertEqual(12,len(da));
455 for elt in enumerate(l):
458 ref=[0.,0.,0.,0.,0.,1.,0.,0.,2.,0.,0.,3.,0.,0.,4.,0.,0.,5.,0.,0.,6.,0.,0.,7.,0.,0.,8.,0.,0.,9.,0.,0.,10.,0.,0.,11.]
459 self.assertEqual(ref,da.getValues());
461 l=[float(elt) for elt in l1]
462 self.assertEqual(ref,da.getValues());
463 self.assertEqual(11.,float(da[-1:]))
466 def testSwigDataArrayIntIterator1(self):
467 da=DataArrayInt.New()
471 # __getitem__ testing
476 self.assertEqual([3, 4, 6, 7, 9, 10, 12, 13],li)
481 self.assertEqual([4, 7, 10, 13],li)
486 self.assertEqual([4, 3, 2, 7, 6, 5, 10, 9, 8, 13, 12, 11],li)
487 # __setitem__ testing
489 da2=DataArrayInt.New()
499 self.assertTrue(da.isEqual(da2))
507 self.assertTrue(da.isUniform(5))
514 self.assertEqual([8, 9, 12, 8, 9, 12, 8, 9, 12, 8, 9, 12],da.getValues())
521 self.assertEqual([2, 3, 7, 5, 6, 7, 8, 9, 7, 11, 12, 7],da.getValues())
524 def testSwigDataArrayDoubleIterator1(self):
525 da=DataArrayDouble.New()
529 # __getitem__ testing
534 self.assertEqual([3, 4, 6, 7, 9, 10, 12, 13],li)
539 self.assertEqual([4, 7, 10, 13],li)
544 self.assertEqual([4, 3, 2, 7, 6, 5, 10, 9, 8, 13, 12, 11],li)
545 # __setitem__ testing
547 da2=DataArrayDouble.New()
557 self.assertTrue(da.isEqual(da2,1e-12))
565 self.assertTrue(da.isUniform(5,1e-12))
572 self.assertEqual([8, 9, 12, 8, 9, 12, 8, 9, 12, 8, 9, 12],da.getValues())
579 self.assertEqual([2, 3, 7, 5, 6, 7, 8, 9, 7, 11, 12, 7],da.getValues())
582 def testSwigUMeshIterator1(self):
583 m=MEDCouplingDataForTest.build2DTargetMesh_1()
587 li1+=cell.getAllConn()[1:]
588 li2+=[cell.getType()]
590 self.assertEqual(li1,[0, 3, 4, 1, 1, 4, 2, 4, 5, 2, 6, 7, 4, 3, 7, 8, 5, 4])
591 self.assertEqual(li2,[4, 3, 3, 4, 4])
594 def testSwigUMeshIterator2(self):
595 m=MEDCouplingDataForTest.build2DTargetMesh_1()
596 self.assertRaises(InterpKernelException,m.cellsByType);
597 m.rearrange2ConsecutiveCellTypes()
601 for cellsByType in m.cellsByType():
602 li1.append(cellsByType.getType())
603 li2.append(cellsByType.getNumberOfElems())
605 for cell in cellsByType:
608 t[1]=cell.getAllConn()[1:]
613 self.assertEqual(li1,[4, 3])
614 self.assertEqual(li2,[3, 2])
615 self.assertEqual(li3,[[[4, (0, 3, 4, 1)], [4, (6, 7, 4, 3)], [4, (7, 8, 5, 4)]], [[3, (1, 4, 2)], [3, (4, 5, 2)]]])
618 def testDAIAggregateMulti1(self):
620 a.setValues(list(range(4)), 2, 2)
623 b.setValues(list(range(6)), 3, 2)
624 c=DataArrayInt.Aggregate([a,b])
625 self.assertEqual(list(range(4)) + list(range(6)), c.getValues())
626 self.assertEqual("aa",c.getName())
627 self.assertEqual(5,c.getNumberOfTuples())
628 self.assertEqual(2,c.getNumberOfComponents())
631 def testMergeUMeshes2(self):
632 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
633 m2=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
634 m3=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
637 m2_2=m2.buildPartOfMySelf(vec1,False);
639 m3_2=m3.buildPartOfMySelf(vec2,False);
643 self.assertRaises(InterpKernelException,MEDCouplingUMesh.MergeUMeshes,ms+[None]);
644 self.assertRaises(InterpKernelException,MEDCouplingUMesh.MergeUMeshes,ms+[3.4])
645 m4=MEDCouplingUMesh.MergeUMeshes(ms);
646 m4.checkConsistencyLight();
647 self.assertEqual(10,m4.getNumberOfCells());
648 self.assertEqual(20,m4.getNumberOfNodes());
649 self.assertEqual(45,m4.getNodalConnectivityArrayLen());
650 m4bis=MEDCouplingMesh.MergeMeshes(ms);
651 self.assertTrue(m4.isEqual(m4bis,1e-12))
655 m4_1=m4.buildPartOfMySelf(vec3,False);
656 m4_1.setName(m1.getName());
657 self.assertTrue(m4_1.isEqual(m1,1e-12));
660 m4_2=m4.buildPartOfMySelf(vec4,False);
661 cellCor,nodeCor=m4_2.checkGeoEquivalWith(m2_2,10,1e-12);
664 m4_3=m4.buildPartOfMySelf(vec5,False);
665 self.assertEqual(2,m4_3.getNumberOfCells());
666 self.assertEqual(3,m4_3.getNumberOfNodes());
668 m4_3.setName(m3_2.getName());
669 self.assertTrue(m4_3.isEqual(m3_2,1e-12));
673 def testBuild0DMeshFromCoords1(self):
674 sourceCoords=[-0.3,-0.3,0., 0.7,-0.3,0., -0.3,0.7,0., 0.7,0.7,0.]
675 coo=DataArrayDouble.New();
676 coo.setValues(sourceCoords,4,3);
678 m=MEDCouplingUMesh.Build0DMeshFromCoords(coo);
679 m.checkConsistencyLight();
680 self.assertEqual(4,m.getNumberOfNodes());
681 self.assertEqual(4,m.getNumberOfCells());
682 self.assertEqual(3,m.getSpaceDimension());
683 self.assertEqual(0,m.getMeshDimension());
684 types1=m.getAllGeoTypes();
685 self.assertEqual([NORM_POINT1],types1);
687 conn=m.getNodeIdsOfCell(i);
688 self.assertEqual([i],conn);
689 self.assertTrue(NORM_POINT1==m.getTypeOfCell(i));
691 self.assertEqual(m.getName(),"My0D");
694 def testDescriptionInMeshTimeUnit1(self):
696 m=MEDCouplingDataForTest.build2DTargetMesh_1();
697 m.setDescription(text1);
698 self.assertEqual(m.getDescription(),text1);
700 self.assertTrue(m.isEqual(m2,1e-12));
701 self.assertEqual(m2.getDescription(),text1);
702 m2.setDescription("ggg");
703 self.assertTrue(not m.isEqual(m2,1e-12));
705 f=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
706 f.setTimeUnit(text1);
707 self.assertEqual(f.getTimeUnit(),text1);
709 self.assertEqual(f2.getTimeUnit(),text1);
713 def testMultiFields1(self):
714 mfs=MEDCouplingDataForTest.buildMultiFields_1();
716 dms,refs=mfs.getDifferentMeshes()
718 das2,refs2=mfs.getDifferentArrays()
719 self.assertEqual(5,len(mfs.getFields()))
720 self.assertEqual(1,len(mfs.getFields()[0].getArrays()));
721 self.assertEqual(2,len(mfs.getFields()[1].getArrays()));
722 self.assertEqual(1,len(mfs.getFields()[2].getArrays()));
723 self.assertEqual(1,len(mfs.getFields()[3].getArrays()));
724 self.assertEqual(1,len(mfs.getFields()[4].getArrays()));
725 self.assertEqual(5,len(ms));
726 self.assertEqual(2,len(dms));
727 self.assertEqual(6,len(das));
728 self.assertEqual(5,len(das2));
730 self.assertTrue(mfs.isEqual(mfs2,1e-12,1e-12))
733 def testFieldOverTime1(self):
734 fs=MEDCouplingDataForTest.buildMultiFields_2();
735 self.assertRaises(InterpKernelException,MEDCouplingFieldOverTime.New,fs);
736 f4bis=fs[4].buildNewTimeReprFromThis(ONE_TIME,False);
738 self.assertRaises(InterpKernelException,MEDCouplingFieldOverTime.New,fs);
739 f4bis.setTime(2.7,20,21);
740 fot=MEDCouplingFieldOverTime.New(fs);
741 dt=fot.getDefinitionTimeZone();
742 hs=dt.getHotSpotsTime();
743 self.assertEqual(6,len(hs));
744 expected1=[0.2,0.7,1.2,1.35,1.7,2.7]
746 self.assertAlmostEqual(expected1[i],hs[i],12);
748 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(0.2);
749 self.assertEqual(0,meshId);
750 self.assertEqual(0,arrId);
751 self.assertEqual(0,arrIdInField);
752 self.assertEqual(0,fieldId);
754 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(0.7);
755 self.assertEqual(0,meshId);
756 self.assertEqual(1,arrId);
757 self.assertEqual(0,arrIdInField);
758 self.assertEqual(1,fieldId);
760 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeLeft(1.2);#**** WARNING left here
761 self.assertEqual(0,meshId);
762 self.assertEqual(2,arrId);
763 self.assertEqual(1,arrIdInField);
764 self.assertEqual(1,fieldId);
766 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(1.2);#**** WARNING right again here
767 self.assertEqual(1,meshId);
768 self.assertEqual(3,arrId);
769 self.assertEqual(0,arrIdInField);
770 self.assertEqual(2,fieldId);
772 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(1.35);
773 self.assertEqual(1,meshId);
774 self.assertEqual(3,arrId);
775 self.assertEqual(0,arrIdInField);
776 self.assertEqual(2,fieldId);
778 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(1.7);
779 self.assertEqual(0,meshId);
780 self.assertEqual(3,arrId);
781 self.assertEqual(0,arrIdInField);
782 self.assertEqual(3,fieldId);
784 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(2.7);
785 self.assertEqual(1,meshId);
786 self.assertEqual(4,arrId);
787 self.assertEqual(0,arrIdInField);
788 self.assertEqual(4,fieldId);
790 dt2=MEDCouplingDefinitionTime();
791 self.assertTrue(not dt2.isEqual(dt));
793 dt2.assign(dt);#to check memory management
794 self.assertTrue(dt2.isEqual(dt));
796 dt3=MEDCouplingDefinitionTime();
800 def testDAICheckAndPreparePermutation1(self):
801 vals1=[9,10,0,6,4,11,3,7];
802 expect1=[5,6,0,3,2,7,1,4];
803 vals2=[9,10,0,6,10,11,3,7];
804 da=DataArrayInt.New();
805 da.setValues(vals1,8,1);
806 da2=da.checkAndPreparePermutation();
807 self.assertEqual(8,da2.getNumberOfTuples());
808 self.assertEqual(1,da2.getNumberOfComponents());
810 self.assertEqual(expect1[i],da2.getIJ(i,0));
813 da=DataArrayInt.New();
816 da2=da.checkAndPreparePermutation();
817 self.assertEqual(1,da2.getNumberOfComponents());
818 self.assertTrue(da2.isIota(8));
820 da=DataArrayInt.New();
822 da.setValues(vals2,8,1);
823 self.assertRaises(InterpKernelException,da.checkAndPreparePermutation);
826 def testDAIChangeSurjectiveFormat1(self):
827 vals1=[0,3,2,3,2,2,1,2]
828 expected1=[0,1,2,6,8]
829 expected2=[0, 6, 2,4,5,7, 1,3]
830 da=DataArrayInt.New();
831 da.setValues(vals1,8,1);
833 da2,da2I=da.changeSurjectiveFormat(4);
834 self.assertEqual(5,da2I.getNumberOfTuples());
835 self.assertEqual(8,da2.getNumberOfTuples());
836 self.assertEqual(expected1,da2I.getValues());
837 self.assertEqual(expected2,da2.getValues());
839 self.assertRaises(InterpKernelException,da.changeSurjectiveFormat,3);
843 def testUMeshGetCellIdsLyingOnNodes1(self):
844 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
847 da=m.getCellIdsLyingOnNodes(nodeIds1,True);
848 self.assertEqual(1,da.getNumberOfTuples());
849 self.assertEqual(1,da.getNumberOfComponents());
850 self.assertEqual(1,da.getIJ(0,0));
851 da2=DataArrayInt.New()
852 da2.setValues(nodeIds2,2,1)
853 da=m.getCellIdsLyingOnNodes(da2,False);
854 self.assertEqual(2,da.getNumberOfTuples());
855 self.assertEqual(1,da.getNumberOfComponents());
856 self.assertEqual(3,da.getIJ(0,0));
857 self.assertEqual(4,da.getIJ(1,0));
860 def testUMeshFindCellIdsOnBoundary1(self):
861 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
862 da5=m.findCellIdsOnBoundary();
863 self.assertTrue(da5.isIota(5));
866 def testMeshSetTime1(self):
867 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
868 m2=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
870 self.assertTrue(m1.isEqual(m2,1e-12));
871 m1.setTime(3.14,6,7);
872 tmp3,tmp1,tmp2=m1.getTime();
873 self.assertEqual(6,tmp1);
874 self.assertEqual(7,tmp2);
875 self.assertAlmostEqual(3.14,tmp3,12);
876 self.assertTrue(not m1.isEqual(m2,1e-12));
877 m2.setTime(3.14,6,7);
878 self.assertTrue(m1.isEqual(m2,1e-12));
879 m1.setTimeUnit("ms");
880 self.assertTrue(m1.getTimeUnit()=="ms");
881 m1.setTimeUnit("us");
882 self.assertTrue(m1.getTimeUnit()=="us");
883 self.assertTrue(not m1.isEqual(m2,1e-12));
884 m2.setTimeUnit("us");
885 self.assertTrue(m1.isEqual(m2,1e-12));
886 m2.setTime(3.14,6,8);
887 self.assertTrue(not m1.isEqual(m2,1e-12));
888 m2.setTime(3.14,7,7);
889 self.assertTrue(not m1.isEqual(m2,1e-12));
890 m2.setTime(3.15,6,7);
891 self.assertTrue(not m1.isEqual(m2,1e-12));
893 m1.setTime(10.34,55,12);
895 self.assertTrue(m1.isEqual(m3,1e-12));
896 tmp3,tmp1,tmp2=m3.getTime();
897 self.assertEqual(55,tmp1);
898 self.assertEqual(12,tmp2);
899 self.assertAlmostEqual(10.34,tmp3,12);
903 a=DataArrayDouble.New();
904 a.setValues(coo1,4,1);
905 b=MEDCouplingCMesh.New();
908 b.setTime(5.67,8,100);
909 tmp3,tmp1,tmp2=b.getTime();
910 self.assertEqual(8,tmp1);
911 self.assertEqual(100,tmp2);
912 self.assertAlmostEqual(5.67,tmp3,12);
914 self.assertTrue(c.isEqual(b,1e-12));
915 tmp3,tmp1,tmp2=c.getTime();
916 self.assertEqual(8,tmp1);
917 self.assertEqual(100,tmp2);
918 self.assertAlmostEqual(5.67,tmp3,12);
921 def testApplyFuncTwo1(self):
922 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
923 f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
926 vals=[1.,11.,21.,2.,12.,22.,3.,13.,23.,4.,14.,24.,5.,15.,25.]
927 da=DataArrayDouble.New();
928 da.setValues(vals,5,3);
931 self.assertRaises(InterpKernelException,da.applyFuncCompo,1,"y+z");
932 da.setInfoOnComponent(0,"x [m]");
933 da.setInfoOnComponent(1,"y [mm]");
934 da.setInfoOnComponent(2,"z [km]");
936 self.assertRaises(InterpKernelException, da.applyFuncCompo, 1, "x+y+zz+zzz");
937 self.assertRaises(InterpKernelException, da.applyFuncCompo, 1, "toto(x+y)");
938 self.assertRaises(InterpKernelException, da.applyFuncCompo, 1, "x/0");
940 da2=da.applyFuncCompo(1,"y+z");
941 self.assertEqual(1,da2.getNumberOfComponents());
942 self.assertEqual(5,da2.getNumberOfTuples());
943 expected1=[32.,34.,36.,38.,40.]
945 self.assertAlmostEqual(expected1[i],da2.getIJ(0,i),12);
947 da2=da.applyFunc(1,"y+z");
948 expected2=[12.,14.,16.,18.,20.]
950 self.assertAlmostEqual(expected2[i],da2.getIJ(0,i),12);
953 self.assertEqual(3,f1.getNumberOfComponents());
954 self.assertEqual(5,f1.getNumberOfTuples());
955 f1.applyFuncCompo(1,"y+z");
956 self.assertEqual(1,f1.getNumberOfComponents());
957 self.assertEqual(5,f1.getNumberOfTuples());
959 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
964 def testApplyFuncThree1(self):
965 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
966 f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
969 vals=[1.,11.,21.,2.,12.,22.,3.,13.,23.,4.,14.,24.,5.,15.,25.]
970 da=DataArrayDouble.New();
971 da.setValues(vals,5,3);
975 vs[0]="x"; vs[1]="Y"; vs[2]="z";
976 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, vs, "y+z");
977 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, vs, "x+Y+z+zz+zzz");
978 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, vs, "x/0");
980 da2=da.applyFuncNamedCompo(1,vs,"y+z");
981 expected1=[32.,34.,36.,38.,40.]
983 self.assertAlmostEqual(expected1[i],da2.getIJ(0,i),12);
985 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, ["x","y","z","a"],"x+a")
987 self.assertEqual(3,f1.getNumberOfComponents());
988 self.assertEqual(5,f1.getNumberOfTuples());
989 f1.applyFuncNamedCompo(1,vs,"y+z");
990 self.assertEqual(1,f1.getNumberOfComponents());
991 self.assertEqual(5,f1.getNumberOfTuples());
993 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
997 def testFillFromAnalyticTwo1(self):
998 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
999 m1.setTime(3.4,5,6); m1.setTimeUnit("us");
1000 self.assertRaises(InterpKernelException,m1.fillFromAnalyticCompo,ON_NODES,1,"y+z");
1001 m1.getCoords().setInfoOnComponent(0,"x [m]");
1002 m1.getCoords().setInfoOnComponent(1,"y");
1003 m1.getCoords().setInfoOnComponent(2,"z");
1004 f1=m1.fillFromAnalyticCompo(ON_NODES,1,"y+z");
1005 self.assertAlmostEqual(3.4,f1.getTime()[0],12) ; self.assertEqual(5,f1.getTime()[1]) ; self.assertEqual(6,f1.getTime()[2])
1006 self.assertEqual("us",f1.getTimeUnit())
1007 self.assertEqual(1,f1.getNumberOfComponents());
1008 self.assertEqual(9,f1.getNumberOfTuples());
1009 expected1=[0.2, 0.7, 1.2, 0.7, 1.2, 1.7, 1.2, 1.7, 2.2]
1011 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
1015 def testFillFromAnalyticThree1(self):
1016 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
1017 m1.setTime(3.4,5,6); m1.setTimeUnit("us");
1019 vs[0]="x"; vs[1]="Y"; vs[2]="z";
1020 self.assertRaises(InterpKernelException,m1.fillFromAnalyticNamedCompo,ON_NODES,1,vs,"y+z");
1022 f1=m1.fillFromAnalyticNamedCompo(ON_NODES,1,vs,"y+z");
1023 self.assertAlmostEqual(3.4,f1.getTime()[0],12) ; self.assertEqual(5,f1.getTime()[1]) ; self.assertEqual(6,f1.getTime()[2])
1024 self.assertEqual("us",f1.getTimeUnit())
1025 self.assertEqual(1,f1.getNumberOfComponents());
1026 self.assertEqual(9,f1.getNumberOfTuples());
1027 expected1=[0.2, 0.7, 1.2, 0.7, 1.2, 1.7, 1.2, 1.7, 2.2]
1029 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
1033 def testDAUnitVar1(self):
1034 da=DataArrayDouble.New();
1036 da.setInfoOnComponent(0,"XPS [m]");
1037 st1=da.getVarOnComponent(0);
1038 self.assertTrue(st1=="XPS");
1039 st2=da.getUnitOnComponent(0);
1040 self.assertTrue(st2=="m");
1042 da.setInfoOnComponent(0,"XPS [m]");
1043 st1=da.getVarOnComponent(0);
1044 self.assertTrue(st1=="XPS");
1045 st2=da.getUnitOnComponent(0);
1046 self.assertTrue(st2=="m");
1048 da.setInfoOnComponent(0,"XPP [m]");
1049 st1=da.getVarOnComponent(0);
1050 self.assertTrue(st1=="XPP");
1051 st2=da.getUnitOnComponent(0);
1052 self.assertTrue(st2=="m");
1054 da.setInfoOnComponent(0,"XPP kdep kefer [ m ]");
1055 st1=da.getVarOnComponent(0);
1056 self.assertTrue(st1=="XPP kdep kefer");
1057 st2=da.getUnitOnComponent(0);
1058 self.assertTrue(st2==" m ");
1060 da.setInfoOnComponent(0," XPP k[ dep k]efer [ m^ 2/s^3*kJ ]");
1061 st1=da.getVarOnComponent(0);
1062 self.assertTrue(st1==" XPP k[ dep k]efer");
1063 st2=da.getUnitOnComponent(0);
1064 self.assertTrue(st2==" m^ 2/s^3*kJ ");
1066 da.setInfoOnComponent(0," XPP kefer ");
1067 st1=da.getVarOnComponent(0);
1068 self.assertTrue(st1==" XPP kefer ");
1069 st2=da.getUnitOnComponent(0);
1070 self.assertTrue(st2=="");
1072 da.setInfoOnComponent(0,"temperature( bof)");
1073 st1=da.getVarOnComponent(0);
1074 self.assertTrue(st1=="temperature( bof)");
1075 st2=da.getUnitOnComponent(0);
1076 self.assertTrue(st2=="");
1078 da.setInfoOnComponent(0,"kkk [m]");
1079 da.setInfoOnComponent(1,"ppp [m^2/kJ]");
1080 da.setInfoOnComponent(2,"abcde [MW/s]");
1082 vs=da.getVarsOnComponent();
1083 self.assertEqual(3,len(vs));
1084 self.assertTrue(vs[0]=="kkk");
1085 self.assertTrue(vs[1]=="ppp");
1086 self.assertTrue(vs[2]=="abcde");
1087 vs=da.getUnitsOnComponent();
1088 self.assertEqual(3,len(vs));
1089 self.assertTrue(vs[0]=="m");
1090 self.assertTrue(vs[1]=="m^2/kJ");
1091 self.assertTrue(vs[2]=="MW/s");
1094 def testGaussCoordinates1(self):
1095 #Testing 1D cell types
1096 m1=MEDCouplingDataForTest.build1DMultiTypes_1();
1097 f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME);
1102 f.setGaussLocalizationOnType(NORM_SEG2,refCoo1,gsCoo1,wg1);
1105 refCoo2=[-1.0,1.0,0.0];
1106 f.setGaussLocalizationOnType(NORM_SEG3,refCoo2,gsCoo2,wg2);
1108 resToTest=f.getLocalizationOfDiscr();
1109 self.assertEqual(3,resToTest.getNumberOfComponents());
1110 self.assertEqual(2,resToTest.getNumberOfTuples());
1111 expected1=[0.6,0.6,0.6, 0.6,0.6,0.6]
1113 self.assertAlmostEqual(expected1[i],resToTest.getIJ(0,i),14);
1116 #Testing 2D cell types
1117 m2=MEDCouplingDataForTest.build2DMultiTypes_1();
1118 f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME);
1121 tria3CooGauss=[ 0.1, 0.8, 0.2, 0.7 ]
1122 gsCoo3=tria3CooGauss
1123 tria3CooRef=[ 0.0, 0.0, 1.0 , 0.0, 0.0, 1.0 ]
1124 refCoo3=tria3CooRef;
1125 f.setGaussLocalizationOnType(NORM_TRI3,refCoo3,gsCoo3,wg3);
1127 tria6CooGauss=[ 0.3, 0.2, 0.2, 0.1, 0.2, 0.4 ]
1128 gsCoo4=tria6CooGauss;
1129 tria6CooRef=[0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.5, 0.0, 0.5, 0.5, 0.0, 0.5]
1130 refCoo4=tria6CooRef;
1131 f.setGaussLocalizationOnType(NORM_TRI6,refCoo4,gsCoo4,wg4);
1132 wg5=[0.3,0.3,0.3,0.3];
1133 quad4CooGauss=[ 0.3, 0.2, 0.2, 0.1, 0.2, 0.4, 0.15, 0.27 ]
1134 gsCoo5=quad4CooGauss;
1135 quad4CooRef=[-1.0, 1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0]
1136 refCoo5=quad4CooRef;
1137 f.setGaussLocalizationOnType(NORM_QUAD4,refCoo5,gsCoo5,wg5);
1138 wg6=[0.3,0.3,0.3,0.3];
1139 quad8CooGauss=[ 0.34, 0.16, 0.21, 0.3, 0.23, 0.4, 0.14, 0.37 ]
1140 gsCoo6=quad8CooGauss;
1141 quad8CooRef=[ -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 0.0, -1.0, 1.0, 0.0, 0.0, 1.0, -1.0, 0.0]
1142 refCoo6=quad8CooRef;
1143 f.setGaussLocalizationOnType(NORM_QUAD8,refCoo6,gsCoo6,wg6);
1145 resToTest=f.getLocalizationOfDiscr();
1146 self.assertEqual(3,resToTest.getNumberOfComponents());
1147 self.assertEqual(13,resToTest.getNumberOfTuples());#2+3+4+4 gauss points for resp TRI3,TRI6,QUAD4,QUAD8
1148 expected2=[5.1,1.55,0.0, 4.7,1.65,0.0,
1149 2.32,1.52,0.0, 1.6,1.32,0.0, 3.52,1.26,0.0,#TRI6
1150 2.6,1.6,0.0, 2.4,1.8,0.0, 2.4,1.2,0.0, 2.3,1.46,0.0,#QUAD4
1151 2.32,2.68,0.0, 2.6,2.42,0.0, 2.8,2.46,0.0, 2.74,2.28,0.0 ];#QUAD8
1153 self.assertAlmostEqual(expected2[i],resToTest.getIJ(0,i),14);
1156 #Testing 3D cell types
1157 m3=MEDCouplingDataForTest.build3DMultiTypes_1();
1158 f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME);
1162 tetra4CooGauss=[0.34, 0.16, 0.21]
1163 gsCoo7=tetra4CooGauss;
1164 tetra4CooRef=[0.0,1.0,0.0, 0.0,0.0,1.0, 0.0,0.0,0.0, 1.0,0.0,0.0]
1165 refCoo7=tetra4CooRef;
1166 f.setGaussLocalizationOnType(NORM_TETRA4,refCoo7,gsCoo7,wg7);
1168 tetra10CooGauss=[0.2, 0.3, 0.1]
1169 gsCoo8=tetra10CooGauss;
1170 tetra10CooRef=[0.0,1.0,0.0, 0.0,0.0,0.0, 0.0,0.0,1.0, 1.0,0.0,0.0, 0.0,0.5,0.0, 0.0,0.0,0.5, 0.0,0.5,0.5, 0.5,0.5,0.0, 0.5,0.0,0.0, 0.5,0.0,0.5]
1171 refCoo8=tetra10CooRef;
1172 f.setGaussLocalizationOnType(NORM_TETRA10,refCoo8,gsCoo8,wg8);
1174 pyra5CooGauss=[0.2, 0.3, 0.1]
1175 gsCoo9=pyra5CooGauss;
1176 pyra5CooRef=[1.0,0.0,0.0, 0.0,1.0,0.0, -1.0,0.0,0.0, 0.0,-1.0,0.0, 0.0,0.0,1.0]
1177 refCoo9=pyra5CooRef;
1178 f.setGaussLocalizationOnType(NORM_PYRA5,refCoo9,gsCoo9,wg9);
1180 pyra13CooGauss=[0.1, 0.2, 0.7]
1181 gsCoo10=pyra13CooGauss;
1182 pyra13CooRef=[1.0,0.0,0.0, 0.0,1.0,0.0,-1.0,0.0,0.0,0.0,-1.0,0.0,0.0,0.0,1.0,0.5,0.5,0.0,-0.5,0.5,0.0,-0.5,-0.5,0.0,0.5,-0.5,0.0,0.5,0.0,0.5,0.0,0.5,0.5,-0.5,0.0,0.5,0.0,-0.5,0.5]
1183 refCoo10=pyra13CooRef;
1184 f.setGaussLocalizationOnType(NORM_PYRA13,refCoo10,gsCoo10,wg10);
1186 penta6CooGauss=[0.2, 0.3, 0.1]
1187 gsCoo11=penta6CooGauss;
1188 penta6CooRef=[-1.0,1.0,0.0,-1.0,-0.0,1.0,-1.0,0.0,0.0,1.0,1.0,0.0,1.0,0.0,1.0,1.0,0.0,0.0]
1189 refCoo11=penta6CooRef;
1190 f.setGaussLocalizationOnType(NORM_PENTA6,refCoo11,gsCoo11,wg11);
1192 penta15CooGauss=[0.2, 0.3,0.15]
1193 gsCoo12=penta15CooGauss;
1194 penta15CooRef=[-1.0,1.0,0.0,-1.0,0.0,1.0,-1.0,0.0,0.0,1.0,1.0,0.0,1.0,0.0,1.0,1.0,0.0,0.0,-1.0,0.5,0.5,-1.0,0.0,0.5,-1.0,0.5,0.0,0.0,1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0,0.5,0.5,1.0,0.0, 0.5,1.0,0.5,0.0]
1195 refCoo12=penta15CooRef;
1196 f.setGaussLocalizationOnType(NORM_PENTA15,refCoo12,gsCoo12,wg12);
1198 hexa8CooGauss=[0.2,0.3,0.15]
1199 gsCoo13=hexa8CooGauss;
1200 hexa8CooRef=[-1.0,-1.0,-1.0,1.0,-1.0,-1.0,1.0,1.0,-1.0,-1.0,1.0,-1.0,-1.0,-1.0,1.0,1.0,-1.0,1.0,1.0,1.0,1.0,-1.0,1.0,1.0]
1201 refCoo13=hexa8CooRef;
1202 f.setGaussLocalizationOnType(NORM_HEXA8,refCoo13,gsCoo13,wg13);
1204 hexa20CooGauss=[0.11,0.3,0.55]
1205 gsCoo14=hexa20CooGauss;
1206 hexa20CooRef=[-1.0,-1.0,-1.0,1.0,-1.0,-1.0,1.0,1.0,-1.0,-1.0,1.0,-1.0,-1.0,-1.0,1.0,1.0,-1.0,1.0,1.0,1.0,1.0,-1.0,1.0,1.0,0.0,-1.0,-1.0,1.0,0.0,-1.0,0.0,1.0,-1.0,-1.0,0.0,-1.0,-1.0,-1.0,0.0,1.0,-1.0,0.0,1.0,1.0,0.0,-1.0,1.0,0.0,0.0,-1.0,1.0,1.0,0.0,1.0,0.0,1.0,1.0,-1.0,0.0,1.0]
1207 refCoo14=hexa20CooRef;
1208 f.setGaussLocalizationOnType(NORM_HEXA20,refCoo14,gsCoo14,wg14);
1210 resToTest=f.getLocalizationOfDiscr();
1211 self.assertEqual(3,resToTest.getNumberOfComponents());
1212 self.assertEqual(8,resToTest.getNumberOfTuples());#2+3+4+4 gauss points for resp TRI3,TRI6,QUAD4,QUAD8
1213 expected3=[1.312,3.15,1.02, 0.56,3.3,0.6, 2.20,1.0,0.2, 2.44, 2.52,1.4, 1.56,0.3,3.6, 1.613,0.801,4.374, 2.6,2.4,2.3, 2.31232,2.3933985,1.553255]
1215 self.assertAlmostEqual(expected3[i],resToTest.getIJ(0,i),14);
1220 def testP2Localization1(self):
1221 m=MEDCouplingUMesh.New("testP2",2);
1222 coords=[0.,2.,3.5,0.,4.5,1.5,1.2,0.32,3.4,1.,2.1,2.4]
1224 coo=DataArrayDouble.New();
1225 coo.setValues(coords,6,2);
1228 m.insertNextCell(NORM_TRI6,6,conn[0:6])
1229 m.finishInsertingCells();
1231 f=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME);
1233 da=DataArrayDouble.New();
1234 vals1=[1.2,2.3,3.4, 2.2,3.3,4.4, 3.2,4.3,5.4, 4.2,5.3,6.4, 5.2,6.3,7.4, 6.2,7.3,8.4]
1235 da.setValues(vals1,6,3);
1239 locs=f.getValueOnMulti(loc);
1240 expected1=[6.0921164547752236, 7.1921164547752232, 8.2921164547752255]
1242 self.assertAlmostEqual(expected1[i],locs.getIJ(0,i),12);
1246 def testP2Localization2(self):
1247 m=MEDCouplingUMesh.New("testP2_2",3);
1248 coords=[0.33312787792955395, -0.35155740179580952, -0.03567564825034563, 1.307146326477638, -0.57234557776250305, -0.08608044208272235, 0.5551834466499993, 0.62324964668794192, -0.014638951108536295, 0.37761817224442129, -0.38324019806913578, 0.96283164472856886, 0.79494856035658679, -0.40628057809270046, 0.0021004190225864614, 1.023740446371799, 0.07665912970471335, -0.072889657161871096, 0.54564584619517376, 0.11132872093429744, 0.039647326652013051, 0.27164784387819052, -0.42018012100866675, 0.46563376500745146, 0.89501965094896418, -0.56148455362735061, 0.43337469695473035, 0.49118025152924394, 0.093884938060727313, 0.47216346905220891]
1249 conn=[0,1,2,3,4,5,6,7,8,9]
1250 coo=DataArrayDouble.New();
1251 coo.setValues(coords,10,3);
1254 m.insertNextCell(NORM_TETRA10,10,conn[0:10])
1255 m.finishInsertingCells();
1257 f=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME);
1259 da=DataArrayDouble.New();
1260 vals1=[1.1,2.1,3.1,4.1,5.2,6.2,7.2,8.2,9.2,10.2]
1261 da.setValues(vals1,10,1);
1263 f.checkConsistencyLight()
1265 loc=[0.64637931739890486, -0.16185896817550552, 0.22678966365273748]
1266 locs=f.getValueOnMulti(loc);
1267 expected1=[10.0844021968047]
1269 self.assertAlmostEqual(expected1[i],locs.getIJ(0,i),12);
1273 def testGetValueOn2(self):
1274 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1275 f=MEDCouplingFieldDouble.New(ON_CELLS,NO_TIME);
1277 arr=DataArrayDouble.New();
1278 nbOfCells=m.getNumberOfCells();
1280 values1=[7.,107.,10007.,8.,108.,10008.,9.,109.,10009.,10.,110.,10010.,11.,111.,10011.]
1281 arr.setValues(values1,nbOfCells,3);
1282 loc=[-0.05,-0.05, 0.55,-0.25, 0.55,0.15, -0.05,0.45, 0.45,0.45]
1283 f.checkConsistencyLight();
1284 locs=f.getValueOnMulti(loc);
1285 self.assertEqual(5,locs.getNumberOfTuples());
1286 self.assertEqual(3,locs.getNumberOfComponents());
1288 self.assertAlmostEqual(values1[j],locs.getIJ(0,j),12);
1291 f=MEDCouplingFieldDouble.New(ON_NODES,NO_TIME);
1293 arr=DataArrayDouble.New();
1294 nbOfNodes=m.getNumberOfNodes();
1296 values2=[7.,107.,10007.,8.,108.,10008.,9.,109.,10009.,10.,110.,10010.,11.,111.,10011.,12.,112.,10012.,13.,113.,10013.,14.,114.,10014.,15.,115.,10015.]
1297 arr.setValues(values2,nbOfNodes,3);
1298 loc2=[0.5432,-0.2432, 0.5478,0.1528, 0.5432,-0.2432, 0.5432,-0.2432]
1299 expected2=[9.0272, 109.0272, 10009.0272, 11.4124,111.4124,10011.4124, 9.0272, 109.0272, 10009.0272, 9.0272, 109.0272, 10009.0272]
1300 f.checkConsistencyLight();
1301 loc3=DataArrayDouble.New()
1302 loc3.setValues(loc2,4,2);
1303 locs=f.getValueOnMulti(loc3);
1304 self.assertEqual(4,locs.getNumberOfTuples());
1305 self.assertEqual(3,locs.getNumberOfComponents());
1307 self.assertAlmostEqual(expected2[i],locs.getIJ(0,i),12);
1312 def testDAIGetIdsNotEqual1(self):
1313 d=DataArrayInt.New();
1314 vals1=[2,3,5,6,8,5,5,6,1,-5]
1315 d.setValues(vals1,10,1);
1316 d2=d.findIdsNotEqual(5);
1317 self.assertEqual(7,d2.getNumberOfTuples());
1318 self.assertEqual(1,d2.getNumberOfComponents());
1319 expected1=[0,1,3,4,7,8,9]
1321 self.assertEqual(expected1[i],d2.getIJ(0,i));
1324 self.assertRaises(InterpKernelException,d.findIdsNotEqual,5);
1328 d3=d.findIdsNotEqualList(vals3);
1329 self.assertEqual(5,d3.getNumberOfTuples());
1330 self.assertEqual(1,d3.getNumberOfComponents());
1331 expected2=[0,1,4,8,9]
1333 self.assertEqual(expected2[i],d3.getIJ(0,i));
1337 def testDAIComputeOffsets1(self):
1338 d=DataArrayInt.New();
1340 expected1=[0,3,8,9,11,11]
1341 d.setValues(vals1,6,1);
1343 self.assertEqual(6,d.getNumberOfTuples());
1344 self.assertEqual(1,d.getNumberOfComponents());
1346 self.assertEqual(expected1[i],d.getIJ(0,i));
1350 def testUMeshHexagonPrism1(self):
1351 coords=[0.8660254037844386, 0.5, 0.0, 0.0, 1.0, 0.0, -0.8660254037844386, 0.5, 0.0, -0.8660254037844386, -0.5, 0.0, 0.0, -1.0, 0.0, 0.8660254037844386, -0.5, 0.0,
1352 0.8660254037844386, 0.5, 2.0, 0.0, 1.0, 2.0, -0.8660254037844386, 0.5, 2.0, -0.8660254037844386, -0.5, 2.0, 0.0, -1.0, 2.0, 0.8660254037844386, -0.5, 2.0];
1353 conn=[1,2,3,4,5,0,7,8,9,10,11,6]
1354 mesh=MEDCouplingUMesh.New("MyFirstHexagonalPrism",3);
1355 coo=DataArrayDouble.New();
1356 coo.setValues(coords,12,3);
1357 mesh.setCoords(coo);
1358 mesh.allocateCells(1);
1359 mesh.insertNextCell(NORM_HEXGP12,12,conn[0:12])
1360 mesh.finishInsertingCells();
1362 mesh.checkConsistencyLight();
1363 vols=mesh.getMeasureField(False);
1364 self.assertEqual(1,vols.getNumberOfTuples());
1365 self.assertEqual(1,vols.getNumberOfComponents());
1366 self.assertAlmostEqual(-5.196152422706632,vols.getIJ(0,0),12);
1367 bary=mesh.computeCellCenterOfMass();
1368 self.assertEqual(1,bary.getNumberOfTuples());
1369 self.assertEqual(3,bary.getNumberOfComponents());
1370 expected1=[0.,0.,1.]
1372 self.assertAlmostEqual(expected1[i],bary.getIJ(0,i),12);
1374 d1=DataArrayInt.New();
1375 d2=DataArrayInt.New();
1376 d3=DataArrayInt.New();
1377 d4=DataArrayInt.New();
1378 m2=mesh.buildDescendingConnectivity(d1,d2,d3,d4);
1379 self.assertEqual(8,m2.getNumberOfCells());
1380 expected4=[[1,2,3,4,5,0],[7,6,11,10,9,8],[1,7,8,2],[2,8,9,3],[3,9,10,4],[4,10,11,5],[5,11,6,0],[0,6,7,1]];
1381 expected2=[NORM_POLYGON, NORM_POLYGON, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4];
1382 expected3=[6,6,4,4,4,4,4,4]
1384 self.assertTrue(m2.getTypeOfCell(i)==expected2[i]);
1385 v=m2.getNodeIdsOfCell(i);
1386 self.assertTrue(len(v)==expected3[i]);
1387 self.assertEqual(expected4[i],v);
1389 mesh.convertAllToPoly();
1390 self.assertTrue(NORM_POLYHED==mesh.getTypeOfCell(0));
1392 self.assertTrue(NORM_HEXGP12==mesh.getTypeOfCell(0));
1393 self.assertEqual(13,mesh.getNodalConnectivityArrayLen());
1397 def testDADCheckIsMonotonic(self):
1398 da=DataArrayDouble.New();
1399 da.setValues([-1.,1.01,2.03,6.],2,2);
1400 self.assertRaises(InterpKernelException,da.isMonotonic,True,1e-12);
1402 self.assertTrue(da.isMonotonic(True,1e-12));
1403 da.checkMonotonic(True,1e-12);
1405 self.assertTrue(not da.isMonotonic(True,1e-12));
1406 self.assertRaises(InterpKernelException,da.checkMonotonic,True,1e-12);
1408 self.assertTrue(da.isMonotonic(True,1e-12));
1409 self.assertTrue(not da.isMonotonic(True,1e-1));
1412 def testCheckCoherencyDeeper1(self):
1413 m=MEDCouplingDataForTest.build3DSourceMesh_1();
1414 m.checkConsistencyLight();
1415 m.checkConsistency();
1416 m.getNodalConnectivity().setIJ(8,0,-1);
1417 m.checkConsistencyLight();
1418 self.assertRaises(InterpKernelException,m.checkConsistency);
1419 m.getNodalConnectivity().setIJ(8,0,-6);
1420 m.checkConsistencyLight();
1421 self.assertRaises(InterpKernelException,m.checkConsistency);
1422 m.getNodalConnectivity().setIJ(8,0,9);#9>=NbOfNodes
1423 m.checkConsistencyLight();
1424 self.assertRaises(InterpKernelException,m.checkConsistency);
1425 m.getNodalConnectivity().setIJ(8,0,8);#OK
1426 m.checkConsistencyLight();
1427 m.checkConsistency();
1429 m.convertToPolyTypes(elts);
1430 m.checkConsistencyLight();
1431 m.checkConsistency();
1432 m.getNodalConnectivity().setIJ(2,0,9);#9>=NbOfNodes
1433 m.checkConsistencyLight();
1434 self.assertRaises(InterpKernelException,m.checkConsistency);
1435 m.getNodalConnectivity().setIJ(2,0,-3);
1436 m.checkConsistencyLight();
1437 self.assertRaises(InterpKernelException,m.checkConsistency);
1438 m.getNodalConnectivity().setIJ(2,0,-1);
1439 m.checkConsistencyLight();
1440 self.assertRaises(InterpKernelException,m.checkConsistency);#Throw because cell#0 is not a polyhedron
1441 m.getNodalConnectivity().setIJ(2,0,4);
1442 m.checkConsistencyLight();
1443 m.checkConsistency();
1444 m.getNodalConnectivity().setIJ(7,0,-1);
1445 m.checkConsistencyLight();
1446 m.checkConsistency();#OK because we are in polyhedron connec
1447 m.getNodalConnectivity().setIJ(36,0,14);
1448 m.checkConsistencyLight();
1449 self.assertRaises(InterpKernelException,m.checkConsistency);#Throw because now cell 5 is a TETRA4 (14) so mismatch of number index and static type.
1452 def testUnPolyze2(self):
1453 m=MEDCouplingUMesh.New("jjj",3);
1454 coo=DataArrayDouble.New();
1461 m.insertNextCell(NORM_TETRA4,4,[0,1,2,3]);
1462 m.insertNextCell(NORM_TETRA4,4,[0,1,2,3]);
1463 m.finishInsertingCells();
1464 m2=MEDCouplingUMesh.MergeUMeshesOnSameCoords(4*[m]);
1465 m2.convertToPolyTypes([2]);
1467 self.assertEqual(NORM_TETRA4,m2.getTypeOfCell(2));
1468 self.assertEqual(40,m2.getNodalConnectivityArrayLen());
1469 temp2=m2.getNodeIdsOfCell(2);
1470 self.assertEqual(temp2,[0,1,2,3]);
1471 m2.checkConsistency();
1474 self.assertTrue(m3.isEqual(m2,1e-12));
1477 def testDACpyFrom1(self):
1478 d=DataArrayDouble.New();
1483 d.setInfoOnComponent(0,"X [m]");
1484 d.setInfoOnComponent(1,"Y [m]");
1485 d.setInfoOnComponent(2,"Z [m]");
1487 d1=DataArrayDouble.New();
1488 self.assertTrue(not d.isEqual(d1,1e-12));
1490 self.assertTrue(d.isEqual(d1,1e-12));
1492 self.assertTrue(d.isEqual(d1,1e-12));
1494 self.assertTrue(not d.isEqual(d1,1e-12));
1496 self.assertTrue(d.isEqual(d1,1e-12));
1498 d2=d.convertToIntArr();
1499 d4=DataArrayInt32.New();
1500 self.assertTrue(not d2.isEqual(d4));
1501 d4.deepCopyFrom(d2);
1502 self.assertTrue(d2.isEqual(d4));
1503 d4.deepCopyFrom(d2);
1504 self.assertTrue(d2.isEqual(d4));
1506 self.assertTrue(not d2.isEqual(d4));
1507 d4.deepCopyFrom(d2);
1508 self.assertTrue(d2.isEqual(d4));
1511 def testDAITransformWithIndArr1(self):
1513 tab2=[0,1,1,3,3,0,1,3,2,2,3,0]
1514 expected=[17,18,18,19,19,17,18,19,22,22,19,17]
1515 d=DataArrayInt.New();
1516 d.setValues(tab1,4,1);
1517 d1=DataArrayInt.New();
1518 d1.setValues(tab2,12,1);
1521 d1.transformWithIndArr(d);
1522 self.assertEqual(12,d1.getNumberOfTuples());
1523 self.assertEqual(1,d1.getNumberOfComponents());
1525 self.assertEqual(expected[i],d1.getIJ(i,0));
1529 d1.transformWithIndArr(tab1)
1530 self.assertEqual(12,d1.getNumberOfTuples());
1531 self.assertEqual(1,d1.getNumberOfComponents());
1533 self.assertEqual(expected[i],d1.getIJ(i,0));
1537 def testDAIBuildPermArrPerLevel1(self):
1538 arr=[2,0,1,1,0,1,2,0,1,1,0,0]
1539 expected1=[10,0,5,6,1,7,11,2,8,9,3,4]
1540 da=DataArrayInt.New();
1541 da.setValues(arr,12,1);
1542 da2=da.buildPermArrPerLevel();
1543 self.assertEqual(12,da2.getNumberOfTuples());
1544 self.assertEqual(1,da2.getNumberOfComponents());
1546 self.assertEqual(expected1[i],da2.getIJ(i,0));
1550 def testDAIOperations1(self):
1551 arr1=[-1,-2,4,7,3,2,6,6,4,3,0,1]
1552 da=DataArrayInt.New();
1553 da.setValues(arr1,4,3);
1554 da1=DataArrayInt.New();
1557 self.assertRaises(InterpKernelException,DataArrayInt.Add,da,da1);#not same number of tuples/Components
1559 da2=DataArrayInt.Add(da,da1);
1560 self.assertEqual(4,da2.getNumberOfTuples());
1561 self.assertEqual(3,da2.getNumberOfComponents());
1562 expected1=[1,1,8,12,9,9,14,15,14,14,12,14]
1564 self.assertEqual(expected1[i],da2.getIJ(0,i));
1566 da1.substractEqual(da);
1567 expected2=[3,5,0,-2,3,5,2,3,6,8,12,12]
1569 self.assertEqual(expected2[i],da1.getIJ(0,i));
1571 da1.rearrange(1); da1.iota(2); da1.rearrange(3);
1574 self.assertEqual(expected1[i],da1.getIJ(0,i));
1576 da1.rearrange(1); da1.iota(2); da1.rearrange(3);
1577 da2=DataArrayInt.Multiply(da,da1);
1578 self.assertEqual(4,da2.getNumberOfTuples());
1579 self.assertEqual(3,da2.getNumberOfComponents());
1580 expected3=[-2,-6,16,35,18,14,48,54,40,33,0,13]
1582 self.assertEqual(expected3[i],da2.getIJ(0,i));
1584 da.divideEqual(da1);
1585 self.assertEqual(4,da.getNumberOfTuples());
1586 self.assertEqual(3,da.getNumberOfComponents());
1587 expected4=[0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0]
1589 self.assertEqual(expected4[i],da.getIJ(0,i));
1591 da.setValues(arr1,4,3);
1592 da1.multiplyEqual(da);
1593 self.assertEqual(4,da1.getNumberOfTuples());
1594 self.assertEqual(3,da1.getNumberOfComponents());
1596 self.assertEqual(expected3[i],da1.getIJ(0,i));
1598 da1.rearrange(1); da1.iota(2); da1.rearrange(3);
1599 da2=DataArrayInt.Divide(da,da1);
1600 self.assertEqual(4,da2.getNumberOfTuples());
1601 self.assertEqual(3,da2.getNumberOfComponents());
1603 self.assertEqual(expected4[i],da2.getIJ(0,i));
1606 self.assertEqual(4,da1.getNumberOfTuples());
1607 self.assertEqual(3,da1.getNumberOfComponents());
1608 expected5=[160,107,80,64,53,45,40,35,32,29,26,24]
1610 self.assertEqual(expected5[i],da1.getIJ(0,i));
1612 da1.applyDivideBy(2);
1613 self.assertEqual(4,da1.getNumberOfTuples());
1614 self.assertEqual(3,da1.getNumberOfComponents());
1615 expected6=[80,53,40,32,26,22,20,17,16,14,13,12]
1617 self.assertEqual(expected6[i],da1.getIJ(0,i));
1619 expected7=[3,4,5,4,5,1,6,3,2,0,6,5]
1620 da1.applyModulus(7);
1622 self.assertEqual(expected7[i],da1.getIJ(0,i));
1625 expected8=[3,3,3,3,3,1,3,3,0,0,3,3]
1626 da1.applyRModulus(3);
1628 self.assertEqual(expected8[i],da1.getIJ(0,i));
1632 def testEmulateMEDMEMBDC1(self):
1633 m,m1=MEDCouplingDataForTest.buildPointe_1();
1634 m2,da1,da2,da3,da4,da5,da0=m.emulateMEDMEMBDC(m1)
1635 expected0=[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,36,37,32,33,34,35,38,39,40,41,42,43,44,45,46]
1636 expected1=[1,32,29,23,41,36]
1637 self.assertEqual(47,da0.getNumberOfTuples());
1638 self.assertEqual(1,da0.getNumberOfComponents());
1640 self.assertEqual(expected0[i],da0.getIJ(0,i));
1642 self.assertEqual(6,da5.getNumberOfTuples());
1643 self.assertEqual(1,da5.getNumberOfComponents());
1645 self.assertEqual(expected1[i],da5.getIJ(0,i));
1647 expected2=[0,1,2,3,4,0,5,6,7,4,8,9,1,7,10,11,12,13,14,5,15,16,17,8,18,19,20,10,21,22,23,2,13,24,25,21,16,26,27,12,19,28,29,15,22,30,31,18,36,26,28,30,24,37,32,33,34,35,38,36,39,40,41,42,37,38,43,44,45,46]
1648 self.assertEqual(70,da1.getNumberOfTuples());
1649 self.assertEqual(1,da1.getNumberOfComponents());
1651 self.assertEqual(expected2[i],da1.getIJ(0,i));
1653 expected3=[0,4,8,12,16,20,24,28,32,36,40,44,48,53,58,64,70]
1654 self.assertEqual(17,da2.getNumberOfTuples());
1655 self.assertEqual(1,da2.getNumberOfComponents());
1657 self.assertEqual(expected3[i],da2.getIJ(0,i));
1659 expected4=[0,2,4,6,7,9,11,12,14,16,17,19,20,22,24,25,27,29,30,32,34,35,37,39,40,42,43,45,46,48,49,51,52,53,54,55,56,58,60,62,63,64,65,66,67,68,69,70]
1660 #expected4=[0,2,4,6,7,9,11,12,14,16,17,19,20,22,24,25,27,29,30,32,34,35,37,39,40,42,43,45,46,48,49,51,52,54,56,57,58,59,60,62,63,64,65,66,67,68,69,70];
1661 self.assertEqual(48,da4.getNumberOfTuples());
1662 self.assertEqual(1,da4.getNumberOfComponents());
1664 self.assertEqual(expected4[i],da4.getIJ(0,i));
1666 expected5=[0,1,0,3,0,7,0,1,2,1,4,1,2,3,2,5,2,3,6,3,4,9,4,8,4,5,10,5,9,5,6,11,6,10,6,7,8,7,11,7,8,12,8,9,12,9,10,12,10,11,12,11,13,13,13,13,12,14,13,15,14,15,14,14,14,14,15,15,15,15]
1667 self.assertEqual(70,da3.getNumberOfTuples());
1668 self.assertEqual(1,da3.getNumberOfComponents());
1670 self.assertEqual(expected5[i],da3.getIJ(0,i));
1674 def testGetLevArrPerCellTypes1(self):
1675 m,m1=MEDCouplingDataForTest.buildPointe_1();
1676 m1,d0,d1,d2,d3=m.buildDescendingConnectivity();
1677 order=[NORM_TRI3,NORM_QUAD4];
1678 da0,da1=m1.getLevArrPerCellTypes(order);
1679 expected0=[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,1,0,0,0,0,1,1,1,1,1,1,1,1,1]
1680 expected1=[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,36,37,32,33,34,35,38,39,40,41,42,43,44,45,46]
1681 self.assertEqual(47,da0.getNumberOfTuples());
1682 self.assertEqual(1,da0.getNumberOfComponents());
1684 self.assertEqual(expected0[i],da0.getIJ(0,i));
1686 self.assertEqual(2,da1.getNumberOfTuples());
1687 self.assertEqual(1,da1.getNumberOfComponents());
1688 self.assertEqual(36,da1.getIJ(0,0));#36 TRI3
1689 self.assertEqual(11,da1.getIJ(1,0));#11 QUAD4
1691 da2=da0.buildPermArrPerLevel();
1693 self.assertEqual(47,da2.getNumberOfTuples());
1694 self.assertEqual(1,da2.getNumberOfComponents());
1696 self.assertEqual(expected1[i],da2.getIJ(0,i));
1700 def testSortCellsInMEDFileFrmt1(self):
1701 m,m1=MEDCouplingDataForTest.buildPointe_1();
1703 da=DataArrayInt.New()
1704 da.setValues([0,1,2,14,3,12,4,5,15,6,7,8,9,10,11,13],16,1)
1705 daa=da.invertArrayN2O2O2N(16)
1706 m.renumberCells(daa,False)
1707 da2=m.sortCellsInMEDFileFrmt()
1708 self.assertEqual(da2.getValues(),[0,1,2,14,3,12,4,5,15,6,7,8,9,10,11,13])
1709 self.assertTrue(m.isEqual(m2,1e-12))
1710 self.assertTrue(da.isEqual(da2))
1713 def testBuildPartAndReduceNodes1(self):
1714 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1716 m2,da=m.buildPartAndReduceNodes(arr);
1717 self.assertEqual(5,m2.getNumberOfNodes());
1718 self.assertEqual(2,m2.getNumberOfCells());
1719 f=m2.getMeasureField(True);
1720 self.assertAlmostEqual(0.125,f.getArray().getIJ(0,0),12);
1721 self.assertAlmostEqual(0.25,f.getArray().getIJ(1,0),12);
1723 arr2=DataArrayInt.New()
1724 arr2.setValues(arr,2,1)
1725 m2,da=m.buildPartAndReduceNodes(arr2);
1726 self.assertEqual(5,m2.getNumberOfNodes());
1727 self.assertEqual(2,m2.getNumberOfCells());
1728 f=m2.getMeasureField(True);
1729 self.assertAlmostEqual(0.125,f.getArray().getIJ(0,0),12);
1730 self.assertAlmostEqual(0.25,f.getArray().getIJ(1,0),12);
1733 def testDAITransformWithIndArrR1(self):
1735 tab2=[-1,-1,0,1,2,3,4,5,-1,-1,-1,-1]
1736 expected=[0,3,1,2,4,5]
1737 d=DataArrayInt.New();
1738 d.setValues(tab1,6,1);
1739 d1=DataArrayInt.New();
1740 d1.setValues(tab2,12,1);
1743 d3=d.transformWithIndArrR(d1);
1744 self.assertEqual(6,d3.getNumberOfTuples());
1745 self.assertEqual(1,d3.getNumberOfComponents());
1747 self.assertEqual(expected[i],d3.getIJ(i,0));
1751 d3=d.transformWithIndArrR(tab2)
1752 self.assertEqual(6,d3.getNumberOfTuples());
1753 self.assertEqual(1,d3.getNumberOfComponents());
1755 self.assertEqual(expected[i],d3.getIJ(i,0));
1759 def testDAISplitByValueRange1(self):
1760 val1=[6,5,0,3,2,7,8,1,4]
1762 d=DataArrayInt.New();
1763 d.setValues(val1,9,1);
1764 e,f,g=d.splitByValueRange(val2);
1765 self.assertEqual(9,e.getNumberOfTuples());
1766 self.assertEqual(1,e.getNumberOfComponents());
1767 self.assertEqual(9,f.getNumberOfTuples());
1768 self.assertEqual(1,f.getNumberOfComponents());
1769 self.assertEqual(2,g.getNumberOfTuples());
1770 self.assertEqual(1,g.getNumberOfComponents());
1772 expected1=[1,1,0,0,0,1,1,0,1]
1773 expected2=[2,1,0,3,2,3,4,1,0]
1775 self.assertEqual(expected1[i],e.getIJ(i,0));
1776 self.assertEqual(expected2[i],f.getIJ(i,0));
1778 self.assertEqual(0,g.getIJ(0,0));
1779 self.assertEqual(1,g.getIJ(1,0));
1782 self.assertRaises(InterpKernelException,d.splitByValueRange,val2);
1783 # non regression test in python wrapping
1784 rg=DataArrayInt([0,10,29,56,75,102,121,148,167,194,213,240,259,286,305,332,351,378,397,424,443,470,489,516])
1785 a,b,c=DataArrayInt([75]).splitByValueRange(rg)
1786 assert(a.isEqual(DataArrayInt([4])))
1787 assert(b.isEqual(DataArrayInt([0])))
1788 assert(c.isEqual(DataArrayInt([4])))
1791 def testUMeshSplitProfilePerType1(self):
1793 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1794 m.renumberCells(val0,False);
1797 d=DataArrayInt.New();
1798 d.setValues(val1,3,1);
1800 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1801 self.assertEqual(2,len(code));
1802 self.assertEqual(2,len(idsInPflPerType));
1803 expected1=[[3,1,0], [4,2,1]]
1804 self.assertEqual(expected1,code)
1805 self.assertEqual(2,len(idsInPflPerType));
1806 self.assertEqual(1,idsInPflPerType[0].getNumberOfTuples());
1807 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1808 self.assertEqual(2,idsInPflPerType[1].getNumberOfTuples());
1809 self.assertEqual(1,idsInPflPerType[1].getIJ(0,0));
1810 self.assertEqual(2,idsInPflPerType[1].getIJ(1,0));
1812 self.assertEqual(2,len(pfls));
1813 self.assertEqual("sup",pfls[0].getName())
1814 self.assertEqual(1,pfls[0].getNumberOfTuples());
1815 self.assertEqual(0,pfls[0].getIJ(0,0));
1816 self.assertEqual("sup",pfls[1].getName())
1817 self.assertEqual(2,pfls[1].getNumberOfTuples());
1818 self.assertEqual(0,pfls[1].getIJ(0,0));
1819 self.assertEqual(1,pfls[1].getIJ(1,0));
1822 d=DataArrayInt.New();
1823 d.setValues(val2,4,1);
1824 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1825 self.assertEqual(2,len(code));
1826 self.assertEqual(2,len(idsInPflPerType));
1827 expected2=[[3,1,0], [4,3,-1]]
1828 self.assertEqual(expected2,code);
1829 self.assertEqual(2,len(idsInPflPerType));
1830 self.assertEqual(1,idsInPflPerType[0].getNumberOfTuples());
1831 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1832 self.assertEqual(3,idsInPflPerType[1].getNumberOfTuples());
1833 self.assertEqual(1,idsInPflPerType[1].getIJ(0,0));
1834 self.assertEqual(2,idsInPflPerType[1].getIJ(1,0));
1835 self.assertEqual(3,idsInPflPerType[1].getIJ(2,0));
1837 self.assertEqual(1,len(pfls));
1838 self.assertEqual(1,pfls[0].getNumberOfTuples());
1839 self.assertEqual(0,pfls[0].getIJ(0,0));
1842 d=DataArrayInt.New();
1843 d.setValues(val3,3,1);
1844 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1845 self.assertEqual(2,len(code));
1846 self.assertEqual(2,len(idsInPflPerType));
1847 expected3=[[3,2,0], [4,1,1]]
1848 self.assertEqual(expected3,code);
1849 self.assertEqual(2,len(idsInPflPerType));
1850 self.assertEqual(2,idsInPflPerType[0].getNumberOfTuples());
1851 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1852 self.assertEqual(1,idsInPflPerType[0].getIJ(1,0));
1853 self.assertEqual(1,idsInPflPerType[1].getNumberOfTuples());
1854 self.assertEqual(2,idsInPflPerType[1].getIJ(0,0));
1856 self.assertEqual(2,len(pfls));
1857 self.assertEqual(2,pfls[0].getNumberOfTuples());
1858 self.assertEqual(1,pfls[0].getIJ(0,0));
1859 self.assertEqual(0,pfls[0].getIJ(1,0));
1860 self.assertEqual(0,pfls[1].getIJ(0,0));
1863 d=DataArrayInt.New();
1864 d.setValues(val4,2,1);
1865 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1866 self.assertEqual(1,len(code));
1867 self.assertEqual(1,len(idsInPflPerType));
1869 self.assertEqual(expected4,code);
1870 self.assertEqual(1,len(idsInPflPerType));
1871 self.assertEqual(2,idsInPflPerType[0].getNumberOfTuples());
1872 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1873 self.assertEqual(1,idsInPflPerType[0].getIJ(1,0));
1875 self.assertEqual(1,len(pfls));
1876 self.assertEqual(2,pfls[0].getNumberOfTuples());
1877 self.assertEqual(1,pfls[0].getIJ(0,0));
1878 self.assertEqual(2,pfls[0].getIJ(1,0));
1881 def testDAIBuildExplicitArrByRanges1(self):
1882 d=DataArrayInt.New();
1884 d.setValues(vals1,3,1);
1885 e=DataArrayInt.New();
1886 vals2=[0,3,6,10,14,20]
1887 e.setValues(vals2,6,1);
1889 f=d.buildExplicitArrByRanges(e);
1890 self.assertEqual(11,f.getNumberOfTuples());
1891 self.assertEqual(1,f.getNumberOfComponents());
1892 expected1=[0,1,2,6,7,8,9,10,11,12,13]
1894 self.assertEqual(expected1[i],f.getIJ(i,0));
1898 def testDAIComputeOffsets2(self):
1899 d=DataArrayInt.New();
1901 expected1=[0,3,8,9,11,11,19]
1902 d.setValues(vals1,6,1);
1903 d.computeOffsetsFull();
1904 self.assertEqual(7,d.getNumberOfTuples());
1905 self.assertEqual(1,d.getNumberOfComponents());
1907 self.assertEqual(expected1[i],d.getIJ(0,i));
1911 def testMergeField3(self):
1912 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1913 m.getCoords().setInfoOnComponent(0,"x [m]");
1914 m.getCoords().setInfoOnComponent(1,"z [km]");
1916 m.setDescription("desc");
1917 f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
1920 arr=DataArrayDouble.New();
1922 arr.setInfoOnComponent(0,"X [m]");
1923 arr.setInfoOnComponent(1,"YY [mm]");
1924 arr.fillWithValue(2.);
1927 f2=MEDCouplingFieldDouble.MergeFields([f1]);
1928 self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
1932 def testGetDistributionOfTypes1(self):
1933 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1935 self.assertRaises(InterpKernelException,m.getDistributionOfTypes);
1936 m.renumberCells(tab1,False);
1937 code=m.getDistributionOfTypes();
1938 self.assertEqual(2,len(code));
1939 self.assertEqual(3,code[0][0]);
1940 self.assertEqual(2,code[0][1]);
1941 self.assertEqual(-1,code[0][2]);
1942 self.assertEqual(4,code[1][0]);
1943 self.assertEqual(3,code[1][1]);
1944 self.assertEqual(-1,code[1][2]);
1947 def testNorm2_1(self):
1948 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1949 f=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
1952 d=DataArrayDouble.New();
1953 tab=[1.2,1.3,2.2,2.3,3.2,3.3,4.2,4.3,5.2,5.3]
1954 d.setValues(tab,5,2);
1956 f.checkConsistencyLight();
1958 self.assertAlmostEqual(11.209371079592289,f.norm2(),14);
1962 def testNormMax1(self):
1963 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1964 f=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
1967 d=DataArrayDouble.New(); d.alloc(0,2)
1968 res = d.normMaxPerComponent()
1969 self.assertAlmostEqual(-1.0, res[0],14)
1970 self.assertAlmostEqual(-1.0, res[1],14)
1972 tab=[2.3,-1.2,6.3,-7.8,2.9,7.7,2.1,0.,3.6,-7.6]
1973 d.setValues(tab,5,2);
1975 res = d.normMaxPerComponent()
1976 self.assertAlmostEqual(6.3, res[0],14)
1977 self.assertAlmostEqual(7.8, res[1],14)
1980 f.checkConsistencyLight();
1983 self.assertAlmostEqual(6.3,res[0],14);
1984 self.assertAlmostEqual(7.8,res[1],14);
1986 self.assertAlmostEqual(6.3,f.normMax(0),14);
1987 self.assertAlmostEqual(7.8,f.normMax(1),14);
1991 def testFindAndCorrectBadOriented3DExtrudedCells1(self):
1992 coords=[0.0011180339887498999, -0.0011755705045849499, 0.0, -0.0012331070204200001, -0.0011755705045849499, 0.0, -0.00067557050458494599, -0.00145964954842536, 0.0, -0.00050000000000000001, -0.00086602540378443902, 0.0, 0.00140211303259031, -0.00061803398874989504, 0.0, 0.00086602540378443902, -0.00050000000000000001, 0.0, 0.001, 0.0, 0.0, 0.00034561537182258202, 0.000269164072574575, 0.0, 0.0, 0.001, 0.0, -0.00050000000000000001, 0.00086602540378443902, 0.0, -0.000269164072574575, 0.00034561537182258202, 0.0, -0.001, 0.0, 0.0, -0.00086602540378443902, -0.00050000000000000001, 0.0, -0.00034561537182258202, -0.000269164072574575, 0.0, 0.0, -0.001, 0.0, 0.00050000000000000001, -0.00086602540378443902, 0.0, 0.000269164072574575, -0.00034561537182258202, 0.0, 0.0015, -6.01853107621011e-36, 0.0, 0.00056049747291484397, -0.00145964954842536, 0.0, 0.0011180339887498999, -0.0011755705045849499, 0.00050000000000000001, -0.0012331070204200001, -0.0011755705045849499, 0.00050000000000000001, -0.00067557050458494599, -0.00145964954842536, 0.00050000000000000001, -0.00050000000000000001, -0.00086602540378443902, 0.00050000000000000001, 0.00140211303259031, -0.00061803398874989504, 0.00050000000000000001, 0.00086602540378443902, -0.00050000000000000001, 0.00050000000000000001, 0.001, 0.0, 0.00050000000000000001, 0.00034561537182258202, 0.000269164072574575, 0.00050000000000000001, 0.0, 0.001, 0.00050000000000000001, -0.00050000000000000001, 0.00086602540378443902, 0.00050000000000000001, -0.000269164072574575, 0.00034561537182258202, 0.00050000000000000001, -0.001, 0.0, 0.00050000000000000001, -0.00086602540378443902, -0.00050000000000000001, 0.00050000000000000001, -0.00034561537182258202, -0.000269164072574575, 0.00050000000000000001, 0.0, -0.001, 0.00050000000000000001, 0.00050000000000000001, -0.00086602540378443902, 0.00050000000000000001, 0.000269164072574575, -0.00034561537182258202, 0.00050000000000000001, 0.0015, -6.01853107621011e-36, 0.00050000000000000001, 0.00056049747291484397, -0.00145964954842536, 0.00050000000000000001];
1993 conn=[2, 1, 3, 21, 20, 22, 4, 0, 5, 23, 19, 24, 8, 9, 10, 27, 28, 29, 11, 12, 13, 30, 31, 32, 0, 18, 15, 5, 19, 37, 34, 24, 6, 17, 4, 5, 25, 36, 23, 24, 3, 14, 16, 13, 22, 33, 35, 32, 13, 16, 7, 10, 32, 35, 26, 29]
1994 connExp=[16, 2, 1, 3, 21, 20, 22, 16, 4, 0, 5, 23, 19, 24, 16, 8, 10, 9, 27, 29, 28, 16, 11, 13, 12, 30, 32, 31, 18, 0, 18, 15, 5, 19, 37, 34, 24,18, 6, 17, 4, 5, 25, 36, 23, 24, 18, 3, 13, 16, 14, 22, 32, 35, 33, 18, 13, 10, 7, 16, 32, 29, 26, 35]
1995 invalidCells=[2,3,6,7]
1996 m=MEDCouplingUMesh.New("Example",3);
1997 coo=DataArrayDouble.New();
1998 coo.setValues(coords,38,3);
2001 m.insertNextCell(NORM_PENTA6,6,conn[0:6])
2002 m.insertNextCell(NORM_PENTA6,6,conn[6:12])
2003 m.insertNextCell(NORM_PENTA6,6,conn[12:18])
2004 m.insertNextCell(NORM_PENTA6,6,conn[18:24])
2005 m.insertNextCell(NORM_HEXA8,8,conn[24:32])
2006 m.insertNextCell(NORM_HEXA8,8,conn[32:40])
2007 m.insertNextCell(NORM_HEXA8,8,conn[40:48])
2008 m.insertNextCell(NORM_HEXA8,8,conn[48:56])
2009 m.finishInsertingCells();
2011 v=m.findAndCorrectBadOriented3DExtrudedCells();
2012 self.assertEqual(4,len(v));
2013 self.assertEqual(v.getValues(),invalidCells);
2014 self.assertEqual(connExp,m.getNodalConnectivity().getValues());
2015 self.assertTrue(m.findAndCorrectBadOriented3DExtrudedCells().empty())
2019 def testConvertExtrudedPolyhedra1(self):
2020 conn=[1,2,3,4, 5,6,7,8,9,10,11,12, 13,14,15,16, 17,18,19,20,21,22, 23,24,25,26,27,28, 29,30,31,32,33,34,35,36,37,38, 39,40,41,42,43,44,45,46, 47,48,49,50,51,52,53,54,55,56,57,58, 59,60,61,62,63,64,65,66,67,68,69,70,71,72]
2021 m=MEDCouplingUMesh.New("Example",3);
2022 coo=DataArrayDouble.New();
2024 coo.rearrange(1); coo.iota(0); coo.rearrange(3);
2027 m.insertNextCell(NORM_TETRA4,4,conn[0:4])
2028 m.insertNextCell(NORM_HEXA8,8,conn[4:12])
2029 m.insertNextCell(NORM_TETRA4,4,conn[12:16])
2030 m.insertNextCell(NORM_POLYHED,6,conn[16:22])
2031 m.insertNextCell(NORM_PENTA6,6,conn[22:28])
2032 m.insertNextCell(NORM_POLYHED,10,conn[28:38])
2033 m.insertNextCell(NORM_HEXA8,8,conn[38:46])
2034 m.insertNextCell(NORM_HEXGP12,12,conn[46:58])
2035 m.insertNextCell(NORM_POLYHED,14,conn[58:72])
2036 m.finishInsertingCells();
2038 m.convertExtrudedPolyhedra();
2039 da=m.getNodalConnectivity();
2040 dai=m.getNodalConnectivityIndex();
2041 self.assertEqual(10,dai.getNbOfElems());
2042 self.assertEqual(159,da.getNbOfElems());
2044 expected1=[14,1,2,3,4,18,5,6,7,8,9,10,11,12,14,13,14,15,16,31,17,18,19,-1,20,22,21,-1,17,20,21,18,-1,18,21,22,19,-1,19,22,20,17,16,23,24,25,26,27,28,31,29,30,31,32,33,-1,34,38,37,36,35,-1,29,34,35,30,-1,30,35,36,31,-1,31,36,37,32,-1,32,37,38,33,-1,33,38,34,29,18,39,40,41,42,43,44,45,46,22,47,48,49,50,51,52,53,54,55,56,57,58,31,59,60,61,62,63,64,65,-1,66,72,71,70,69,68,67,-1,59,66,67,60,-1,60,67,68,61,-1,61,68,69,62,-1,62,69,70,63,-1,63,70,71,64,-1,64,71,72,65,-1,65,72,66,59];
2045 expected2=[0,5,14,19,42,49,86,95,108,159]
2046 self.assertEqual(expected1,da.getValues());
2047 self.assertEqual(expected2,dai.getValues());
2048 m.checkConsistency()
2051 def testNonRegressionCopyTinyStrings(self):
2052 m=MEDCouplingDataForTest.build2DTargetMesh_1()
2053 f1=m.getMeasureField(True)
2054 f1.getArray().setInfoOnComponent(0,"P [N/m^2]")
2055 bary=m.computeCellCenterOfMass()
2056 f2=f1.buildNewTimeReprFromThis(NO_TIME,False)
2058 self.assertRaises(InterpKernelException,f1.copyTinyAttrFrom,f2)
2061 def testDaDSetPartOfValuesAdv1(self):
2062 tab1=[3.,4.,5., 13.,14.,15., 23.,24.,25., 33.,34.,35., 43.,44.,45., 53.,54.,55.]
2063 tab2=[6.,7.,8., 16.,17.,18., 26.,27.,28.]
2064 tab3=[4,1, 2,2, 3,0]
2065 a=DataArrayDouble.New();
2066 a.setValues(tab1,6,3);
2067 b=DataArrayDouble.New();
2068 b.setValues(tab2,3,3);
2069 c=DataArrayInt.New();
2070 c.setValues(tab3,3,2);
2072 a.setPartOfValuesAdv(b,c);
2073 expected1=[3.,4.,5., 13.,14.,15., 26.,27.,28., 6.,7.,8., 16.,17.,18., 53.,54.,55.]
2074 self.assertEqual(expected1,a.getValues());
2077 def testUMeshBuildSetInstanceFromThis1(self):
2078 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
2079 m2=m.buildSetInstanceFromThis(3);
2080 self.assertTrue(m.isEqual(m2,1e-12));
2082 m=MEDCouplingUMesh.New("toto",2);
2083 m2=m.buildSetInstanceFromThis(3);
2084 self.assertEqual(0,m2.getNumberOfNodes());
2085 self.assertEqual(0,m2.getNumberOfCells());
2088 def testUMeshMergeMeshesCVW1(self):
2089 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
2090 m2=MEDCouplingUMesh.New("toto",2);
2091 m3=MEDCouplingUMesh.MergeUMeshes([m,m2]);
2092 m3.setName(m.getName());
2093 self.assertTrue(m.isEqual(m3,1e-12));
2096 def testChangeUnderlyingMeshWithCMesh1(self):
2097 mesh=MEDCouplingCMesh.New();
2098 coordsX=DataArrayDouble.New();
2099 arrX=[ -1., 1., 2., 4. ]
2100 coordsX.setValues(arrX,4,1);
2101 coordsY=DataArrayDouble.New();
2102 arrY=[ -2., 2., 4., 8. ]
2103 coordsY.setValues(arrY,4,1);
2104 coordsZ=DataArrayDouble.New();
2105 arrZ=[ -3., 3., 6., 12. ]
2106 coordsZ.setValues(arrZ,4,1);
2107 mesh.setCoords(coordsX,coordsY,coordsZ);
2108 f=mesh.getMeasureField(True)
2109 mesh2=mesh.deepCopy()
2110 for myId in [0,1,2,10,11,12,20,21,22]:
2111 f=mesh.getMeasureField(True)
2112 f.changeUnderlyingMesh(mesh2,myId,1e-12);
2114 mesh2.setName("uuuu")
2115 for myId in [1,2,10,11,12,20,21,22]:
2116 f=mesh.getMeasureField(True)
2117 f.changeUnderlyingMesh(mesh2,myId,1e-12);
2121 def testDADFindCommonTuples1(self):
2122 da=DataArrayDouble.New();
2124 array1=[2.3,1.2,1.3,2.3,2.301,0.8]
2125 da.setValues(array1,6,1)
2126 c,cI=da.findCommonTuples(1e-2);
2129 self.assertEqual(3,c.getNbOfElems());
2130 self.assertEqual(2,cI.getNbOfElems());
2131 self.assertEqual(expected1,c.getValues())
2132 self.assertEqual(expected2,cI.getValues())
2133 c,cI=da.findCommonTuples(2e-1)
2134 expected3=[0,3,4,1,2]
2136 self.assertEqual(5,c.getNbOfElems());
2137 self.assertEqual(3,cI.getNbOfElems());
2138 self.assertEqual(expected3,c.getValues())
2139 self.assertEqual(expected4,cI.getValues())
2141 array2=[2.3,2.3,1.2,1.2,1.3,1.3,2.3,2.3,2.301,2.301,0.8,0.8]
2142 da.setValues(array2,6,2)
2143 c,cI=da.findCommonTuples(1e-2);
2144 self.assertEqual(3,c.getNbOfElems());
2145 self.assertEqual(2,cI.getNbOfElems());
2146 self.assertEqual(expected1,c.getValues())
2147 self.assertEqual(expected2,cI.getValues())
2148 c,cI=da.findCommonTuples(2e-1)
2149 self.assertEqual(5,c.getNbOfElems());
2150 self.assertEqual(3,cI.getNbOfElems());
2151 self.assertEqual(expected3,c.getValues())
2152 self.assertEqual(expected4,cI.getValues())
2154 array3=[2.3,2.3,2.3,1.2,1.2,1.2,1.3,1.3,1.3,2.3,2.3,2.3,2.301,2.301,2.301,0.8,0.8,0.8]
2155 da.setValues(array3,6,3)
2156 c,cI=da.findCommonTuples(1e-2);
2157 self.assertEqual(3,c.getNbOfElems());
2158 self.assertEqual(2,cI.getNbOfElems());
2159 self.assertEqual(expected1,c.getValues())
2160 self.assertEqual(expected2,cI.getValues())
2161 c,cI=da.findCommonTuples(2e-1)
2162 self.assertEqual(5,c.getNbOfElems());
2163 self.assertEqual(3,cI.getNbOfElems());
2164 self.assertEqual(expected3,c.getValues())
2165 self.assertEqual(expected4,cI.getValues())
2166 # nbOftuples=1, no common groups
2167 array11=[2.3,1.2,1.3,2.4,2.5,0.8]
2168 da.setValues(array11,6,1)
2169 c,cI=da.findCommonTuples(1e-2);
2170 self.assertEqual(0,c.getNbOfElems());
2171 self.assertEqual(1,cI.getNbOfElems());
2172 self.assertEqual([0],cI.getValues())
2175 da.setValues(array12,6,5) #bad NumberOfComponents
2176 self.assertRaises(InterpKernelException, da.findCommonTuples, 1e-2);
2179 def testDABack1(self):
2180 da=DataArrayDouble.New();
2181 array1=[2.3,1.2,1.3,2.3,2.301,0.8]
2182 da.setValues(array1,6,1);
2183 self.assertAlmostEqual(0.8,da.back(),14);
2185 self.assertRaises(InterpKernelException,da.back);
2187 self.assertRaises(InterpKernelException,da.back);
2189 da=DataArrayInt.New();
2191 da.setValues(array2,4,1);
2192 self.assertEqual(2,da.back());
2194 self.assertRaises(InterpKernelException,da.back);
2196 self.assertRaises(InterpKernelException,da.back);
2199 def testDADGetDifferentValues1(self):
2200 da=DataArrayDouble.New();
2201 array1=[2.3,1.2,1.3,2.3,2.301,0.8]
2202 da.setValues(array1,6,1)
2204 expected1=[2.301,1.2,1.3,0.8]
2205 dv=da.getDifferentValues(1e-2);
2206 self.assertEqual(4,dv.getNbOfElems());
2208 self.assertAlmostEqual(expected1[i],dv.getIJ(i,0),14);
2211 dv=da.getDifferentValues(2e-1);
2212 expected2=[2.301,1.3,0.8]
2213 self.assertEqual(3,dv.getNbOfElems());
2215 self.assertAlmostEqual(expected2[i],dv.getIJ(i,0),14);
2219 def testDAIBuildOld2NewArrayFromSurjectiveFormat2(self):
2222 a=DataArrayInt.New();
2223 a.setValues(arr,5,1);
2224 b=DataArrayInt.New();
2225 b.setValues(arrI,3,1);
2226 ret,newNbTuple=DataArrayInt.ConvertIndexArrayToO2N(10,a,b);
2227 expected=[0,1,2,0,3,4,5,4,6,4]
2228 self.assertEqual(10,ret.getNbOfElems());
2229 self.assertEqual(7,newNbTuple);
2230 self.assertEqual(1,ret.getNumberOfComponents());
2231 self.assertEqual(expected,ret.getValues());
2232 self.assertRaises(InterpKernelException,DataArrayInt.ConvertIndexArrayToO2N,9,a,b);
2235 def testDADIReverse1(self):
2237 a=DataArrayInt.New();
2238 a.setValues(arr,6,1);
2239 self.assertEqual(2,a.back());
2242 self.assertEqual(arr[5-i],a.getIJ(i,0));
2244 a.setValues(arr[:-1],5,1);
2247 self.assertEqual(arr[4-i],a.getIJ(i,0));
2250 arr2=[0.,3.,5.,7.,9.,2.]
2251 b=DataArrayDouble.New();
2252 b.setValues(arr2,6,1);
2255 self.assertAlmostEqual(arr2[5-i],b.getIJ(i,0),14);
2257 b.setValues(arr2[:5],5,1);
2258 self.assertAlmostEqual(9.,b.back(),14)
2261 self.assertAlmostEqual(arr2[4-i],b.getIJ(i,0),14);
2265 def testGetNodeIdsInUse1(self):
2266 m0=MEDCouplingDataForTest.build2DTargetMesh_1();
2268 m1=m0.buildPartOfMySelf(CellIds,True);
2269 arr,newNbOfNodes=m1.getNodeIdsInUse();
2270 expected=[-1,0,1,-1,2,3,-1,-1,-1]
2271 self.assertEqual(4,newNbOfNodes);
2272 self.assertEqual(9,arr.getNbOfElems());
2273 self.assertEqual(expected,arr.getValues());
2274 arr2=arr.invertArrayO2N2N2O(newNbOfNodes);
2275 self.assertEqual(4,arr2.getNbOfElems());
2277 self.assertEqual(expected2,arr2.getValues());
2280 def testBuildDescendingConnec2(self):
2281 mesh=MEDCouplingDataForTest.build2DTargetMesh_1();
2283 mesh2,desc,descIndx,revDesc,revDescIndx=mesh.buildDescendingConnectivity2();
2284 mesh2.checkConsistencyLight();
2285 self.assertEqual(1,mesh2.getMeshDimension());
2286 self.assertEqual(13,mesh2.getNumberOfCells());
2287 self.assertEqual(14,revDescIndx.getNbOfElems()); self.assertEqual(14,revDescIndx.getNumberOfTuples());
2288 self.assertEqual(6,descIndx.getNbOfElems()); self.assertEqual(6,descIndx.getNumberOfTuples());
2289 self.assertEqual(18,desc.getNbOfElems()); self.assertEqual(18,desc.getNumberOfTuples());
2290 self.assertEqual(18,revDesc.getNbOfElems()); self.assertEqual(18,revDesc.getNumberOfTuples());
2291 expected1=[1,2,3,4,-3,5,6, 7,8,-5,9,10,-2,11, 12,13,-7,-10]
2292 self.assertEqual(expected1,desc.getValues());
2293 expected2=[0,4,7,10,14,18]
2294 self.assertEqual(expected2,descIndx.getValues());
2295 expected3=[0,1,3,5,6,8,9,11,12,13,15,16,17,18]
2296 self.assertEqual(expected3,revDescIndx.getValues());
2297 expected4=[0, 0,3, 0,1, 0, 1,2, 1, 2,4, 2, 3, 3,4, 3, 4, 4]
2298 self.assertEqual(expected4,revDesc.getValues());
2299 conn=mesh2.getNodalConnectivity();
2300 connIndex=mesh2.getNodalConnectivityIndex();
2301 expected5=[0,3,6,9,12,15,18,21,24,27,30,33,36,39]
2302 self.assertEqual(expected5,connIndex.getValues());
2303 expected6=[1, 0, 3, 1, 3, 4, 1, 4, 1, 1, 1, 0, 1, 4, 2, 1, 2, 1, 1, 4, 5, 1, 5, 2, 1, 6, 7, 1, 7, 4, 1, 3, 6, 1, 7, 8, 1, 8, 5]
2304 self.assertEqual(expected6,conn.getValues());
2307 def testIntersect2DMeshesTmp1(self):
2308 m1c=MEDCouplingCMesh.New();
2309 coordsX=DataArrayDouble.New();
2310 arrX=[ -1., 1., 2., 4. ]
2311 coordsX.setValues(arrX,4,1);
2312 m1c.setCoordsAt(0,coordsX);
2313 coordsY=DataArrayDouble.New();
2314 arrY=[ -2., 2., 4., 8. ]
2315 coordsY.setValues(arrY,4,1);
2316 m1c.setCoordsAt(1,coordsY);
2317 m1=m1c.buildUnstructured()
2318 m1bis=m1.buildPartOfMySelf([3,4,5],False)
2320 m2=m2.buildPartOfMySelf([0,1,2],False)
2321 m2.translate([0.5,0.5])
2323 m3,d1,d2=MEDCouplingUMesh.Intersect2DMeshes(m1bis,m2,1e-10)
2324 expected1=[0,0,1,1,1,2,2,2]
2325 expected2=[0,-1,0,1,-1,1,2,-1]
2326 self.assertEqual(8,d1.getNumberOfTuples());
2327 self.assertEqual(8,d2.getNumberOfTuples());
2328 self.assertEqual(8,m3.getNumberOfCells());
2329 self.assertEqual(22,m3.getNumberOfNodes());
2330 self.assertEqual(2,m3.getSpaceDimension());
2331 self.assertEqual(expected1,d1.getValues());
2332 self.assertEqual(expected2,d2.getValues());
2333 expected3=[5,17,1,16,12,5,16,0,4,5,17,12,5,18,1,17,13,5,19,2,18,13,5,17,5,6,19,13,5,20,2,19,14,5,21,3,20,14,5,19,6,7,21,14]
2334 expected4=[0,5,12,17,22,28,33,38,44]
2335 expected5=[-1.0,2.0,1.0,2.0,2.0,2.0,4.0,2.0,-1.0,4.0,1.0,4.0,2.0,4.0,4.0,4.0,-0.5,-1.5,1.5,-1.5,2.5,-1.5,4.5,-1.5,-0.5,2.5,1.5,2.5,2.5,2.5,4.5,2.5,-0.5,2.0,1.0,2.5,1.5,2.0,2.0,2.5,2.5,2.0,4.0,2.5]
2336 self.assertEqual(44,m3.getNodalConnectivity().getNumberOfTuples());
2337 self.assertEqual(9,m3.getNodalConnectivityIndex().getNumberOfTuples());
2338 self.assertEqual(expected3,m3.getNodalConnectivity().getValues());
2339 self.assertEqual(expected4,m3.getNodalConnectivityIndex().getValues());
2341 self.assertAlmostEqual(expected5[i],m3.getCoords().getIJ(0,i),12);
2345 def testFindNodesOnLine1(self):
2346 mesh=MEDCouplingDataForTest.build2DTargetMesh_1();
2354 res=mesh.findNodesOnLine(pt,vec,1e-12);
2355 self.assertEqual(3,len(res));
2356 self.assertEqual(expected1,res.getValues());
2358 mesh.changeSpaceDimension(3);
2359 mesh.rotate(pt2,vec2,pi/4.);
2360 res=mesh.findNodesOnLine(pt3,vec3,1e-12);
2361 self.assertEqual(3,len(res));
2362 self.assertEqual(expected1,res.getValues());
2365 def testIntersect2DMeshesTmp2(self):
2366 m1c=MEDCouplingCMesh.New();
2367 coordsX1=DataArrayDouble.New();
2368 arrX1=[ 0., 1., 1.5, 2. ]
2369 coordsX1.setValues(arrX1,4,1);
2370 m1c.setCoordsAt(0,coordsX1);
2371 coordsY1=DataArrayDouble.New();
2372 arrY1=[ 0., 1.5, 3.]
2373 coordsY1.setValues(arrY1,3,1);
2374 m1c.setCoordsAt(1,coordsY1);
2375 m1=m1c.buildUnstructured();
2376 m2c=MEDCouplingCMesh.New();
2377 coordsX2=DataArrayDouble.New();
2378 arrX2=[ 0., 1., 2. ]
2379 coordsX2.setValues(arrX2,3,1);
2380 m2c.setCoordsAt(0,coordsX2);
2381 coordsY2=DataArrayDouble.New();
2383 coordsY2.setValues(arrY2,3,1);
2384 m2c.setCoordsAt(1,coordsY2);
2385 m2=m2c.buildUnstructured();
2387 m3,d1,d2=MEDCouplingUMesh.Intersect2DMeshes(m1,m2,1e-10)
2389 expected1=[0,0,1,1,2,2,3,4,5]
2390 expected2=[0,2,1,3,1,3,2,3,3]
2391 self.assertEqual(9,d1.getNumberOfTuples());
2392 self.assertEqual(9,d2.getNumberOfTuples());
2393 self.assertEqual(9,m3.getNumberOfCells());
2394 self.assertEqual(22,m3.getNumberOfNodes());
2395 self.assertEqual(2,m3.getSpaceDimension());
2396 self.assertEqual(expected1,d1.getValues());
2397 self.assertEqual(expected2,d2.getValues());
2398 expected3=[5,16,13,12,15,5,15,4,5,16,5,21,2,13,16,5,16,5,6,21,5,17,14,2,21,5,21,6,7,17,5,4,18,19,5,5,5,19,10,6,5,6,10,20,7]
2399 expected4=[0,5,10,15,20,25,30,35,40,45]
2400 expected5=[0.0,0.0,1.0,0.0,1.5,0.0,2.0,0.0,0.0,1.5,1.0,1.5,1.5,1.5,2.0,1.5,0.0,3.0,1.0,3.0,1.5,3.0,2.0,3.0,0.0,0.0,1.0,0.0,2.0,0.0,0.0,1.0,1.0,1.0,2.0,1.0,0.0,3.0,1.0,3.0,2.0,3.0,1.5,1.0]
2401 self.assertEqual(45,m3.getNodalConnectivity().getNumberOfTuples());
2402 self.assertEqual(10,m3.getNodalConnectivityIndex().getNumberOfTuples());
2403 self.assertEqual(expected3,m3.getNodalConnectivity().getValues());
2404 self.assertEqual(expected4,m3.getNodalConnectivityIndex().getValues());
2406 self.assertAlmostEqual(expected5[i],m3.getCoords().getIJ(0,i),12);
2410 def testBuildPartOfMySelfSafe1(self):
2411 mesh=MEDCouplingDataForTest.build2DTargetMesh_1()
2412 self.assertRaises(InterpKernelException,mesh.buildPartOfMySelf,[0,-1,4,2],True)
2413 self.assertRaises(InterpKernelException,mesh.buildPartOfMySelf,[0,4,5,4],True)
2416 def testIntersect2DMeshesTmp3(self):
2417 m1Coords=[0.,0.,1.,0.,1.5,0.,0.,1.,0.,1.5,-1.,0.,-1.5,0.,0.,-1,0.,-1.5,0.5,0.,1.25,0.,0.70710678118654757,0.70710678118654757,1.0606601717798214,1.0606601717798214,0.,0.5,0.,1.25,-0.70710678118654757,0.70710678118654757,-1.0606601717798214,1.0606601717798214,-0.5,0.,-1.25,0.,-0.70710678118654757,-0.70710678118654757,-1.0606601717798214,-1.0606601717798214,0.,-0.5,0.,-1.25,0.70710678118654757,-0.70710678118654757,1.0606601717798214,-1.0606601717798214];
2418 m1Conn=[0,3,1,13,11,9, 3,4,2,1,14,12,10,11, 5,3,0,15,13,17, 6,4,3,5,16,14,15,18, 5,0,7,17,21,19, 6,5,7,8,18,19,22,20, 0,1,7,9,23,21, 1,2,8,7,10,24,22,23];
2419 m1=MEDCouplingUMesh.New();
2420 m1.setMeshDimension(2);
2421 m1.allocateCells(8);
2422 m1.insertNextCell(NORM_TRI6,6,m1Conn[0:6]);
2423 m1.insertNextCell(NORM_QUAD8,8,m1Conn[6:14]);
2424 m1.insertNextCell(NORM_TRI6,6,m1Conn[14:20]);
2425 m1.insertNextCell(NORM_QUAD8,8,m1Conn[20:28]);
2426 m1.insertNextCell(NORM_TRI6,6,m1Conn[28:34]);
2427 m1.insertNextCell(NORM_QUAD8,8,m1Conn[34:42]);
2428 m1.insertNextCell(NORM_TRI6,6,m1Conn[42:48]);
2429 m1.insertNextCell(NORM_QUAD8,8,m1Conn[48:56]);
2430 m1.finishInsertingCells();
2431 myCoords1=DataArrayDouble.New();
2432 myCoords1.setValues(m1Coords,25,2);
2433 m1.setCoords(myCoords1);
2435 m2Coords=[0.,0.,1.1,0.,1.1,1.,0.,1.,1.7,0.,1.7,1.,-1.1,1.,-1.1,0.,-1.7,0.,-1.7,1.,-1.7,-1,-1.1,-1.,0.,-1.,1.1,-1,1.7,-1.]
2436 m2Conn=[0,3,2,1, 1,2,5,4, 7,6,3,0, 8,9,6,7, 7,0,12,11, 8,7,11,10, 0,1,13,12, 1,4,14,13]
2437 m2=MEDCouplingUMesh.New();
2438 m2.setMeshDimension(2);
2439 m2.allocateCells(8);
2441 m2.insertNextCell(NORM_QUAD4,4,m2Conn[4*i:4*(i+1)])
2443 m2.finishInsertingCells();
2444 myCoords2=DataArrayDouble.New();
2445 myCoords2.setValues(m2Coords,15,2);
2446 m2.setCoords(myCoords2);
2448 m3,d1,d2=MEDCouplingUMesh.Intersect2DMeshes(m1,m2,1e-10)
2451 expected1=[0,1,1,1,2,3,3,3,4,5,5,5,6,7,7,7]
2452 expected2=[0,0,1,-1,2,2,3,-1,4,4,5,-1,6,6,7,-1]
2453 self.assertEqual(16,d1.getNumberOfTuples());
2454 self.assertEqual(16,d2.getNumberOfTuples());
2455 self.assertEqual(16,m3.getNumberOfCells());
2456 self.assertEqual(104,m3.getNumberOfNodes());
2457 self.assertEqual(2,m3.getSpaceDimension());
2458 self.assertEqual(expected1,d1.getValues());
2459 self.assertEqual(expected2,d2.getValues());
2460 expected3=[6,28,1,25,44,45,46,8,26,1,28,27,47,48,49,50,8,40,2,26,27,51,52,53,54,8,28,4,40,27,55,56,57,58,6,28,25,5,59,60,61,8,28,5,32,31,62,63,64,65,8,32,6,41,31,66,67,68,69,8,41,4,28,31,70,71,72,73,6,25,37,5,74,75,76,8,32,5,37,36,77,78,79,80,8,42,6,32,36,81,82,83,84,8,37,8,42,36,85,86,87,88,6,1,37,25,89,90,91,8,37,1,26,38,92,93,94,95,8,26,2,43,38,96,97,98,99,8,43,8,37,38,100,101,102,103]
2461 expected4=[0,7,16,25,34,41,50,59,68,75,84,93,102,109,118,127,136]
2462 expected5=[0.,0.,1.,0.,1.5,0.,0.,1.,0.,1.5,-1.,0.,-1.5,0.,0.,-1.,0.,-1.5,0.5,0.,1.25,0.,0.7071067811865476,0.7071067811865476,1.0606601717798214,1.0606601717798214,0.,0.5,0.,1.25,-0.7071067811865476,0.7071067811865476,-1.0606601717798214,1.0606601717798214,-0.5,0.,-1.25,0.,-0.7071067811865476,-0.7071067811865476,-1.0606601717798214,-1.0606601717798214,0.,-0.5,0.,-1.25,0.7071067811865476,-0.7071067811865476,1.0606601717798214,-1.0606601717798214,0.,0.,1.1,0.,1.1,1.,0.,1.,1.7,0.,1.7,1.,-1.1,1.,-1.1,0.,-1.7,0.,-1.7,1.,-1.7,-1.,-1.1,-1.,0.,-1.,1.1,-1.,1.7,-1.,1.118033988749895,1.,-1.118033988749895,1.,-1.118033988749895,-1.,1.118033988749895,-1.,0.7071067811865477,0.7071067811865476,0.5,0.,0.,0.5,1.05,0.,0.7071067811865475,0.7071067811865477,0.55,1.,1.1,0.5,1.4012585384440737,0.535233134659635,1.3,0.,1.1,0.5,1.1090169943749475,1.,0.,1.25,0.6123724356957946,1.369306393762915,1.1090169943749475,1.,0.55,1.,0.,0.5,-0.5,0.,-0.7071067811865477,0.7071067811865476,-0.7071067811865475,0.7071067811865477,-1.05,0.,-1.1,0.5,-0.55,1.,-1.3,0.,-1.4012585384440737,0.5352331346596344,-1.1090169943749475,1.,-1.1,0.5,-0.6123724356957941,1.3693063937629155,0.,1.25,-0.55,1.,-1.1090169943749475,1.,0.,-0.5,-0.7071067811865475,-0.7071067811865477,-0.5,0.,-1.05,0.,-0.7071067811865478,-0.7071067811865475,-0.55,-1.,-1.1,-0.5,-1.4012585384440734,-0.5352331346596354,-1.3,0.,-1.1,-0.5,-1.1090169943749475,-1.,0.,-1.25,-0.6123724356957945,-1.369306393762915,-1.1090169943749475,-1.,-0.55,-1.,0.7071067811865475,-0.7071067811865477,0.,-0.5,0.5,0.,0.7071067811865477,-0.7071067811865475,1.05,0.,1.1,-0.5,0.55,-1.,1.3,0.,1.4012585384440737,-0.535233134659635,1.1090169943749475,-1.,1.1,-0.5,0.6123724356957946,-1.369306393762915,0.,-1.25,0.55,-1.,1.1090169943749475,-1.0]
2463 self.assertEqual(136,m3.getNodalConnectivity().getNumberOfTuples());
2464 self.assertEqual(17,m3.getNodalConnectivityIndex().getNumberOfTuples());
2465 self.assertEqual(expected3,m3.getNodalConnectivity().getValues());
2466 self.assertEqual(expected4,m3.getNodalConnectivityIndex().getValues());
2467 for i in range(208):
2468 self.assertAlmostEqual(expected5[i],m3.getCoords().getIJ(0,i),12);
2472 def testUMeshTessellate2D1(self):
2473 m1Coords=[0.,0.,1.,0.,1.5,0.,0.,1.,0.,1.5,-1.,0.,-1.5,0.,0.,-1,0.,-1.5,0.5,0.,1.25,0.,0.70710678118654757,0.70710678118654757,1.0606601717798214,1.0606601717798214,0.,0.5,0.,1.25,-0.70710678118654757,0.70710678118654757,-1.0606601717798214,1.0606601717798214,-0.5,0.,-1.25,0.,-0.70710678118654757,-0.70710678118654757,-1.0606601717798214,-1.0606601717798214,0.,-0.5,0.,-1.25,0.70710678118654757,-0.70710678118654757,1.0606601717798214,-1.0606601717798214];
2474 m1Conn=[0,3,1,13,11,9, 3,4,2,1,14,12,10,11, 5,3,0,15,13,17, 6,4,3,5,16,14,15,18, 5,0,7,17,21,19, 6,5,7,8,18,19,22,20, 0,1,7,9,23,21, 1,2,8,7,10,24,22,23];
2475 m1=MEDCouplingUMesh.New();
2476 m1.setMeshDimension(2);
2477 m1.allocateCells(8);
2478 m1.insertNextCell(NORM_TRI6,6,m1Conn[0:6]);
2479 m1.insertNextCell(NORM_QUAD8,8,m1Conn[6:14]);
2480 m1.insertNextCell(NORM_TRI6,6,m1Conn[14:20]);
2481 m1.insertNextCell(NORM_QUAD8,8,m1Conn[20:28]);
2482 m1.insertNextCell(NORM_TRI6,6,m1Conn[28:34]);
2483 m1.insertNextCell(NORM_QUAD8,8,m1Conn[34:42]);
2484 m1.insertNextCell(NORM_TRI6,6,m1Conn[42:48]);
2485 m1.insertNextCell(NORM_QUAD8,8,m1Conn[48:56]);
2486 m1.finishInsertingCells();
2487 myCoords1=DataArrayDouble.New();
2488 myCoords1.setValues(m1Coords,25,2);
2489 m1.setCoords(myCoords1);
2492 m11.tessellate2D(1.);
2493 self.assertTrue(m11.getCoords().isEqual(m11.getCoords(),1e-12));
2494 expected1=[5,0,3,11,1,5,3,4,12,2,1,11,5,5,15,3,0,5,6,16,4,3,15,5,5,5,0,7,19,5,6,5,19,7,8,20,5,0,1,23,7,5,1,2,24,8,7,23]
2495 expected2=[0,5,12,17,24,29,36,41,48]
2496 self.assertEqual(48,m11.getNodalConnectivity().getNumberOfTuples());
2497 self.assertEqual(9,m11.getNodalConnectivityIndex().getNumberOfTuples());
2498 self.assertEqual(expected1,m11.getNodalConnectivity().getValues());
2499 self.assertEqual(expected2,m11.getNodalConnectivityIndex().getValues());
2502 m12.tessellate2D(0.5);
2503 self.assertEqual(41,m12.getNumberOfNodes());
2504 expected3=[5,0,3,25,26,1,5,3,4,27,28,2,1,26,25,5,5,29,30,3,0,5,6,31,32,4,3,30,29,5,5,5,0,7,33,34,5,6,5,34,33,7,8,35,36,5,0,1,37,38,7,5,1,2,39,40,8,7,38,37]
2505 expected4=[0,6,15,21,30,36,45,51,60]
2506 expected5=[0.,0.,1.,0.,1.5,0.,0.,1.,0.,1.5,-1.,0.,-1.5,0.,0.,-1.,0.,-1.5,0.5,0.,1.25,0.,0.7071067811865476,0.7071067811865476,1.0606601717798214,1.0606601717798214,0.,0.5,0.,1.25,-0.7071067811865476,0.7071067811865476,-1.0606601717798214,1.0606601717798214,-0.5,0.,-1.25,0.,-0.7071067811865476,-0.7071067811865476,-1.0606601717798214,-1.0606601717798214,0.,-0.5,0.,-1.25,0.7071067811865476,-0.7071067811865476,1.0606601717798214,-1.0606601717798214,0.479425538604203,0.8775825618903728,0.8414709848078964,0.54030230586814,0.7191383079063044,1.3163738428355591,1.2622064772118446,0.8104534588022099,-0.877582561890373,0.4794255386042027,-0.5403023058681399,0.8414709848078964,-1.3163738428355596,0.7191383079063038,-0.8104534588022098,1.2622064772118446,-0.4794255386042031,-0.8775825618903728,-0.8414709848078965,-0.5403023058681399,-0.7191383079063045,-1.3163738428355591,-1.2622064772118449,-0.8104534588022098,0.8775825618903729,-0.47942553860420295,0.54030230586814,-0.8414709848078964,1.3163738428355594,-0.7191383079063043,0.8104534588022099,-1.2622064772118446]
2508 self.assertAlmostEqual(expected5[i],m12.getCoords().getIJ(0,i),12);
2510 self.assertEqual(60,m12.getNodalConnectivity().getNumberOfTuples());
2511 self.assertEqual(9,m12.getNodalConnectivityIndex().getNumberOfTuples());
2512 self.assertEqual(expected3,m12.getNodalConnectivity().getValues());
2513 self.assertEqual(expected4,m12.getNodalConnectivityIndex().getValues());
2516 def testUMeshTessellate2DCurve1(self):
2517 # A quarter of circle:
2518 mcoords = [0.4,0.0, 0.0,-0.4, 0.283,-0.283]
2521 m1 = MEDCouplingUMesh.New()
2522 m1.setMeshDimension(1)
2524 m1.insertNextCell(NORM_SEG3, mconnec)
2526 myCoords = DataArrayDouble.New(mcoords, 3, 2)
2527 m1.setCoords(myCoords)
2530 m2.tessellate2D(0.1)
2531 # If the following raises, the test will fail automatically:
2532 m2.checkConsistency(0.0) # eps param not used
2534 def testGetCellIdsCrossingPlane1(self):
2535 mesh3D,mesh2D=MEDCouplingDataForTest.build3DExtrudedUMesh_1();
2537 origin=[1.524,1.4552,1.74768]
2538 ids1=mesh3D.getCellIdsCrossingPlane(origin,vec,1e-10)
2539 self.assertEqual([1,3,4,7,9,10,13,15,16],ids1.getValues())
2541 ids2=mesh3D.getCellIdsCrossingPlane(origin,vec2,1e-10)
2542 self.assertEqual([6,7,8,9,10,11],ids2.getValues())
2545 def testBuildSlice3D1(self):
2546 mesh3D,mesh2D=MEDCouplingDataForTest.build3DExtrudedUMesh_1();
2547 vec1=[-0.07,1.,0.07]
2548 origin1=[1.524,1.4552,1.74768]
2549 slice1,ids=mesh3D.buildSlice3D(origin1,vec1,1e-10);
2550 expected1=[1,3,4,7,9,10,13,15,16]
2551 expected2=[5,42,41,40,43,44,5,42,46,45,41,5,44,43,40,47,48,5,49,42,44,50,5,49,51,46,42,5,50,44,48,52,5,53,49,50,54,5,53,55,51,49,5,54,50,52,56]
2552 expected3=[0,6,11,17,22,27,32,37,42,47]
2553 expected4=[1.,1.,0.,1.,1.25,0.,1.,1.5,0.,2.,1.,0.,1.,2.,0.,0.,2.,0.,3.,1.,0.,3.,2.,0.,0.,1.,0.,2.,2.,0.,1.,1.,1.,1.,1.25,1.,1.,1.5,1.,2.,1.,1.,1.,2.,1.,0.,2.,1.,3.,1.,1.,3.,2.,1.,0.,1.,1.,2.,2.,1.,1.,1.,2.,1.,1.25,2.,1.,1.5,2.,2.,1.,2.,1.,2.,2.,0.,2.,2.,3.,1.,2.,3.,2.,2.,0.,1.,2.,2.,2.,2.,1.,1.,3.,1.,1.25,3.,1.,1.5,3.,2.,1.,3.,1.,2.,3.,0.,2.,3.,3.,1.,3.,3.,2.,3.,0.,1.,3.,2.,2.,3.,1.,1.5408576,0.,2.,1.6108576000000001,0.,2.,1.5408576,1.,1.,1.5,0.5836800000000008,1.,1.4708576,1.,3.,1.6808576,0.,3.,1.6108576000000001,1.,0.,1.4708576,0.,0.,1.4008576,1.,2.,1.4708576,2.,1.,1.4008576000000001,2.,3.,1.5408575999999998,2.,0.,1.3308575999999999,2.,2.,1.4008576,3.,1.,1.3308576,3.,3.,1.4708576,3.,0.,1.2608576,3.]
2554 self.assertEqual(2,slice1.getMeshDimension());
2555 self.assertEqual(3,slice1.getSpaceDimension());
2556 self.assertEqual(57,slice1.getNumberOfNodes());
2557 self.assertEqual(9,slice1.getNumberOfCells());
2558 self.assertEqual(9,ids.getNumberOfTuples());
2559 self.assertEqual(47,slice1.getNodalConnectivity().getNumberOfTuples());
2560 self.assertEqual(10,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2561 self.assertEqual(expected1,ids.getValues());
2562 self.assertEqual(expected2,slice1.getNodalConnectivity().getValues());
2563 self.assertEqual(expected3,slice1.getNodalConnectivityIndex().getValues());
2564 for i in range(171):
2565 self.assertAlmostEqual(expected4[i],slice1.getCoords().getIJ(0,i),12);
2567 # 2nd slice based on already existing nodes of mesh3D.
2570 slice1,ids=mesh3D.buildSlice3D(origin2,vec2,1e-10);
2571 expected5=[5,50,10,4,51,5,50,52,7,10,5,51,4,5,53,5,54,50,51,55,56,5,54,57,52,50,5,56,55,51,53,58,5,38,59,56,54,43,5,54,57,46,43,5,38,59,56,58,48]
2572 expected6=[0,5,10,15,21,26,32,38,43,49]
2573 expected7=[1.,1.,0.,1.,1.25,0.,1.,1.5,0.,2.,1.,0.,1.,2.,0.,0.,2.,0.,3.,1.,0.,3.,2.,0.,0.,1.,0.,1.,3.,0.,2.,2.,0.,2.,3.,0.,1.,1.,1.,1.,1.25,1.,1.,1.5,1.,2.,1.,1.,1.,2.,1.,0.,2.,1.,3.,1.,1.,3.,2.,1.,0.,1.,1.,1.,3.,1.,2.,2.,1.,2.,3.,1.,0.,0.,2.,1.,1.,2.,1.,1.25,2.,1.,0.,2.,1.,1.5,2.,2.,0.,2.,2.,1.,2.,1.,2.,2.,0.,2.,2.,3.,1.,2.,3.,2.,2.,0.,1.,2.,2.,2.,2.,0.,0.,3.,1.,1.,3.,1.,1.25,3.,1.,0.,3.,1.,1.5,3.,2.,0.,3.,2.,1.,3.,1.,2.,3.,0.,2.,3.,3.,1.,3.,3.,2.,3.,0.,1.,3.,2.,2.,3.,2.,1.6666666666666667,1.,1.,1.6666666666666667,1.,3.,1.6666666666666667,1.,0.,1.6666666666666667,1.,2.,1.3333333333333335,2.,1.,1.5,1.5,1.,1.3333333333333333,2.,3.,1.3333333333333335,2.,0.,1.3333333333333335,2.,1.,1.25,2.25]
2574 self.assertEqual(2,slice1.getMeshDimension());
2575 self.assertEqual(3,slice1.getSpaceDimension());
2576 self.assertEqual(60,slice1.getNumberOfNodes());
2577 self.assertEqual(9,slice1.getNumberOfCells());
2578 self.assertEqual(9,ids.getNumberOfTuples());
2579 self.assertEqual(49,slice1.getNodalConnectivity().getNumberOfTuples());
2580 self.assertEqual(10,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2581 self.assertEqual(expected1,ids.getValues());
2582 self.assertEqual(expected5,slice1.getNodalConnectivity().getValues());
2583 self.assertEqual(expected6,slice1.getNodalConnectivityIndex().getValues());
2584 for i in range(180):
2585 self.assertAlmostEqual(expected7[i],slice1.getCoords().getIJ(0,i),12);
2587 # 3rd slice based on shared face of mesh3D.
2590 slice1,ids=mesh3D.buildSlice3D(origin3,vec3,1e-10);
2591 expected8=[6,7,8,9,10,11,12,13,14,15,16,17]
2592 expected9=[5,15,26,16,18,5,16,21,28,22,19,17,5,18,20,21,16,5,21,24,25,28,5,26,16,17,19,22,23,5,22,27,29,28,5,15,26,16,18,5,16,21,28,22,19,17,5,18,20,21,16,5,21,24,25,28,5,26,16,17,19,22,23,5,22,27,29,28]
2593 expected10=[0,5,12,17,22,29,34,39,46,51,56,63,68]
2594 expected11=[0.,0.,1.,1.,1.,1.,1.,1.25,1.,1.,0.,1.,1.,1.5,1.,2.,0.,1.,2.,1.,1.,1.,2.,1.,0.,2.,1.,3.,1.,1.,3.,2.,1.,0.,1.,1.,1.,3.,1.,2.,2.,1.,2.,3.,1.,0.,0.,2.,1.,1.,2.,1.,1.25,2.,1.,0.,2.,1.,1.5,2.,2.,0.,2.,2.,1.,2.,1.,2.,2.,0.,2.,2.,3.,1.,2.,3.,2.,2.,0.,1.,2.,1.,3.,2.,2.,2.,2.,2.,3.,2.,0.,0.,3.,1.,1.,3.,1.,1.25,3.,1.,0.,3.,1.,1.5,3.,2.,0.,3.,2.,1.,3.,1.,2.,3.,0.,2.,3.,3.,1.,3.,3.,2.,3.,0.,1.,3.,1.,3.,3.,2.,2.,3.,2.,3.,3.]
2595 self.assertEqual(2,slice1.getMeshDimension());
2596 self.assertEqual(3,slice1.getSpaceDimension());
2597 self.assertEqual(45,slice1.getNumberOfNodes());
2598 self.assertEqual(12,slice1.getNumberOfCells());
2599 self.assertEqual(12,ids.getNumberOfTuples());
2600 self.assertEqual(68,slice1.getNodalConnectivity().getNumberOfTuples());
2601 self.assertEqual(13,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2602 self.assertEqual(expected8,ids.getValues());
2603 self.assertEqual(expected9,slice1.getNodalConnectivity().getValues());
2604 self.assertEqual(expected10,slice1.getNodalConnectivityIndex().getValues());
2605 for i in range(135):
2606 self.assertAlmostEqual(expected11[i],slice1.getCoords().getIJ(0,i),12);
2609 coo = DataArrayDouble([1.8686305176182501,0.85678785370447097,7.4, 1.8686305176182501,0.85678785370447097,7.5, 1.8663408654534299,0.82910039403216995,7.4, 1.8663408654534299,0.82910039403216995,7.5, 1.8370211426271501,0.83286926189135702,7.4, 1.8370211426271501,0.83286926189135702,7.5, 1.84595839792064,0.86012397150595199,7.4, 1.84595839792064,0.86012397150595199,7.5], 8,3)
2610 conn = DataArrayInt([18,0,2,4,6,1,3,5,7])
2611 connI = DataArrayInt([0,9])
2613 mesh3D_2=MEDCouplingUMesh.New();
2614 mesh3D_2.setName("3DMesh_2");
2615 mesh3D_2.setMeshDimension(3);
2616 mesh3D_2.setCoords(coo);
2617 mesh3D_2.setConnectivity(conn,connI,True);
2620 expected13=[5,5,9,8,4]
2622 expected15=[1.8686305176182501,0.85678785370447097,7.4, 1.8686305176182501,0.85678785370447097,7.5, 1.8663408654534299,0.82910039403216995,7.4, 1.8663408654534299,0.82910039403216995,7.5, 1.8370211426271501,0.83286926189135702,7.4, 1.8370211426271501,0.83286926189135702,7.5, 1.84595839792064,0.86012397150595199, 7.4, 1.84595839792064,0.86012397150595199,7.5, 1.8666525378798646,0.83286927118760312,7.4, 1.8666525378798646,0.83286927118760312,7.5]
2624 y_cut = 0.8328692711876031
2625 ori, vec = [0.0, y_cut, 0.0], [0.0,1.0,0.0]
2626 slice1, ids = mesh3D_2.buildSlice3D(DataArrayDouble(ori,1,3), DataArrayDouble(vec,1,3), 1.0e-8)
2627 self.assertEqual(2,slice1.getMeshDimension());
2628 self.assertEqual(3,slice1.getSpaceDimension());
2629 self.assertEqual(10,slice1.getNumberOfNodes());
2630 self.assertEqual(1,slice1.getNumberOfCells());
2631 self.assertEqual(1,ids.getNumberOfTuples());
2632 self.assertEqual(5,slice1.getNodalConnectivity().getNumberOfTuples());
2633 self.assertEqual(2,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2634 self.assertEqual(expected12,ids.getValues());
2635 self.assertEqual(expected13,slice1.getNodalConnectivity().getValues());
2636 self.assertEqual(expected14,slice1.getNodalConnectivityIndex().getValues());
2638 self.assertAlmostEqual(expected15[i],slice1.getCoords().getIJ(0,i),12);
2641 def testBuildSlice3DSurf1(self):
2642 mesh3D,mesh2D=MEDCouplingDataForTest.build3DExtrudedUMesh_1();
2643 mesh2D=mesh3D.buildDescendingConnectivity()[0];
2644 vec1=[-0.07,1.,0.07]
2645 origin1=[1.524,1.4552,1.74768]
2646 slice1,ids=mesh2D.buildSlice3DSurf(origin1,vec1,1e-10);
2647 expected1=[6,8,10,11,13,18,19,21,23,25,26,38,41,43,47,49,52,53,64,67,69,73,75,78,79]
2648 expected2=[1,40,41,1,42,41,1,40,43,1,44,43,1,42,44,1,45,41,1,42,46,1,46,45,1,47,40,1,47,48,1,44,48,1,49,42,1,44,50,1,49,50,1,49,51,1,51,46,1,48,52,1,50,52,1,53,49,1,50,54,1,53,54,1,53,55,1,55,51,1,52,56,1,54,56]
2649 expected3=[0,3,6,9,12,15,18,21,24,27,30,33,36,39,42,45,48,51,54,57,60,63,66,69,72,75];
2650 expected4=[1.,1.,0.,1.,1.25,0.,1.,1.5,0.,2.,1.,0.,1.,2.,0.,0.,2.,0.,3.,1.,0.,3.,2.,0.,0.,1.,0.,2.,2.,0.,1.,1.,1.,1.,1.25,1.,1.,1.5,1.,2.,1.,1.,1.,2.,1.,0.,2.,1.,3.,1.,1.,3.,2.,1.,0.,1.,1.,2.,2.,1.,1.,1.,2.,1.,1.25,2.,1.,1.5,2.,2.,1.,2.,1.,2.,2.,0.,2.,2.,3.,1.,2.,3.,2.,2.,0.,1.,2.,2.,2.,2.,1.,1.,3.,1.,1.25,3.,1.,1.5,3.,2.,1.,3.,1.,2.,3.,0.,2.,3.,3.,1.,3.,3.,2.,3.,0.,1.,3.,2.,2.,3.,1.,1.5408576,0.,2.,1.6108576000000001,0.,2.,1.5408576,1.,1.,1.5,0.5836800000000008,1.,1.4708576,1.,3.,1.6808576,0.,3.,1.6108576000000001,1.,0.,1.4708576,0.,0.,1.4008576,1.,2.,1.4708576,2.,1.,1.4008576000000001,2.,3.,1.5408575999999998,2.,0.,1.3308575999999999,2.,2.,1.4008576,3.,1.,1.3308576,3.,3.,1.4708576,3.,0.,1.2608576,3.]
2651 self.assertEqual(1,slice1.getMeshDimension());
2652 self.assertEqual(3,slice1.getSpaceDimension());
2653 self.assertEqual(57,slice1.getNumberOfNodes());
2654 self.assertEqual(25,slice1.getNumberOfCells());
2655 self.assertEqual(25,ids.getNumberOfTuples());
2656 self.assertEqual(75,slice1.getNodalConnectivity().getNumberOfTuples());
2657 self.assertEqual(26,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2658 self.assertEqual(expected1,ids.getValues());
2659 self.assertEqual(expected2,slice1.getNodalConnectivity().getValues());
2660 self.assertEqual(expected3,slice1.getNodalConnectivityIndex().getValues());
2661 for i in range(171):
2662 self.assertAlmostEqual(expected4[i],slice1.getCoords().getIJ(0,i),12);
2667 slice1,ids=mesh2D.buildSlice3DSurf(origin2,vec2,1e-10);
2668 expected5=[32,32,32,32,33,34,35,36,37,38,39,40,41,42,43,43,43,43,43,43,44,44,44,44,45,46,47,47,47,47,48,49,50,51,52,53,53,53,53,53,53,54,54,54,54,55,56,57,59,60,61,62,63,64,65,66,67,68,71,72,74,75,76,77,78,81,82,83]
2669 expected6=[1,15,18,1,18,16,1,16,26,1,26,15,1,26,15,1,16,26,1,18,16,1,15,18,1,16,21,1,21,28,1,22,28,1,19,22,1,17,19,1,16,17,1,16,21,1,21,28,1,28,22,1,22,19,1,19,17,1,17,16,1,16,18,1,18,20,1,20,21,1,21,16,1,20,21,1,18,20,1,28,21,1,21,24,1,24,25,1,25,28,1,25,28,1,24,25,1,21,24,1,23,22,1,26,23,1,26,16,1,16,17,1,17,19,1,19,22,1,22,23,1,23,26,1,22,28,1,28,29,1,29,27,1,27,22,1,27,22,1,29,27,1,28,29,1,26,15,1,16,26,1,18,16,1,15,18,1,16,21,1,21,28,1,22,28,1,19,22,1,17,19,1,16,17,1,20,21,1,18,20,1,25,28,1,24,25,1,21,24,1,23,22,1,26,23,1,27,22,1,29,27,1,28,29]
2670 expected7=[0,3,6,9,12,15,18,21,24,27,30,33,36,39,42,45,48,51,54,57,60,63,66,69,72,75,78,81,84,87,90,93,96,99,102,105,108,111,114,117,120,123,126,129,132,135,138,141,144,147,150,153,156,159,162,165,168,171,174,177,180,183,186,189,192,195,198,201,204];
2671 expected8=[0.,0.,1.,1.,1.,1.,1.,1.25, 1.,1.,0.,1.,1.,1.5, 1.,2.,0.,1.,2.,1.,1.,1.,2.,1.,0.,2.,1.,3.,1.,1.,3.,2.,1.,0.,1.,1.,1.,3.,1.,2.,2.,1.,2.,3.,1.,0.,0.,2.,1.,1.,2.,1.,1.25, 2.,1.,0.,2.,1.,1.5, 2.,2.,0.,2.,2.,1.,2.,1.,2.,2.,0.,2.,2.,3.,1.,2.,3.,2.,2.,0.,1.,2.,1.,3.,2.,2.,2.,2.,2.,3.,2.,0.,0.,3.,1.,1.,3.,1.,1.25, 3.,1.,0.,3.,1.,1.5, 3.,2.,0.,3.,2.,1.,3.,1.,2.,3.,0.,2.,3.,3.,1.,3.,3.,2.,3.,0.,1.,3.,1.,3.,3.,2.,2.,3.,2.,3.,3.]
2672 self.assertEqual(1,slice1.getMeshDimension());
2673 self.assertEqual(3,slice1.getSpaceDimension());
2674 self.assertEqual(45,slice1.getNumberOfNodes());
2675 self.assertEqual(68,slice1.getNumberOfCells());
2676 self.assertEqual(68,ids.getNumberOfTuples());
2677 self.assertEqual(204,slice1.getNodalConnectivity().getNumberOfTuples());
2678 self.assertEqual(69,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2679 self.assertEqual(expected5,ids.getValues());
2680 self.assertEqual(expected6,slice1.getNodalConnectivity().getValues());
2681 self.assertEqual(expected7,slice1.getNodalConnectivityIndex().getValues());
2682 for i in range(135):
2683 self.assertAlmostEqual(expected8[i],slice1.getCoords().getIJ(0,i),12);
2687 def testDataArrayDoubleAdvSetting1(self):
2688 data1=[1.,11.,2.,12.,3.,13.,4.,14.,5.,15.,6.,16.,7.,17.]
2689 data2=[8.,38.,9.,39.,0.,30.,11.,41.,12.,42.]
2690 compsCpp=["comp1","comp2"]
2691 da=DataArrayDouble.New();
2692 da.setInfoAndChangeNbOfCompo(compsCpp);
2695 compsCpp=compsCpp[:-1]
2696 self.assertRaises(InterpKernelException,da.setInfoAndChangeNbOfCompo,compsCpp);
2697 da.setValues(data1,7,2)
2699 p=[(0,3),(3,5),(5,7)]
2700 tmp=da.selectByTupleRanges(p);
2701 self.assertTrue(tmp.isEqual(da,1e-14));
2702 p=[(0,2),(3,4),(5,7)]
2703 tmp=da.selectByTupleRanges(p);
2704 expected1=[1.,11.,2.,12.,4.,14.,6.,16.,7.,17.]
2705 self.assertEqual(5,tmp.getNumberOfTuples());
2706 self.assertEqual(2,tmp.getNumberOfComponents());
2708 self.assertAlmostEqual(expected1[i],tmp.getIJ(0,i),14);
2710 p=[(0,2),(0,2),(5,6)]
2711 tmp=da.selectByTupleRanges(p);
2712 expected2=[1.,11.,2.,12.,1.,11.,2.,12.,6.,16.]
2713 self.assertEqual(5,tmp.getNumberOfTuples());
2714 self.assertEqual(2,tmp.getNumberOfComponents());
2716 self.assertAlmostEqual(expected2[i],tmp.getIJ(0,i),14);
2718 p=[(0,2),(-1,2),(5,6)]
2719 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2720 p=[(0,2),(0,2),(5,8)]
2721 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2723 da2=DataArrayDouble.New();
2724 da2.setValues(data2,5,2);
2727 dac.setContigPartOfSelectedValuesSlice(1,da2,2,4,1);
2728 expected3=[1.,11.,0.,30.,11.,41.,4.,14.,5.,15.,6.,16.,7.,17.]
2730 self.assertAlmostEqual(expected3[i],dac.getIJ(0,i),14);
2734 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,0,5,1);
2735 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,0,da2,4,6,1);
2736 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,5,0,1);
2737 dac.setContigPartOfSelectedValuesSlice(3,da2,1,5,1);
2738 expected4=[1.,11.,2.,12.,3.,13.,9.,39.,0.,30.,11.,41.,12.,42.]
2740 self.assertAlmostEqual(expected4[i],dac.getIJ(0,i),14);
2743 ids=DataArrayInt.New();
2746 ids.setIJ(0,0,2); ids.setIJ(1,0,0); ids.setIJ(2,0,4);
2747 dac.setContigPartOfSelectedValues(2,da2,ids);
2748 expected5=[1.,11.,2.,12.,0.,30.,8.,38.,12.,42.,6.,16.,7.,17.]
2750 self.assertAlmostEqual(expected5[i],dac.getIJ(0,i),14);
2754 ids.setIJ(0,0,2); ids.setIJ(1,0,5); ids.setIJ(2,0,4);
2755 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2756 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,-1);
2757 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2758 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2759 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,5,da2,ids);
2761 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2763 dac.setContigPartOfSelectedValues(4,da2,ids);
2764 expected6=[1.,11.,2.,12.,3.,13.,4.,14.,0.,30.,0.,30.,9.,39.]
2766 self.assertAlmostEqual(expected6[i],dac.getIJ(0,i),14);
2770 def testDataArrayIntAdvSetting1(self):
2771 data1=[1,11,2,12,3,13,4,14,5,15,6,16,7,17]
2772 data2=[8,38,9,39,0,30,11,41,12,42]
2773 compsCpp=["comp1","comp2"]
2774 da=DataArrayInt.New();
2775 da.setInfoAndChangeNbOfCompo(compsCpp);
2778 compsCpp=compsCpp[:-1]
2779 self.assertRaises(InterpKernelException,da.setInfoAndChangeNbOfCompo,compsCpp);
2780 da.setValues(data1,7,2)
2782 p=[(0,3),(3,5),(5,7)]
2783 tmp=da.selectByTupleRanges(p);
2784 self.assertTrue(tmp.isEqual(da));
2785 p=[(0,2),(3,4),(5,7)]
2786 tmp=da.selectByTupleRanges(p);
2787 expected1=[1,11,2,12,4,14,6,16,7,17]
2788 self.assertEqual(5,tmp.getNumberOfTuples());
2789 self.assertEqual(2,tmp.getNumberOfComponents());
2791 self.assertEqual(expected1[i],tmp.getIJ(0,i));
2793 p=[(0,2),(0,2),(5,6)]
2794 tmp=da.selectByTupleRanges(p);
2795 expected2=[1,11,2,12,1,11,2,12,6,16]
2796 self.assertEqual(5,tmp.getNumberOfTuples());
2797 self.assertEqual(2,tmp.getNumberOfComponents());
2799 self.assertEqual(expected2[i],tmp.getIJ(0,i));
2801 p=[(0,2),(-1,2),(5,6)]
2802 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2803 p=[(0,2),(0,2),(5,8)]
2804 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2806 da2=DataArrayInt.New();
2807 da2.setValues(data2,5,2);
2810 dac.setContigPartOfSelectedValuesSlice(1,da2,2,4,1);
2811 expected3=[1,11,0,30,11,41,4,14,5,15,6,16,7,17]
2813 self.assertEqual(expected3[i],dac.getIJ(0,i));
2817 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,0,5,1);
2818 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,0,da2,4,6,1);
2819 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,5,0,1);
2820 dac.setContigPartOfSelectedValuesSlice(3,da2,1,5,1);
2821 expected4=[1,11,2,12,3,13,9,39,0,30,11,41,12,42]
2823 self.assertEqual(expected4[i],dac.getIJ(0,i));
2826 ids=DataArrayInt.New();
2829 ids.setIJ(0,0,2); ids.setIJ(1,0,0); ids.setIJ(2,0,4);
2830 dac.setContigPartOfSelectedValues(2,da2,ids);
2831 expected5=[1,11,2,12,0,30,8,38,12,42,6,16,7,17]
2833 self.assertEqual(expected5[i],dac.getIJ(0,i));
2837 ids.setIJ(0,0,2); ids.setIJ(1,0,5); ids.setIJ(2,0,4);
2838 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2839 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,-1);
2840 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2841 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2842 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,5,da2,ids);
2844 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2846 dac.setContigPartOfSelectedValues(4,da2,ids);
2847 expected6=[1,11,2,12,3,13,4,14,0,30,0,30,9,39]
2849 self.assertEqual(expected6[i],dac.getIJ(0,i));
2853 def testBuildDescendingConnec2Of3DMesh1(self):
2854 mesh=MEDCouplingDataForTest.build3DSourceMesh_1();
2856 mesh2,desc,descIndx,revDesc,revDescIndx=mesh.buildDescendingConnectivity2();
2857 mesh2.checkConsistencyLight();
2858 self.assertEqual(2,mesh2.getMeshDimension());
2859 self.assertEqual(30,mesh2.getNumberOfCells());
2860 self.assertEqual(31,revDescIndx.getNbOfElems()); self.assertEqual(31,revDescIndx.getNumberOfTuples());
2861 self.assertEqual(13,descIndx.getNbOfElems()); self.assertEqual(13,descIndx.getNumberOfTuples());
2862 self.assertEqual(48,desc.getNbOfElems()); self.assertEqual(48,desc.getNumberOfTuples());
2863 self.assertEqual(48,revDesc.getNbOfElems()); self.assertEqual(48,revDesc.getNumberOfTuples());
2864 expected1=[1,2,3,4,5,6,7,8,9,10,11,12,13,14,-10,15,-5,-13,16,17,-14,18,-4,19,-2,20,21,22,23,24,25,-11,26,-1,-12,-25,-22,27,28,-7,-20,-24,29,-16,-18,30,-8,-28]
2865 self.assertEqual(expected1,desc.getValues());
2866 expected2=[0,4,8,12,16,20,24,28,32,36,40,44,48]
2867 self.assertEqual(expected2,descIndx.getValues());
2868 expected3=[0,2,4,5,7,9,10,12,14,15,17,19,21,23,25,26,28,29,31,32,34,35,37,38,40,42,43,44,46,47,48]
2869 self.assertEqual(expected3,revDescIndx.getValues());
2870 expected4=[0,8,0,6,0,0,5,1,4,1,1,9,1,11,2,2,3,2,7,2,8,3,4,3,5,3,4,10,4,5,11,5,6,10,6,6,9,7,7,10,7,8,8,9,9,11,10,11]
2871 self.assertEqual(expected4,revDesc.getValues());
2872 conn=mesh2.getNodalConnectivity();
2873 connIndex=mesh2.getNodalConnectivityIndex();
2874 expected5=[0,4,8,12,16,20,24,28,32,36,40,44,48,52,56,60,64,68,72,76,80,84,88,92,96,100,104,108,112,116,120]
2875 self.assertEqual(expected5,connIndex.getValues());
2876 expected6=[3,8,1,7,3,8,3,1,3,1,3,7,3,7,3,8,3,6,0,8,3,6,2,0,3,0,2,8,3,8,2,6,3,7,4,5,3,7,8,4,3,4,8,5,3,5,8,7,3,6,8,4,3,6,7,8,3,4,7,6,3,8,4,0,3,0,4,6,3,6,3,8,3,7,3,6,3,8,0,1,3,1,0,3,3,3,0,8,3,4,1,5,3,4,8,1,3,1,8,5,3,1,7,5,3,0,2,3,3,3,2,8,3,1,4,0,3,3,2,6]
2877 self.assertEqual(expected6,conn.getValues());
2880 def testAre2DCellsNotCorrectlyOriented1(self):
2881 m1Coords=[1.,1.,-1.,-1.,-1.,-1.,1.,-1.]
2883 m1=MEDCouplingUMesh.New();
2884 m1.setMeshDimension(2);
2885 m1.allocateCells(1);
2886 m1.insertNextCell(NORM_QUAD4,4,m1Conn[0:4])
2887 m1.finishInsertingCells();
2888 myCoords1=DataArrayDouble.New();
2889 myCoords1.setValues(m1Coords,4,2);
2890 m1.setCoords(myCoords1);
2894 vec2=[3.*cos(pi/9.*i),3.*sin(pi/9.*i)];
2895 m1Cpy=m1.deepCopy();
2896 m1Cpy.translate(vec2);
2897 self.assertRaises(InterpKernelException,m1Cpy.are2DCellsNotCorrectlyOriented,vec1,False);
2898 m1Cpy.changeSpaceDimension(3);
2899 res=m1Cpy.are2DCellsNotCorrectlyOriented(vec1,False)
2900 self.assertEqual([0],res.getValues());
2904 def testDataArrayAbs1(self):
2905 d1=DataArrayDouble.New();
2906 val1=[2.,-3.,-5.,6.,-7.,-8.,9.,10.,-11.,-12.,-13.,-15.]
2907 expected1=[2.,3.,5.,6.,7.,8.,9.,10.,11.,12.,13.,15.]
2908 d1.setValues(val1,6,2);
2909 d2=d1.convertToIntArr();
2913 self.assertAlmostEqual(expected1[i],d1.getIJ(0,i),14);
2916 expected2=[2,3,5,6,7,8,9,10,11,12,13,15]
2919 self.assertEqual(expected2[i],d2.getIJ(0,i));
2925 def testGetValueOn3(self):
2927 v2=[0.7,1.25,0.,2.,1.5]
2928 disp=[5.,50.,500.,6.,60.,600.,7.,70.,700.,8.,80.,800.]
2929 m=MEDCouplingUMesh.New("myMesh",1)
2932 m.allocateCells(nbCells)
2933 coords=DataArrayDouble.New() ; coords.setValues(v,nbNodes,1)
2935 m.insertNextCell(NORM_SEG2,2,[0,1])
2936 m.insertNextCell(NORM_SEG2,2,[2,1])
2937 m.insertNextCell(NORM_SEG2,2,[2,3])
2938 m.finishInsertingCells()
2939 f=MEDCouplingFieldDouble.New(ON_NODES)
2941 array=DataArrayDouble.New(); array.setValues(disp,m.getNumberOfNodes(),3)
2943 arr1=f.getValueOnMulti(v2)
2944 self.assertEqual(5,arr1.getNumberOfTuples());
2945 self.assertEqual(3,arr1.getNumberOfComponents());
2946 expected1=[5.7,57.,570.,6.5,65.,650.,5.,50.,500.,8.,80.,800.,7.,70.,700.]
2948 self.assertAlmostEqual(expected1[i],arr1.getIJ(0,i),14);
2952 def testGetNodeIdsOfCell2(self):
2953 m1c=MEDCouplingCMesh.New();
2954 coordsX=DataArrayDouble.New();
2955 arrX=[ -1., 1., 2., 4., 4.5 ]
2956 coordsX.setValues(arrX,5,1);
2957 coordsY=DataArrayDouble.New();
2958 arrY=[ -2., 2., 4., 8.]
2959 coordsY.setValues(arrY,4,1);
2960 coordsZ=DataArrayDouble.New();
2962 coordsZ.setValues(arrZ,3,1);
2964 m1c.setCoordsAt(0,coordsX);
2965 expected1=[[0,1],[1,2],[2,3],[3,4]]
2966 self.assertEqual(4,m1c.getNumberOfCells())
2967 for i in range(m1c.getNumberOfCells()):
2968 self.assertEqual(expected1[i],m1c.getNodeIdsOfCell(i))
2971 m1c.setCoordsAt(1,coordsY);
2972 self.assertEqual(12,m1c.getNumberOfCells())
2973 self.assertEqual(20,m1c.getNumberOfNodes())
2974 expected2=[[0,1,6,5],[1,2,7,6],[2,3,8,7],[3,4,9,8],[5,6,11,10],[6,7,12,11],[7,8,13,12],[8,9,14,13],[10,11,16,15],[11,12,17,16],[12,13,18,17],[13,14,19,18]]
2975 for i in range(m1c.getNumberOfCells()):
2976 self.assertEqual(expected2[i],m1c.getNodeIdsOfCell(i))
2979 m1c.setCoordsAt(2,coordsZ);
2980 self.assertEqual(24,m1c.getNumberOfCells())
2981 self.assertEqual(60,m1c.getNumberOfNodes())
2982 expected3=[[0,1,6,5,20,21,26,25],[1,2,7,6,21,22,27,26],[2,3,8,7,22,23,28,27],[3,4,9,8,23,24,29,28],[5,6,11,10,25,26,31,30],[6,7,12,11,26,27,32,31],[7,8,13,12,27,28,33,32],[8,9,14,13,28,29,34,33],[10,11,16,15,30,31,36,35],[11,12,17,16,31,32,37,36],[12,13,18,17,32,33,38,37],[13,14,19,18,33,34,39,38],[20,21,26,25,40,41,46,45],[21,22,27,26,41,42,47,46],[22,23,28,27,42,43,48,47],[23,24,29,28,43,44,49,48],[25,26,31,30,45,46,51,50],[26,27,32,31,46,47,52,51],[27,28,33,32,47,48,53,52],[28,29,34,33,48,49,54,53],[30,31,36,35,50,51,56,55],[31,32,37,36,51,52,57,56],[32,33,38,37,52,53,58,57],[33,34,39,38,53,54,59,58]]
2983 self.assertEqual(24,m1c.getNumberOfCells())
2984 for i in range(m1c.getNumberOfCells()):
2985 self.assertEqual(expected3[i],m1c.getNodeIdsOfCell(i))
2991 if __name__ == '__main__':