1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2007-2021 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 if sys.platform == "win32":
23 from MEDCouplingCompat import *
25 from medcoupling import *
27 from math import pi,e,sqrt,cos,sin
28 from datetime import datetime
29 from MEDCouplingDataForTest import MEDCouplingDataForTest
30 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
32 class MEDCouplingBasicsTest3(unittest.TestCase):
33 def testSwigGetItem1(self):
39 da.setInfoOnComponent(0,"X [m]")
40 da.setInfoOnComponent(1,"Y [m]")
41 da.setInfoOnComponent(2,"Z [km]")
43 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())
45 self.assertEqual([19, 20, 21],da2.getValues())
48 except InterpKernelException as e:
51 self.assertTrue(False)
54 self.assertEqual([24, 27, 30, 33, 36, 39, 42, 45, 48],da2.getValues())
56 self.assertEqual([22, 23, 24, 25, 26, 27, 28, 29, 30],da2.getValues())
58 self.assertTrue(da2.isEqual(da))
60 self.assertTrue(da2.isEqual(da))
63 except InterpKernelException as e:
66 self.assertTrue(False)
68 self.assertTrue(da[5:8,-2].isEqualWithoutConsideringStr(DataArrayInt([23,26,29])))
70 self.assertEqual([22, 25, 28],da2.getValues())
73 except InterpKernelException as e:
76 self.assertTrue(False)
79 self.assertEqual([],da2.getValues())
82 def testSwigGetItem2(self):
83 da=DataArrayDouble.New()
88 da.setInfoOnComponent(0,"X [m]")
89 da.setInfoOnComponent(1,"Y [m]")
90 da.setInfoOnComponent(2,"Z [km]")
92 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())
94 self.assertEqual([19., 20., 21],da2.getValues())
97 except InterpKernelException as e:
100 self.assertTrue(False)
103 self.assertEqual([24., 27., 30., 33., 36., 39., 42., 45., 48.],da2.getValues())
105 self.assertEqual([22., 23., 24., 25., 26., 27., 28., 29., 30.],da2.getValues())
107 self.assertTrue(da2.isEqual(da,1e-12))
109 self.assertTrue(da2.isEqual(da,1e-12))
112 except InterpKernelException as e:
113 self.assertTrue(True)
115 self.assertTrue(False)
117 self.assertTrue(da[5:8,-2].isEqualWithoutConsideringStr(DataArrayDouble([23.,26.,29.]),1e-12))
119 self.assertEqual([22., 25., 28.],da2.getValues())
122 except InterpKernelException as e:
123 self.assertTrue(True)
125 self.assertTrue(False)
128 self.assertEqual([],da2.getValues())
131 def testSwigSetItem1(self):
132 da=DataArrayInt.New()
136 da.setInfoOnComponent(0,"X [m]") ; da.setInfoOnComponent(1,"Y [km]") ; da.setInfoOnComponent(2,"Y [m]")
137 da.setInfoOnComponent(3,"Z [W]") ; da.setInfoOnComponent(4,"ZZ [km]") ;
139 self.assertEqual([7, 8, 3, 10, 11, 12, 13, 3, 15, 16, 17, 18, 3, 20, 21, 22, 23, 3, 25, 26],da.getValues())
140 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
142 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 3, 3, 3, 3, 3, 22, 23, 24, 25, 26],da.getValues())
143 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
145 self.assertEqual([-1, -1, -1, -1, -1, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, -1, -1, -1, -1, -1],da.getValues())
146 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
148 self.assertEqual([7, -3, 9, -3, -3, 12, -3, 14, -3, -3, 17, -3, 19, -3, -3, 22, -3, 24, -3, -3],da.getValues())
149 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
150 da2=DataArrayInt.New() ; da2.setValues([0,2,3],3,1)
152 self.assertEqual([-7, -7, -7, -7, -7, 12, 13, 14, 15, 16, -7, -7, -7, -7, -7, -7, -7, -7, -7, -7],da.getValues())
153 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
155 self.assertEqual([7, 8, 9, -7, -7, 12, 13, 14, 15, 16, 17, 18, 19, -7, -7, 22, 23, 24, -7, -7],da.getValues())
156 # Let's test with DAI right hand side
157 da1=DataArrayInt.New()
158 da1.setValues([25,26,27,125,126,127],2,3)
160 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
162 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 25, 26, 27, 21, 22, 125, 126, 127, 26],da.getValues())
163 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
164 da[1:,3]=[225,226,227]
165 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 225, 16, 17, 18, 19, 226, 21, 22, 23, 24, 227, 26],da.getValues())
166 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
167 da[1,2:]=[225,226,227]
168 self.assertEqual([7, 8, 9, 10, 11, 12, 13, 225, 226, 227, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26],da.getValues())
169 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
170 da[da2,-2:]=[88,99,1010,1111,1212,1313]
171 self.assertEqual([7, 8, 9, 88, 99, 12, 13, 14, 15, 16, 17, 18, 19, 1010, 1111, 22, 23, 24, 1212, 1313],da.getValues())
172 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
173 da3=DataArrayInt.New(); da3.setValues([88,99,1010,1111,1212,1313],3,2)
175 self.assertEqual([7, 8, 9, 88, 99, 12, 13, 14, 15, 16, 17, 18, 19, 1010, 1111, 22, 23, 24, 1212, 1313],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([88, 8, 99, 10, 11, 12, 13, 14, 15, 16, 1010, 18, 1111, 20, 21, 1212, 23, 1313, 25, 26],da.getValues())
182 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
184 self.assertEqual([-8, 8, -8, 10, 11, 12, 13, 14, 15, 16, -8, 18, -8, 20, 21, -8, 23, -8, 25, 26],da.getValues())
187 def testSwigSetItem2(self):
188 da=DataArrayDouble.New()
192 da.setInfoOnComponent(0,"X [m]") ; da.setInfoOnComponent(1,"Y [km]") ; da.setInfoOnComponent(2,"Y [m]")
193 da.setInfoOnComponent(3,"Z [W]") ; da.setInfoOnComponent(4,"ZZ [km]") ;
195 self.assertEqual([7., 8., 3., 10., 11., 12., 13., 3., 15., 16., 17., 18., 3., 20., 21., 22., 23., 3., 25., 26.],da.getValues())
196 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
198 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 14., 15., 16., 3., 3., 3., 3., 3., 22., 23., 24., 25., 26.],da.getValues())
199 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
201 self.assertEqual([-1., -1., -1., -1., -1., 12., 13., 14., 15., 16., 17., 18., 19., 20., 21., -1., -1., -1., -1., -1.],da.getValues())
202 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
204 self.assertEqual([7., -3., 9., -3., -3., 12., -3., 14., -3., -3., 17., -3., 19., -3., -3., 22., -3., 24., -3., -3.],da.getValues())
205 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
206 da2=DataArrayInt.New() ; da2.setValues([0,2,3],3,1)
208 self.assertEqual([-7., -7., -7., -7., -7., 12., 13., 14., 15., 16., -7., -7., -7., -7., -7., -7., -7., -7., -7., -7.],da.getValues())
209 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
211 self.assertEqual([7., 8., 9., -7., -7., 12., 13., 14., 15., 16., 17., 18., 19., -7., -7., 22., 23., 24., -7., -7.],da.getValues())
212 # Let's test with DAI right hand side
213 da1=DataArrayDouble.New()
214 da1.setValues([25,26,27,125,126,127],2,3)
216 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
218 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 14., 15., 16., 17., 25., 26., 27., 21., 22., 125., 126., 127., 26.],da.getValues())
219 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
220 da[1:,3]=[225.,226.,227.]
221 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 14., 225., 16., 17., 18., 19., 226., 21., 22., 23., 24., 227., 26.],da.getValues())
222 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
223 da[1,2:]=[225,226,227]
224 self.assertEqual([7., 8., 9., 10., 11., 12., 13., 225., 226., 227., 17., 18., 19., 20., 21., 22., 23., 24., 25., 26.],da.getValues())
225 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
226 da[da2,-2:]=[88,99,1010,1111,1212,1313]
227 self.assertEqual([7., 8., 9., 88., 99., 12., 13., 14., 15., 16., 17., 18., 19., 1010., 1111., 22., 23., 24., 1212., 1313.],da.getValues())
228 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
229 da3=DataArrayDouble.New(); da3.setValues([88,99,1010,1111,1212,1313],3,2)
231 self.assertEqual([7., 8., 9., 88., 99., 12., 13., 14., 15., 16., 17., 18., 19., 1010., 1111., 22., 23., 24., 1212., 1313.],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([88., 8., 99., 10., 11., 12., 13., 14., 15., 16., 1010., 18., 1111., 20., 21., 1212., 23., 1313., 25., 26.],da.getValues())
238 da.rearrange(1) ; da.iota(7) ; da.rearrange(5)
240 self.assertEqual([-8., 8., -8., 10., 11., 12., 13., 14., 15., 16., -8., 18., -8., 20., 21., -8., 23., -8., 25., 26.],da.getValues())
243 def testSwigDADOp(self):
244 da=DataArrayDouble.New()
247 da1=DataArrayDouble.New()
251 self.assertEqual([15., 17., 19., 21., 23., 25., 27., 29., 31., 33., 35., 37.],da2.getValues())
254 self.assertTrue(da2.isEqual(da3,1e-12))
256 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())
258 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())
260 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())
262 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())
264 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())
266 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())
269 self.assertTrue(da4.isUniform(1.,1e-12))
270 st1=da.getHiddenCppPointer()
272 st2=da.getHiddenCppPointer()
273 self.assertEqual(st1,st2)
274 self.assertTrue(da.isEqual(da1,1e-12))
276 st2=da.getHiddenCppPointer()
277 self.assertEqual(st1,st2)
278 self.assertEqual(list(range(12)), da.getValues())
280 st2=da.getHiddenCppPointer()
281 self.assertEqual(st1,st2)
282 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())
284 st2=da.getHiddenCppPointer()
285 self.assertEqual(st1,st2)
286 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())
288 st2=da.getHiddenCppPointer()
289 self.assertEqual(st1,st2)
290 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())
292 self.assertEqual(st1,st2)
293 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())
295 st2=da.getHiddenCppPointer()
296 self.assertEqual(st1,st2)
297 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())
299 da5=DataArrayDouble.New()
300 da5.setValues([5.,4.,3.,2.],4,1)
301 da*=da5 # it works with unmathing number of compo
302 st2=da.getHiddenCppPointer()
303 self.assertEqual(st1,st2)
304 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())
309 ids=DataArrayInt.New()
310 ids.setValues([3,4,7],3,1)
312 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())
314 da.rearrange(1) ; da.iota(7) ; da.rearrange(3)
316 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())
319 def testSwigDAIOp(self):
320 da=DataArrayInt.New()
323 da1=DataArrayInt.New()
327 self.assertEqual([15,17,19,21,23,25,27,29,31,33,35,37],da2.getValues())
330 self.assertTrue(da2.isEqual(da3))
332 self.assertEqual([6,7,8,9,10,11,12,13,14,15,16,17],da2.getValues())
334 self.assertEqual([-6,-7,-8,-9,-10,-11,-12,-13,-14,-15,-16,-17],da2.getValues())
336 self.assertEqual([21,24,27,30,33,36,39,42,45,48,51,54.0],da2.getValues())
338 self.assertEqual([21,24,27,30,33,36,39,42,45,48,51,54.0],da2.getValues())
340 self.assertEqual([56,72,90,110,132,156,182,210,240,272,306,342.0],da2.getValues())
342 self.assertEqual([1,2,2,2,2,3,3,3,3,4,4,4],da2.getValues())
345 self.assertTrue(da4.isUniform(0))
346 st1=da.getHiddenCppPointer()
348 st2=da.getHiddenCppPointer()
349 self.assertEqual(st1,st2)
350 self.assertTrue(da.isEqual(da1))
352 st2=da.getHiddenCppPointer()
353 self.assertEqual(st1,st2)
354 self.assertEqual(list(range(12)), da.getValues())
356 st2=da.getHiddenCppPointer()
357 self.assertEqual(st1,st2)
358 self.assertEqual([8,10,12,14,16,18,20,22,24,26,28,30],da.getValues())
360 st2=da.getHiddenCppPointer()
361 self.assertEqual(st1,st2)
362 self.assertEqual([4,5,6,7,8,9,10,11,12,13,14,15],da.getValues())
364 st2=da.getHiddenCppPointer()
365 self.assertEqual(st1,st2)
366 self.assertEqual([32,45,60,77,96,117,140,165,192,221,252,285],da.getValues())
368 self.assertEqual(st1,st2)
369 self.assertEqual([4,5,6,7,8,9,10,11,12,13,14,15],da.getValues())
371 st2=da.getHiddenCppPointer()
372 self.assertEqual(st1,st2)
373 self.assertEqual([2,2, 3,3, 4,4, 5,5, 6,6, 7,7],da.getValues())
375 da5=DataArrayInt.New()
376 da5.setValues([5,4,3,2],4,1)
377 da*=da5 # it works with unmathing number of compo
378 st2=da.getHiddenCppPointer()
379 self.assertEqual(st1,st2)
380 self.assertEqual([10,10, 15,12,16,16,15,15, 18,12,14,14],da.getValues())
382 st2=da.getHiddenCppPointer()
383 self.assertEqual(st1,st2)
384 self.assertEqual([4,4,3,0,4,4,3,3,0,0,2,2],da.getValues())
389 ids=DataArrayInt.New()
390 ids.setValues([3,4,7],3,1)
392 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())
394 da.rearrange(1) ; da.iota(7) ; da.rearrange(3)
396 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())
399 def testSwigDAIOp2(self):
400 da=DataArrayInt.New()
401 st=da.getHiddenCppPointer()
411 self.assertEqual(st,da.getHiddenCppPointer())
412 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])
415 def testSwigDAIOp3(self):
416 da=DataArrayInt.New()
417 self.assertRaises(InterpKernelException,da.__len__)
418 self.assertRaises(InterpKernelException,da.__int__)
420 self.assertTrue(False)
423 da.rearrange(1) ; da.fillWithZero()
425 self.assertEqual(36,len(da));
428 self.assertRaises(InterpKernelException,tmp.__int__)
429 self.assertEqual(12,len(da));
431 for elt in enumerate(l):
434 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]
435 self.assertEqual(ref,da.getValues());
437 l=[int(elt) for elt in l1]
438 self.assertEqual(ref,da.getValues());
439 self.assertEqual(11,int(da[-1:]))
442 def testSwigDADOp3(self):
443 da=DataArrayDouble.New()
444 self.assertRaises(InterpKernelException,da.__len__)
445 self.assertRaises(InterpKernelException,da.__float__)
447 self.assertTrue(False)
450 da.rearrange(1) ; da.fillWithZero()
452 self.assertEqual(36,len(da));
455 self.assertRaises(InterpKernelException,tmp.__float__)
456 self.assertEqual(12,len(da));
458 for elt in enumerate(l):
461 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.]
462 self.assertEqual(ref,da.getValues());
464 l=[float(elt) for elt in l1]
465 self.assertEqual(ref,da.getValues());
466 self.assertEqual(11.,float(da[-1:]))
469 def testSwigDataArrayIntIterator1(self):
470 da=DataArrayInt.New()
474 # __getitem__ testing
479 self.assertEqual([3, 4, 6, 7, 9, 10, 12, 13],li)
484 self.assertEqual([4, 7, 10, 13],li)
489 self.assertEqual([4, 3, 2, 7, 6, 5, 10, 9, 8, 13, 12, 11],li)
490 # __setitem__ testing
492 da2=DataArrayInt.New()
502 self.assertTrue(da.isEqual(da2))
510 self.assertTrue(da.isUniform(5))
517 self.assertEqual([8, 9, 12, 8, 9, 12, 8, 9, 12, 8, 9, 12],da.getValues())
524 self.assertEqual([2, 3, 7, 5, 6, 7, 8, 9, 7, 11, 12, 7],da.getValues())
527 def testSwigDataArrayDoubleIterator1(self):
528 da=DataArrayDouble.New()
532 # __getitem__ testing
537 self.assertEqual([3, 4, 6, 7, 9, 10, 12, 13],li)
542 self.assertEqual([4, 7, 10, 13],li)
547 self.assertEqual([4, 3, 2, 7, 6, 5, 10, 9, 8, 13, 12, 11],li)
548 # __setitem__ testing
550 da2=DataArrayDouble.New()
560 self.assertTrue(da.isEqual(da2,1e-12))
568 self.assertTrue(da.isUniform(5,1e-12))
575 self.assertEqual([8, 9, 12, 8, 9, 12, 8, 9, 12, 8, 9, 12],da.getValues())
582 self.assertEqual([2, 3, 7, 5, 6, 7, 8, 9, 7, 11, 12, 7],da.getValues())
585 def testSwigUMeshIterator1(self):
586 m=MEDCouplingDataForTest.build2DTargetMesh_1()
590 li1+=cell.getAllConn()[1:]
591 li2+=[cell.getType()]
593 self.assertEqual(li1,[0, 3, 4, 1, 1, 4, 2, 4, 5, 2, 6, 7, 4, 3, 7, 8, 5, 4])
594 self.assertEqual(li2,[4, 3, 3, 4, 4])
597 def testSwigUMeshIterator2(self):
598 m=MEDCouplingDataForTest.build2DTargetMesh_1()
599 self.assertRaises(InterpKernelException,m.cellsByType);
600 m.rearrange2ConsecutiveCellTypes()
604 for cellsByType in m.cellsByType():
605 li1.append(cellsByType.getType())
606 li2.append(cellsByType.getNumberOfElems())
608 for cell in cellsByType:
611 t[1]=cell.getAllConn()[1:]
616 self.assertEqual(li1,[4, 3])
617 self.assertEqual(li2,[3, 2])
618 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)]]])
621 def testDAIAggregateMulti1(self):
623 a.setValues(list(range(4)), 2, 2)
626 b.setValues(list(range(6)), 3, 2)
627 c=DataArrayInt.Aggregate([a,b])
628 self.assertEqual(list(range(4)) + list(range(6)), c.getValues())
629 self.assertEqual("aa",c.getName())
630 self.assertEqual(5,c.getNumberOfTuples())
631 self.assertEqual(2,c.getNumberOfComponents())
634 def testMergeUMeshes2(self):
635 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
636 m2=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
637 m3=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
640 m2_2=m2.buildPartOfMySelf(vec1,False);
642 m3_2=m3.buildPartOfMySelf(vec2,False);
646 self.assertRaises(InterpKernelException,MEDCouplingUMesh.MergeUMeshes,ms+[None]);
647 self.assertRaises(InterpKernelException,MEDCouplingUMesh.MergeUMeshes,ms+[3.4])
648 m4=MEDCouplingUMesh.MergeUMeshes(ms);
649 m4.checkConsistencyLight();
650 self.assertEqual(10,m4.getNumberOfCells());
651 self.assertEqual(20,m4.getNumberOfNodes());
652 self.assertEqual(45,m4.getNodalConnectivityArrayLen());
653 m4bis=MEDCouplingMesh.MergeMeshes(ms);
654 self.assertTrue(m4.isEqual(m4bis,1e-12))
658 m4_1=m4.buildPartOfMySelf(vec3,False);
659 m4_1.setName(m1.getName());
660 self.assertTrue(m4_1.isEqual(m1,1e-12));
663 m4_2=m4.buildPartOfMySelf(vec4,False);
664 cellCor,nodeCor=m4_2.checkGeoEquivalWith(m2_2,10,1e-12);
667 m4_3=m4.buildPartOfMySelf(vec5,False);
668 self.assertEqual(2,m4_3.getNumberOfCells());
669 self.assertEqual(3,m4_3.getNumberOfNodes());
671 m4_3.setName(m3_2.getName());
672 self.assertTrue(m4_3.isEqual(m3_2,1e-12));
676 def testBuild0DMeshFromCoords1(self):
677 sourceCoords=[-0.3,-0.3,0., 0.7,-0.3,0., -0.3,0.7,0., 0.7,0.7,0.]
678 coo=DataArrayDouble.New();
679 coo.setValues(sourceCoords,4,3);
681 m=MEDCouplingUMesh.Build0DMeshFromCoords(coo);
682 m.checkConsistencyLight();
683 self.assertEqual(4,m.getNumberOfNodes());
684 self.assertEqual(4,m.getNumberOfCells());
685 self.assertEqual(3,m.getSpaceDimension());
686 self.assertEqual(0,m.getMeshDimension());
687 types1=m.getAllGeoTypes();
688 self.assertEqual([NORM_POINT1],types1);
690 conn=m.getNodeIdsOfCell(i);
691 self.assertEqual([i],conn);
692 self.assertTrue(NORM_POINT1==m.getTypeOfCell(i));
694 self.assertEqual(m.getName(),"My0D");
697 def testDescriptionInMeshTimeUnit1(self):
699 m=MEDCouplingDataForTest.build2DTargetMesh_1();
700 m.setDescription(text1);
701 self.assertEqual(m.getDescription(),text1);
703 self.assertTrue(m.isEqual(m2,1e-12));
704 self.assertEqual(m2.getDescription(),text1);
705 m2.setDescription("ggg");
706 self.assertTrue(not m.isEqual(m2,1e-12));
708 f=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
709 f.setTimeUnit(text1);
710 self.assertEqual(f.getTimeUnit(),text1);
712 self.assertEqual(f2.getTimeUnit(),text1);
716 def testMultiFields1(self):
717 mfs=MEDCouplingDataForTest.buildMultiFields_1();
719 dms,refs=mfs.getDifferentMeshes()
721 das2,refs2=mfs.getDifferentArrays()
722 self.assertEqual(5,len(mfs.getFields()))
723 self.assertEqual(1,len(mfs.getFields()[0].getArrays()));
724 self.assertEqual(2,len(mfs.getFields()[1].getArrays()));
725 self.assertEqual(1,len(mfs.getFields()[2].getArrays()));
726 self.assertEqual(1,len(mfs.getFields()[3].getArrays()));
727 self.assertEqual(1,len(mfs.getFields()[4].getArrays()));
728 self.assertEqual(5,len(ms));
729 self.assertEqual(2,len(dms));
730 self.assertEqual(6,len(das));
731 self.assertEqual(5,len(das2));
733 self.assertTrue(mfs.isEqual(mfs2,1e-12,1e-12))
736 def testFieldOverTime1(self):
737 fs=MEDCouplingDataForTest.buildMultiFields_2();
738 self.assertRaises(InterpKernelException,MEDCouplingFieldOverTime.New,fs);
739 f4bis=fs[4].buildNewTimeReprFromThis(ONE_TIME,False);
741 self.assertRaises(InterpKernelException,MEDCouplingFieldOverTime.New,fs);
742 f4bis.setTime(2.7,20,21);
743 fot=MEDCouplingFieldOverTime.New(fs);
744 dt=fot.getDefinitionTimeZone();
745 hs=dt.getHotSpotsTime();
746 self.assertEqual(6,len(hs));
747 expected1=[0.2,0.7,1.2,1.35,1.7,2.7]
749 self.assertAlmostEqual(expected1[i],hs[i],12);
751 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(0.2);
752 self.assertEqual(0,meshId);
753 self.assertEqual(0,arrId);
754 self.assertEqual(0,arrIdInField);
755 self.assertEqual(0,fieldId);
757 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(0.7);
758 self.assertEqual(0,meshId);
759 self.assertEqual(1,arrId);
760 self.assertEqual(0,arrIdInField);
761 self.assertEqual(1,fieldId);
763 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeLeft(1.2);#**** WARNING left here
764 self.assertEqual(0,meshId);
765 self.assertEqual(2,arrId);
766 self.assertEqual(1,arrIdInField);
767 self.assertEqual(1,fieldId);
769 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(1.2);#**** WARNING right again here
770 self.assertEqual(1,meshId);
771 self.assertEqual(3,arrId);
772 self.assertEqual(0,arrIdInField);
773 self.assertEqual(2,fieldId);
775 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(1.35);
776 self.assertEqual(1,meshId);
777 self.assertEqual(3,arrId);
778 self.assertEqual(0,arrIdInField);
779 self.assertEqual(2,fieldId);
781 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(1.7);
782 self.assertEqual(0,meshId);
783 self.assertEqual(3,arrId);
784 self.assertEqual(0,arrIdInField);
785 self.assertEqual(3,fieldId);
787 meshId,arrId,arrIdInField,fieldId=dt.getIdsOnTimeRight(2.7);
788 self.assertEqual(1,meshId);
789 self.assertEqual(4,arrId);
790 self.assertEqual(0,arrIdInField);
791 self.assertEqual(4,fieldId);
793 dt2=MEDCouplingDefinitionTime();
794 self.assertTrue(not dt2.isEqual(dt));
796 dt2.assign(dt);#to check memory management
797 self.assertTrue(dt2.isEqual(dt));
799 dt3=MEDCouplingDefinitionTime();
803 def testDAICheckAndPreparePermutation1(self):
804 vals1=[9,10,0,6,4,11,3,7];
805 expect1=[5,6,0,3,2,7,1,4];
806 vals2=[9,10,0,6,10,11,3,7];
807 da=DataArrayInt.New();
808 da.setValues(vals1,8,1);
809 da2=da.checkAndPreparePermutation();
810 self.assertEqual(8,da2.getNumberOfTuples());
811 self.assertEqual(1,da2.getNumberOfComponents());
813 self.assertEqual(expect1[i],da2.getIJ(i,0));
816 da=DataArrayInt.New();
819 da2=da.checkAndPreparePermutation();
820 self.assertEqual(1,da2.getNumberOfComponents());
821 self.assertTrue(da2.isIota(8));
823 da=DataArrayInt.New();
825 da.setValues(vals2,8,1);
826 self.assertRaises(InterpKernelException,da.checkAndPreparePermutation);
829 def testDAIChangeSurjectiveFormat1(self):
830 vals1=[0,3,2,3,2,2,1,2]
831 expected1=[0,1,2,6,8]
832 expected2=[0, 6, 2,4,5,7, 1,3]
833 da=DataArrayInt.New();
834 da.setValues(vals1,8,1);
836 da2,da2I=da.changeSurjectiveFormat(4);
837 self.assertEqual(5,da2I.getNumberOfTuples());
838 self.assertEqual(8,da2.getNumberOfTuples());
839 self.assertEqual(expected1,da2I.getValues());
840 self.assertEqual(expected2,da2.getValues());
842 self.assertRaises(InterpKernelException,da.changeSurjectiveFormat,3);
846 def testUMeshGetCellIdsLyingOnNodes1(self):
847 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
850 da=m.getCellIdsLyingOnNodes(nodeIds1,True);
851 self.assertEqual(1,da.getNumberOfTuples());
852 self.assertEqual(1,da.getNumberOfComponents());
853 self.assertEqual(1,da.getIJ(0,0));
854 da2=DataArrayInt.New()
855 da2.setValues(nodeIds2,2,1)
856 da=m.getCellIdsLyingOnNodes(da2,False);
857 self.assertEqual(2,da.getNumberOfTuples());
858 self.assertEqual(1,da.getNumberOfComponents());
859 self.assertEqual(3,da.getIJ(0,0));
860 self.assertEqual(4,da.getIJ(1,0));
863 def testUMeshFindCellIdsOnBoundary1(self):
864 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
865 da5=m.findCellIdsOnBoundary();
866 self.assertTrue(da5.isIota(5));
869 def testMeshSetTime1(self):
870 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
871 m2=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
873 self.assertTrue(m1.isEqual(m2,1e-12));
874 m1.setTime(3.14,6,7);
875 tmp3,tmp1,tmp2=m1.getTime();
876 self.assertEqual(6,tmp1);
877 self.assertEqual(7,tmp2);
878 self.assertAlmostEqual(3.14,tmp3,12);
879 self.assertTrue(not m1.isEqual(m2,1e-12));
880 m2.setTime(3.14,6,7);
881 self.assertTrue(m1.isEqual(m2,1e-12));
882 m1.setTimeUnit("ms");
883 self.assertTrue(m1.getTimeUnit()=="ms");
884 m1.setTimeUnit("us");
885 self.assertTrue(m1.getTimeUnit()=="us");
886 self.assertTrue(not m1.isEqual(m2,1e-12));
887 m2.setTimeUnit("us");
888 self.assertTrue(m1.isEqual(m2,1e-12));
889 m2.setTime(3.14,6,8);
890 self.assertTrue(not m1.isEqual(m2,1e-12));
891 m2.setTime(3.14,7,7);
892 self.assertTrue(not m1.isEqual(m2,1e-12));
893 m2.setTime(3.15,6,7);
894 self.assertTrue(not m1.isEqual(m2,1e-12));
896 m1.setTime(10.34,55,12);
898 self.assertTrue(m1.isEqual(m3,1e-12));
899 tmp3,tmp1,tmp2=m3.getTime();
900 self.assertEqual(55,tmp1);
901 self.assertEqual(12,tmp2);
902 self.assertAlmostEqual(10.34,tmp3,12);
906 a=DataArrayDouble.New();
907 a.setValues(coo1,4,1);
908 b=MEDCouplingCMesh.New();
911 b.setTime(5.67,8,100);
912 tmp3,tmp1,tmp2=b.getTime();
913 self.assertEqual(8,tmp1);
914 self.assertEqual(100,tmp2);
915 self.assertAlmostEqual(5.67,tmp3,12);
917 self.assertTrue(c.isEqual(b,1e-12));
918 tmp3,tmp1,tmp2=c.getTime();
919 self.assertEqual(8,tmp1);
920 self.assertEqual(100,tmp2);
921 self.assertAlmostEqual(5.67,tmp3,12);
924 def testApplyFuncTwo1(self):
925 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
926 f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
929 vals=[1.,11.,21.,2.,12.,22.,3.,13.,23.,4.,14.,24.,5.,15.,25.]
930 da=DataArrayDouble.New();
931 da.setValues(vals,5,3);
934 self.assertRaises(InterpKernelException,da.applyFuncCompo,1,"y+z");
935 da.setInfoOnComponent(0,"x [m]");
936 da.setInfoOnComponent(1,"y [mm]");
937 da.setInfoOnComponent(2,"z [km]");
939 self.assertRaises(InterpKernelException, da.applyFuncCompo, 1, "x+y+zz+zzz");
940 self.assertRaises(InterpKernelException, da.applyFuncCompo, 1, "toto(x+y)");
941 self.assertRaises(InterpKernelException, da.applyFuncCompo, 1, "x/0");
943 da2=da.applyFuncCompo(1,"y+z");
944 self.assertEqual(1,da2.getNumberOfComponents());
945 self.assertEqual(5,da2.getNumberOfTuples());
946 expected1=[32.,34.,36.,38.,40.]
948 self.assertAlmostEqual(expected1[i],da2.getIJ(0,i),12);
950 da2=da.applyFunc(1,"y+z");
951 expected2=[12.,14.,16.,18.,20.]
953 self.assertAlmostEqual(expected2[i],da2.getIJ(0,i),12);
956 self.assertEqual(3,f1.getNumberOfComponents());
957 self.assertEqual(5,f1.getNumberOfTuples());
958 f1.applyFuncCompo(1,"y+z");
959 self.assertEqual(1,f1.getNumberOfComponents());
960 self.assertEqual(5,f1.getNumberOfTuples());
962 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
967 def testApplyFuncThree1(self):
968 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
969 f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
972 vals=[1.,11.,21.,2.,12.,22.,3.,13.,23.,4.,14.,24.,5.,15.,25.]
973 da=DataArrayDouble.New();
974 da.setValues(vals,5,3);
978 vs[0]="x"; vs[1]="Y"; vs[2]="z";
979 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, vs, "y+z");
980 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, vs, "x+Y+z+zz+zzz");
981 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, vs, "x/0");
983 da2=da.applyFuncNamedCompo(1,vs,"y+z");
984 expected1=[32.,34.,36.,38.,40.]
986 self.assertAlmostEqual(expected1[i],da2.getIJ(0,i),12);
988 self.assertRaises(InterpKernelException, da.applyFuncNamedCompo, 1, ["x","y","z","a"],"x+a")
990 self.assertEqual(3,f1.getNumberOfComponents());
991 self.assertEqual(5,f1.getNumberOfTuples());
992 f1.applyFuncNamedCompo(1,vs,"y+z");
993 self.assertEqual(1,f1.getNumberOfComponents());
994 self.assertEqual(5,f1.getNumberOfTuples());
996 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
1000 def testFillFromAnalyticTwo1(self):
1001 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
1002 m1.setTime(3.4,5,6); m1.setTimeUnit("us");
1003 self.assertRaises(InterpKernelException,m1.fillFromAnalyticCompo,ON_NODES,1,"y+z");
1004 m1.getCoords().setInfoOnComponent(0,"x [m]");
1005 m1.getCoords().setInfoOnComponent(1,"y");
1006 m1.getCoords().setInfoOnComponent(2,"z");
1007 f1=m1.fillFromAnalyticCompo(ON_NODES,1,"y+z");
1008 self.assertAlmostEqual(3.4,f1.getTime()[0],12) ; self.assertEqual(5,f1.getTime()[1]) ; self.assertEqual(6,f1.getTime()[2])
1009 self.assertEqual("us",f1.getTimeUnit())
1010 self.assertEqual(1,f1.getNumberOfComponents());
1011 self.assertEqual(9,f1.getNumberOfTuples());
1012 expected1=[0.2, 0.7, 1.2, 0.7, 1.2, 1.7, 1.2, 1.7, 2.2]
1014 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
1018 def testFillFromAnalyticThree1(self):
1019 m1=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
1020 m1.setTime(3.4,5,6); m1.setTimeUnit("us");
1022 vs[0]="x"; vs[1]="Y"; vs[2]="z";
1023 self.assertRaises(InterpKernelException,m1.fillFromAnalyticNamedCompo,ON_NODES,1,vs,"y+z");
1025 f1=m1.fillFromAnalyticNamedCompo(ON_NODES,1,vs,"y+z");
1026 self.assertAlmostEqual(3.4,f1.getTime()[0],12) ; self.assertEqual(5,f1.getTime()[1]) ; self.assertEqual(6,f1.getTime()[2])
1027 self.assertEqual("us",f1.getTimeUnit())
1028 self.assertEqual(1,f1.getNumberOfComponents());
1029 self.assertEqual(9,f1.getNumberOfTuples());
1030 expected1=[0.2, 0.7, 1.2, 0.7, 1.2, 1.7, 1.2, 1.7, 2.2]
1032 self.assertAlmostEqual(expected1[i],f1.getArray().getIJ(0,i),12);
1036 def testDAUnitVar1(self):
1037 da=DataArrayDouble.New();
1039 da.setInfoOnComponent(0,"XPS [m]");
1040 st1=da.getVarOnComponent(0);
1041 self.assertTrue(st1=="XPS");
1042 st2=da.getUnitOnComponent(0);
1043 self.assertTrue(st2=="m");
1045 da.setInfoOnComponent(0,"XPS [m]");
1046 st1=da.getVarOnComponent(0);
1047 self.assertTrue(st1=="XPS");
1048 st2=da.getUnitOnComponent(0);
1049 self.assertTrue(st2=="m");
1051 da.setInfoOnComponent(0,"XPP [m]");
1052 st1=da.getVarOnComponent(0);
1053 self.assertTrue(st1=="XPP");
1054 st2=da.getUnitOnComponent(0);
1055 self.assertTrue(st2=="m");
1057 da.setInfoOnComponent(0,"XPP kdep kefer [ m ]");
1058 st1=da.getVarOnComponent(0);
1059 self.assertTrue(st1=="XPP kdep kefer");
1060 st2=da.getUnitOnComponent(0);
1061 self.assertTrue(st2==" m ");
1063 da.setInfoOnComponent(0," XPP k[ dep k]efer [ m^ 2/s^3*kJ ]");
1064 st1=da.getVarOnComponent(0);
1065 self.assertTrue(st1==" XPP k[ dep k]efer");
1066 st2=da.getUnitOnComponent(0);
1067 self.assertTrue(st2==" m^ 2/s^3*kJ ");
1069 da.setInfoOnComponent(0," XPP kefer ");
1070 st1=da.getVarOnComponent(0);
1071 self.assertTrue(st1==" XPP kefer ");
1072 st2=da.getUnitOnComponent(0);
1073 self.assertTrue(st2=="");
1075 da.setInfoOnComponent(0,"temperature( bof)");
1076 st1=da.getVarOnComponent(0);
1077 self.assertTrue(st1=="temperature( bof)");
1078 st2=da.getUnitOnComponent(0);
1079 self.assertTrue(st2=="");
1081 da.setInfoOnComponent(0,"kkk [m]");
1082 da.setInfoOnComponent(1,"ppp [m^2/kJ]");
1083 da.setInfoOnComponent(2,"abcde [MW/s]");
1085 vs=da.getVarsOnComponent();
1086 self.assertEqual(3,len(vs));
1087 self.assertTrue(vs[0]=="kkk");
1088 self.assertTrue(vs[1]=="ppp");
1089 self.assertTrue(vs[2]=="abcde");
1090 vs=da.getUnitsOnComponent();
1091 self.assertEqual(3,len(vs));
1092 self.assertTrue(vs[0]=="m");
1093 self.assertTrue(vs[1]=="m^2/kJ");
1094 self.assertTrue(vs[2]=="MW/s");
1097 def testGaussCoordinates1(self):
1098 #Testing 1D cell types
1099 m1=MEDCouplingDataForTest.build1DMultiTypes_1();
1100 f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME);
1105 f.setGaussLocalizationOnType(NORM_SEG2,refCoo1,gsCoo1,wg1);
1108 refCoo2=[-1.0,1.0,0.0];
1109 f.setGaussLocalizationOnType(NORM_SEG3,refCoo2,gsCoo2,wg2);
1111 resToTest=f.getLocalizationOfDiscr();
1112 self.assertEqual(3,resToTest.getNumberOfComponents());
1113 self.assertEqual(2,resToTest.getNumberOfTuples());
1114 expected1=[0.6,0.6,0.6, 0.6,0.6,0.6]
1116 self.assertAlmostEqual(expected1[i],resToTest.getIJ(0,i),14);
1119 #Testing 2D cell types
1120 m2=MEDCouplingDataForTest.build2DMultiTypes_1();
1121 f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME);
1124 tria3CooGauss=[ 0.1, 0.8, 0.2, 0.7 ]
1125 gsCoo3=tria3CooGauss
1126 tria3CooRef=[ 0.0, 0.0, 1.0 , 0.0, 0.0, 1.0 ]
1127 refCoo3=tria3CooRef;
1128 f.setGaussLocalizationOnType(NORM_TRI3,refCoo3,gsCoo3,wg3);
1130 tria6CooGauss=[ 0.3, 0.2, 0.2, 0.1, 0.2, 0.4 ]
1131 gsCoo4=tria6CooGauss;
1132 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]
1133 refCoo4=tria6CooRef;
1134 f.setGaussLocalizationOnType(NORM_TRI6,refCoo4,gsCoo4,wg4);
1135 wg5=[0.3,0.3,0.3,0.3];
1136 quad4CooGauss=[ 0.3, 0.2, 0.2, 0.1, 0.2, 0.4, 0.15, 0.27 ]
1137 gsCoo5=quad4CooGauss;
1138 quad4CooRef=[-1.0, 1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0]
1139 refCoo5=quad4CooRef;
1140 f.setGaussLocalizationOnType(NORM_QUAD4,refCoo5,gsCoo5,wg5);
1141 wg6=[0.3,0.3,0.3,0.3];
1142 quad8CooGauss=[ 0.34, 0.16, 0.21, 0.3, 0.23, 0.4, 0.14, 0.37 ]
1143 gsCoo6=quad8CooGauss;
1144 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]
1145 refCoo6=quad8CooRef;
1146 f.setGaussLocalizationOnType(NORM_QUAD8,refCoo6,gsCoo6,wg6);
1148 resToTest=f.getLocalizationOfDiscr();
1149 self.assertEqual(3,resToTest.getNumberOfComponents());
1150 self.assertEqual(13,resToTest.getNumberOfTuples());#2+3+4+4 gauss points for resp TRI3,TRI6,QUAD4,QUAD8
1151 expected2=[5.1,1.55,0.0, 4.7,1.65,0.0,
1152 2.32,1.52,0.0, 1.6,1.32,0.0, 3.52,1.26,0.0,#TRI6
1153 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
1154 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
1156 self.assertAlmostEqual(expected2[i],resToTest.getIJ(0,i),14);
1159 #Testing 3D cell types
1160 m3=MEDCouplingDataForTest.build3DMultiTypes_1();
1161 f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME);
1165 tetra4CooGauss=[0.34, 0.16, 0.21]
1166 gsCoo7=tetra4CooGauss;
1167 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]
1168 refCoo7=tetra4CooRef;
1169 f.setGaussLocalizationOnType(NORM_TETRA4,refCoo7,gsCoo7,wg7);
1171 tetra10CooGauss=[0.2, 0.3, 0.1]
1172 gsCoo8=tetra10CooGauss;
1173 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]
1174 refCoo8=tetra10CooRef;
1175 f.setGaussLocalizationOnType(NORM_TETRA10,refCoo8,gsCoo8,wg8);
1177 pyra5CooGauss=[0.2, 0.3, 0.1]
1178 gsCoo9=pyra5CooGauss;
1179 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]
1180 refCoo9=pyra5CooRef;
1181 f.setGaussLocalizationOnType(NORM_PYRA5,refCoo9,gsCoo9,wg9);
1183 pyra13CooGauss=[0.1, 0.2, 0.7]
1184 gsCoo10=pyra13CooGauss;
1185 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]
1186 refCoo10=pyra13CooRef;
1187 f.setGaussLocalizationOnType(NORM_PYRA13,refCoo10,gsCoo10,wg10);
1189 penta6CooGauss=[0.2, 0.3, 0.1]
1190 gsCoo11=penta6CooGauss;
1191 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]
1192 refCoo11=penta6CooRef;
1193 f.setGaussLocalizationOnType(NORM_PENTA6,refCoo11,gsCoo11,wg11);
1195 penta15CooGauss=[0.2, 0.3,0.15]
1196 gsCoo12=penta15CooGauss;
1197 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]
1198 refCoo12=penta15CooRef;
1199 f.setGaussLocalizationOnType(NORM_PENTA15,refCoo12,gsCoo12,wg12);
1201 hexa8CooGauss=[0.2,0.3,0.15]
1202 gsCoo13=hexa8CooGauss;
1203 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]
1204 refCoo13=hexa8CooRef;
1205 f.setGaussLocalizationOnType(NORM_HEXA8,refCoo13,gsCoo13,wg13);
1207 hexa20CooGauss=[0.11,0.3,0.55]
1208 gsCoo14=hexa20CooGauss;
1209 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]
1210 refCoo14=hexa20CooRef;
1211 f.setGaussLocalizationOnType(NORM_HEXA20,refCoo14,gsCoo14,wg14);
1213 resToTest=f.getLocalizationOfDiscr();
1214 self.assertEqual(3,resToTest.getNumberOfComponents());
1215 self.assertEqual(8,resToTest.getNumberOfTuples());#2+3+4+4 gauss points for resp TRI3,TRI6,QUAD4,QUAD8
1216 expected3=[1.312,3.15,1.02, 0.56,3.3,0.6, 2.18,1.1,0.2, 1.18,1.54,0.98, 1.56,0.3,3.6, 1.613,0.801,4.374, 2.6,2.4,2.3, 2.31232,2.3933985,1.553255]
1218 self.assertAlmostEqual(expected3[i],resToTest.getIJ(0,i),14);
1223 def testP2Localization1(self):
1224 m=MEDCouplingUMesh.New("testP2",2);
1225 coords=[0.,2.,3.5,0.,4.5,1.5,1.2,0.32,3.4,1.,2.1,2.4]
1227 coo=DataArrayDouble.New();
1228 coo.setValues(coords,6,2);
1231 m.insertNextCell(NORM_TRI6,6,conn[0:6])
1232 m.finishInsertingCells();
1234 f=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME);
1236 da=DataArrayDouble.New();
1237 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]
1238 da.setValues(vals1,6,3);
1242 locs=f.getValueOnMulti(loc);
1243 expected1=[6.0921164547752236, 7.1921164547752232, 8.2921164547752255]
1245 self.assertAlmostEqual(expected1[i],locs.getIJ(0,i),12);
1249 def testP2Localization2(self):
1250 m=MEDCouplingUMesh.New("testP2_2",3);
1251 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]
1252 conn=[0,1,2,3,4,5,6,7,8,9]
1253 coo=DataArrayDouble.New();
1254 coo.setValues(coords,10,3);
1257 m.insertNextCell(NORM_TETRA10,10,conn[0:10])
1258 m.finishInsertingCells();
1260 f=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME);
1262 da=DataArrayDouble.New();
1263 vals1=[1.1,2.1,3.1,4.1,5.2,6.2,7.2,8.2,9.2,10.2]
1264 da.setValues(vals1,10,1);
1266 f.checkConsistencyLight()
1268 loc=[0.64637931739890486, -0.16185896817550552, 0.22678966365273748]
1269 locs=f.getValueOnMulti(loc);
1270 expected1=[10.0844021968047]
1272 self.assertAlmostEqual(expected1[i],locs.getIJ(0,i),12);
1276 def testGetValueOn2(self):
1277 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1278 f=MEDCouplingFieldDouble.New(ON_CELLS,NO_TIME);
1280 arr=DataArrayDouble.New();
1281 nbOfCells=m.getNumberOfCells();
1283 values1=[7.,107.,10007.,8.,108.,10008.,9.,109.,10009.,10.,110.,10010.,11.,111.,10011.]
1284 arr.setValues(values1,nbOfCells,3);
1285 loc=[-0.05,-0.05, 0.55,-0.25, 0.55,0.15, -0.05,0.45, 0.45,0.45]
1286 f.checkConsistencyLight();
1287 locs=f.getValueOnMulti(loc);
1288 self.assertEqual(5,locs.getNumberOfTuples());
1289 self.assertEqual(3,locs.getNumberOfComponents());
1291 self.assertAlmostEqual(values1[j],locs.getIJ(0,j),12);
1294 f=MEDCouplingFieldDouble.New(ON_NODES,NO_TIME);
1296 arr=DataArrayDouble.New();
1297 nbOfNodes=m.getNumberOfNodes();
1299 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.]
1300 arr.setValues(values2,nbOfNodes,3);
1301 loc2=[0.5432,-0.2432, 0.5478,0.1528, 0.5432,-0.2432, 0.5432,-0.2432]
1302 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]
1303 f.checkConsistencyLight();
1304 loc3=DataArrayDouble.New()
1305 loc3.setValues(loc2,4,2);
1306 locs=f.getValueOnMulti(loc3);
1307 self.assertEqual(4,locs.getNumberOfTuples());
1308 self.assertEqual(3,locs.getNumberOfComponents());
1310 self.assertAlmostEqual(expected2[i],locs.getIJ(0,i),12);
1315 def testDAIGetIdsNotEqual1(self):
1316 d=DataArrayInt.New();
1317 vals1=[2,3,5,6,8,5,5,6,1,-5]
1318 d.setValues(vals1,10,1);
1319 d2=d.findIdsNotEqual(5);
1320 self.assertEqual(7,d2.getNumberOfTuples());
1321 self.assertEqual(1,d2.getNumberOfComponents());
1322 expected1=[0,1,3,4,7,8,9]
1324 self.assertEqual(expected1[i],d2.getIJ(0,i));
1327 self.assertRaises(InterpKernelException,d.findIdsNotEqual,5);
1331 d3=d.findIdsNotEqualList(vals3);
1332 self.assertEqual(5,d3.getNumberOfTuples());
1333 self.assertEqual(1,d3.getNumberOfComponents());
1334 expected2=[0,1,4,8,9]
1336 self.assertEqual(expected2[i],d3.getIJ(0,i));
1340 def testDAIComputeOffsets1(self):
1341 d=DataArrayInt.New();
1343 expected1=[0,3,8,9,11,11]
1344 d.setValues(vals1,6,1);
1346 self.assertEqual(6,d.getNumberOfTuples());
1347 self.assertEqual(1,d.getNumberOfComponents());
1349 self.assertEqual(expected1[i],d.getIJ(0,i));
1353 def testUMeshHexagonPrism1(self):
1354 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,
1355 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];
1356 conn=[1,2,3,4,5,0,7,8,9,10,11,6]
1357 mesh=MEDCouplingUMesh.New("MyFirstHexagonalPrism",3);
1358 coo=DataArrayDouble.New();
1359 coo.setValues(coords,12,3);
1360 mesh.setCoords(coo);
1361 mesh.allocateCells(1);
1362 mesh.insertNextCell(NORM_HEXGP12,12,conn[0:12])
1363 mesh.finishInsertingCells();
1365 mesh.checkConsistencyLight();
1366 vols=mesh.getMeasureField(False);
1367 self.assertEqual(1,vols.getNumberOfTuples());
1368 self.assertEqual(1,vols.getNumberOfComponents());
1369 self.assertAlmostEqual(-5.196152422706632,vols.getIJ(0,0),12);
1370 bary=mesh.computeCellCenterOfMass();
1371 self.assertEqual(1,bary.getNumberOfTuples());
1372 self.assertEqual(3,bary.getNumberOfComponents());
1373 expected1=[0.,0.,1.]
1375 self.assertAlmostEqual(expected1[i],bary.getIJ(0,i),12);
1377 d1=DataArrayInt.New();
1378 d2=DataArrayInt.New();
1379 d3=DataArrayInt.New();
1380 d4=DataArrayInt.New();
1381 m2=mesh.buildDescendingConnectivity(d1,d2,d3,d4);
1382 self.assertEqual(8,m2.getNumberOfCells());
1383 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]];
1384 expected2=[NORM_POLYGON, NORM_POLYGON, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4, NORM_QUAD4];
1385 expected3=[6,6,4,4,4,4,4,4]
1387 self.assertTrue(m2.getTypeOfCell(i)==expected2[i]);
1388 v=m2.getNodeIdsOfCell(i);
1389 self.assertTrue(len(v)==expected3[i]);
1390 self.assertEqual(expected4[i],v);
1392 mesh.convertAllToPoly();
1393 self.assertTrue(NORM_POLYHED==mesh.getTypeOfCell(0));
1395 self.assertTrue(NORM_HEXGP12==mesh.getTypeOfCell(0));
1396 self.assertEqual(13,mesh.getNodalConnectivityArrayLen());
1400 def testDADCheckIsMonotonic(self):
1401 da=DataArrayDouble.New();
1402 da.setValues([-1.,1.01,2.03,6.],2,2);
1403 self.assertRaises(InterpKernelException,da.isMonotonic,True,1e-12);
1405 self.assertTrue(da.isMonotonic(True,1e-12));
1406 da.checkMonotonic(True,1e-12);
1408 self.assertTrue(not da.isMonotonic(True,1e-12));
1409 self.assertRaises(InterpKernelException,da.checkMonotonic,True,1e-12);
1411 self.assertTrue(da.isMonotonic(True,1e-12));
1412 self.assertTrue(not da.isMonotonic(True,1e-1));
1415 def testCheckCoherencyDeeper1(self):
1416 m=MEDCouplingDataForTest.build3DSourceMesh_1();
1417 m.checkConsistencyLight();
1418 m.checkConsistency();
1419 m.getNodalConnectivity().setIJ(8,0,-1);
1420 m.checkConsistencyLight();
1421 self.assertRaises(InterpKernelException,m.checkConsistency);
1422 m.getNodalConnectivity().setIJ(8,0,-6);
1423 m.checkConsistencyLight();
1424 self.assertRaises(InterpKernelException,m.checkConsistency);
1425 m.getNodalConnectivity().setIJ(8,0,9);#9>=NbOfNodes
1426 m.checkConsistencyLight();
1427 self.assertRaises(InterpKernelException,m.checkConsistency);
1428 m.getNodalConnectivity().setIJ(8,0,8);#OK
1429 m.checkConsistencyLight();
1430 m.checkConsistency();
1432 m.convertToPolyTypes(elts);
1433 m.checkConsistencyLight();
1434 m.checkConsistency();
1435 m.getNodalConnectivity().setIJ(2,0,9);#9>=NbOfNodes
1436 m.checkConsistencyLight();
1437 self.assertRaises(InterpKernelException,m.checkConsistency);
1438 m.getNodalConnectivity().setIJ(2,0,-3);
1439 m.checkConsistencyLight();
1440 self.assertRaises(InterpKernelException,m.checkConsistency);
1441 m.getNodalConnectivity().setIJ(2,0,-1);
1442 m.checkConsistencyLight();
1443 self.assertRaises(InterpKernelException,m.checkConsistency);#Throw because cell#0 is not a polyhedron
1444 m.getNodalConnectivity().setIJ(2,0,4);
1445 m.checkConsistencyLight();
1446 m.checkConsistency();
1447 m.getNodalConnectivity().setIJ(7,0,-1);
1448 m.checkConsistencyLight();
1449 m.checkConsistency();#OK because we are in polyhedron connec
1450 m.getNodalConnectivity().setIJ(36,0,14);
1451 m.checkConsistencyLight();
1452 self.assertRaises(InterpKernelException,m.checkConsistency);#Throw because now cell 5 is a TETRA4 (14) so mismatch of number index and static type.
1455 def testUnPolyze2(self):
1456 m=MEDCouplingUMesh.New("jjj",3);
1457 coo=DataArrayDouble.New();
1464 m.insertNextCell(NORM_TETRA4,4,[0,1,2,3]);
1465 m.insertNextCell(NORM_TETRA4,4,[0,1,2,3]);
1466 m.finishInsertingCells();
1467 m2=MEDCouplingUMesh.MergeUMeshesOnSameCoords(4*[m]);
1468 m2.convertToPolyTypes([2]);
1470 self.assertEqual(NORM_TETRA4,m2.getTypeOfCell(2));
1471 self.assertEqual(40,m2.getNodalConnectivityArrayLen());
1472 temp2=m2.getNodeIdsOfCell(2);
1473 self.assertEqual(temp2,[0,1,2,3]);
1474 m2.checkConsistency();
1477 self.assertTrue(m3.isEqual(m2,1e-12));
1480 def testDACpyFrom1(self):
1481 d=DataArrayDouble.New();
1486 d.setInfoOnComponent(0,"X [m]");
1487 d.setInfoOnComponent(1,"Y [m]");
1488 d.setInfoOnComponent(2,"Z [m]");
1490 d1=DataArrayDouble.New();
1491 self.assertTrue(not d.isEqual(d1,1e-12));
1493 self.assertTrue(d.isEqual(d1,1e-12));
1495 self.assertTrue(d.isEqual(d1,1e-12));
1497 self.assertTrue(not d.isEqual(d1,1e-12));
1499 self.assertTrue(d.isEqual(d1,1e-12));
1501 d2=d.convertToIntArr();
1502 d4=DataArrayInt32.New();
1503 self.assertTrue(not d2.isEqual(d4));
1504 d4.deepCopyFrom(d2);
1505 self.assertTrue(d2.isEqual(d4));
1506 d4.deepCopyFrom(d2);
1507 self.assertTrue(d2.isEqual(d4));
1509 self.assertTrue(not d2.isEqual(d4));
1510 d4.deepCopyFrom(d2);
1511 self.assertTrue(d2.isEqual(d4));
1514 def testDAITransformWithIndArr1(self):
1516 tab2=[0,1,1,3,3,0,1,3,2,2,3,0]
1517 expected=[17,18,18,19,19,17,18,19,22,22,19,17]
1518 d=DataArrayInt.New();
1519 d.setValues(tab1,4,1);
1520 d1=DataArrayInt.New();
1521 d1.setValues(tab2,12,1);
1524 d1.transformWithIndArr(d);
1525 self.assertEqual(12,d1.getNumberOfTuples());
1526 self.assertEqual(1,d1.getNumberOfComponents());
1528 self.assertEqual(expected[i],d1.getIJ(i,0));
1532 d1.transformWithIndArr(tab1)
1533 self.assertEqual(12,d1.getNumberOfTuples());
1534 self.assertEqual(1,d1.getNumberOfComponents());
1536 self.assertEqual(expected[i],d1.getIJ(i,0));
1540 def testDAIBuildPermArrPerLevel1(self):
1541 arr=[2,0,1,1,0,1,2,0,1,1,0,0]
1542 expected1=[10,0,5,6,1,7,11,2,8,9,3,4]
1543 da=DataArrayInt.New();
1544 da.setValues(arr,12,1);
1545 da2=da.buildPermArrPerLevel();
1546 self.assertEqual(12,da2.getNumberOfTuples());
1547 self.assertEqual(1,da2.getNumberOfComponents());
1549 self.assertEqual(expected1[i],da2.getIJ(i,0));
1553 def testDAIOperations1(self):
1554 arr1=[-1,-2,4,7,3,2,6,6,4,3,0,1]
1555 da=DataArrayInt.New();
1556 da.setValues(arr1,4,3);
1557 da1=DataArrayInt.New();
1560 self.assertRaises(InterpKernelException,DataArrayInt.Add,da,da1);#not same number of tuples/Components
1562 da2=DataArrayInt.Add(da,da1);
1563 self.assertEqual(4,da2.getNumberOfTuples());
1564 self.assertEqual(3,da2.getNumberOfComponents());
1565 expected1=[1,1,8,12,9,9,14,15,14,14,12,14]
1567 self.assertEqual(expected1[i],da2.getIJ(0,i));
1569 da1.substractEqual(da);
1570 expected2=[3,5,0,-2,3,5,2,3,6,8,12,12]
1572 self.assertEqual(expected2[i],da1.getIJ(0,i));
1574 da1.rearrange(1); da1.iota(2); da1.rearrange(3);
1577 self.assertEqual(expected1[i],da1.getIJ(0,i));
1579 da1.rearrange(1); da1.iota(2); da1.rearrange(3);
1580 da2=DataArrayInt.Multiply(da,da1);
1581 self.assertEqual(4,da2.getNumberOfTuples());
1582 self.assertEqual(3,da2.getNumberOfComponents());
1583 expected3=[-2,-6,16,35,18,14,48,54,40,33,0,13]
1585 self.assertEqual(expected3[i],da2.getIJ(0,i));
1587 da.divideEqual(da1);
1588 self.assertEqual(4,da.getNumberOfTuples());
1589 self.assertEqual(3,da.getNumberOfComponents());
1590 expected4=[0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0]
1592 self.assertEqual(expected4[i],da.getIJ(0,i));
1594 da.setValues(arr1,4,3);
1595 da1.multiplyEqual(da);
1596 self.assertEqual(4,da1.getNumberOfTuples());
1597 self.assertEqual(3,da1.getNumberOfComponents());
1599 self.assertEqual(expected3[i],da1.getIJ(0,i));
1601 da1.rearrange(1); da1.iota(2); da1.rearrange(3);
1602 da2=DataArrayInt.Divide(da,da1);
1603 self.assertEqual(4,da2.getNumberOfTuples());
1604 self.assertEqual(3,da2.getNumberOfComponents());
1606 self.assertEqual(expected4[i],da2.getIJ(0,i));
1609 self.assertEqual(4,da1.getNumberOfTuples());
1610 self.assertEqual(3,da1.getNumberOfComponents());
1611 expected5=[160,107,80,64,53,45,40,35,32,29,26,24]
1613 self.assertEqual(expected5[i],da1.getIJ(0,i));
1615 da1.applyDivideBy(2);
1616 self.assertEqual(4,da1.getNumberOfTuples());
1617 self.assertEqual(3,da1.getNumberOfComponents());
1618 expected6=[80,53,40,32,26,22,20,17,16,14,13,12]
1620 self.assertEqual(expected6[i],da1.getIJ(0,i));
1622 expected7=[3,4,5,4,5,1,6,3,2,0,6,5]
1623 da1.applyModulus(7);
1625 self.assertEqual(expected7[i],da1.getIJ(0,i));
1628 expected8=[3,3,3,3,3,1,3,3,0,0,3,3]
1629 da1.applyRModulus(3);
1631 self.assertEqual(expected8[i],da1.getIJ(0,i));
1635 def testEmulateMEDMEMBDC1(self):
1636 m,m1=MEDCouplingDataForTest.buildPointe_1();
1637 m2,da1,da2,da3,da4,da5,da0=m.emulateMEDMEMBDC(m1)
1638 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]
1639 expected1=[1,32,29,23,41,36]
1640 self.assertEqual(47,da0.getNumberOfTuples());
1641 self.assertEqual(1,da0.getNumberOfComponents());
1643 self.assertEqual(expected0[i],da0.getIJ(0,i));
1645 self.assertEqual(6,da5.getNumberOfTuples());
1646 self.assertEqual(1,da5.getNumberOfComponents());
1648 self.assertEqual(expected1[i],da5.getIJ(0,i));
1650 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]
1651 self.assertEqual(70,da1.getNumberOfTuples());
1652 self.assertEqual(1,da1.getNumberOfComponents());
1654 self.assertEqual(expected2[i],da1.getIJ(0,i));
1656 expected3=[0,4,8,12,16,20,24,28,32,36,40,44,48,53,58,64,70]
1657 self.assertEqual(17,da2.getNumberOfTuples());
1658 self.assertEqual(1,da2.getNumberOfComponents());
1660 self.assertEqual(expected3[i],da2.getIJ(0,i));
1662 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]
1663 #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];
1664 self.assertEqual(48,da4.getNumberOfTuples());
1665 self.assertEqual(1,da4.getNumberOfComponents());
1667 self.assertEqual(expected4[i],da4.getIJ(0,i));
1669 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]
1670 self.assertEqual(70,da3.getNumberOfTuples());
1671 self.assertEqual(1,da3.getNumberOfComponents());
1673 self.assertEqual(expected5[i],da3.getIJ(0,i));
1677 def testGetLevArrPerCellTypes1(self):
1678 m,m1=MEDCouplingDataForTest.buildPointe_1();
1679 m1,d0,d1,d2,d3=m.buildDescendingConnectivity();
1680 order=[NORM_TRI3,NORM_QUAD4];
1681 da0,da1=m1.getLevArrPerCellTypes(order);
1682 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]
1683 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]
1684 self.assertEqual(47,da0.getNumberOfTuples());
1685 self.assertEqual(1,da0.getNumberOfComponents());
1687 self.assertEqual(expected0[i],da0.getIJ(0,i));
1689 self.assertEqual(2,da1.getNumberOfTuples());
1690 self.assertEqual(1,da1.getNumberOfComponents());
1691 self.assertEqual(36,da1.getIJ(0,0));#36 TRI3
1692 self.assertEqual(11,da1.getIJ(1,0));#11 QUAD4
1694 da2=da0.buildPermArrPerLevel();
1696 self.assertEqual(47,da2.getNumberOfTuples());
1697 self.assertEqual(1,da2.getNumberOfComponents());
1699 self.assertEqual(expected1[i],da2.getIJ(0,i));
1703 def testSortCellsInMEDFileFrmt1(self):
1704 m,m1=MEDCouplingDataForTest.buildPointe_1();
1706 da=DataArrayInt.New()
1707 da.setValues([0,1,2,14,3,12,4,5,15,6,7,8,9,10,11,13],16,1)
1708 daa=da.invertArrayN2O2O2N(16)
1709 m.renumberCells(daa,False)
1710 da2=m.sortCellsInMEDFileFrmt()
1711 self.assertEqual(da2.getValues(),[0,1,2,14,3,12,4,5,15,6,7,8,9,10,11,13])
1712 self.assertTrue(m.isEqual(m2,1e-12))
1713 self.assertTrue(da.isEqual(da2))
1716 def testBuildPartAndReduceNodes1(self):
1717 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1719 m2,da=m.buildPartAndReduceNodes(arr);
1720 self.assertEqual(5,m2.getNumberOfNodes());
1721 self.assertEqual(2,m2.getNumberOfCells());
1722 f=m2.getMeasureField(True);
1723 self.assertAlmostEqual(0.125,f.getArray().getIJ(0,0),12);
1724 self.assertAlmostEqual(0.25,f.getArray().getIJ(1,0),12);
1726 arr2=DataArrayInt.New()
1727 arr2.setValues(arr,2,1)
1728 m2,da=m.buildPartAndReduceNodes(arr2);
1729 self.assertEqual(5,m2.getNumberOfNodes());
1730 self.assertEqual(2,m2.getNumberOfCells());
1731 f=m2.getMeasureField(True);
1732 self.assertAlmostEqual(0.125,f.getArray().getIJ(0,0),12);
1733 self.assertAlmostEqual(0.25,f.getArray().getIJ(1,0),12);
1736 def testDAITransformWithIndArrR1(self):
1738 tab2=[-1,-1,0,1,2,3,4,5,-1,-1,-1,-1]
1739 expected=[0,3,1,2,4,5]
1740 d=DataArrayInt.New();
1741 d.setValues(tab1,6,1);
1742 d1=DataArrayInt.New();
1743 d1.setValues(tab2,12,1);
1746 d3=d.transformWithIndArrR(d1);
1747 self.assertEqual(6,d3.getNumberOfTuples());
1748 self.assertEqual(1,d3.getNumberOfComponents());
1750 self.assertEqual(expected[i],d3.getIJ(i,0));
1754 d3=d.transformWithIndArrR(tab2)
1755 self.assertEqual(6,d3.getNumberOfTuples());
1756 self.assertEqual(1,d3.getNumberOfComponents());
1758 self.assertEqual(expected[i],d3.getIJ(i,0));
1762 def testDAISplitByValueRange1(self):
1763 val1=[6,5,0,3,2,7,8,1,4]
1765 d=DataArrayInt.New();
1766 d.setValues(val1,9,1);
1767 e,f,g=d.splitByValueRange(val2);
1768 self.assertEqual(9,e.getNumberOfTuples());
1769 self.assertEqual(1,e.getNumberOfComponents());
1770 self.assertEqual(9,f.getNumberOfTuples());
1771 self.assertEqual(1,f.getNumberOfComponents());
1772 self.assertEqual(2,g.getNumberOfTuples());
1773 self.assertEqual(1,g.getNumberOfComponents());
1775 expected1=[1,1,0,0,0,1,1,0,1]
1776 expected2=[2,1,0,3,2,3,4,1,0]
1778 self.assertEqual(expected1[i],e.getIJ(i,0));
1779 self.assertEqual(expected2[i],f.getIJ(i,0));
1781 self.assertEqual(0,g.getIJ(0,0));
1782 self.assertEqual(1,g.getIJ(1,0));
1785 self.assertRaises(InterpKernelException,d.splitByValueRange,val2);
1786 # non regression test in python wrapping
1787 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])
1788 a,b,c=DataArrayInt([75]).splitByValueRange(rg)
1789 assert(a.isEqual(DataArrayInt([4])))
1790 assert(b.isEqual(DataArrayInt([0])))
1791 assert(c.isEqual(DataArrayInt([4])))
1794 def testUMeshSplitProfilePerType1(self):
1796 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1797 m.renumberCells(val0,False);
1800 d=DataArrayInt.New();
1801 d.setValues(val1,3,1);
1803 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1804 self.assertEqual(2,len(code));
1805 self.assertEqual(2,len(idsInPflPerType));
1806 expected1=[[3,1,0], [4,2,1]]
1807 self.assertEqual(expected1,code)
1808 self.assertEqual(2,len(idsInPflPerType));
1809 self.assertEqual(1,idsInPflPerType[0].getNumberOfTuples());
1810 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1811 self.assertEqual(2,idsInPflPerType[1].getNumberOfTuples());
1812 self.assertEqual(1,idsInPflPerType[1].getIJ(0,0));
1813 self.assertEqual(2,idsInPflPerType[1].getIJ(1,0));
1815 self.assertEqual(2,len(pfls));
1816 self.assertEqual("sup",pfls[0].getName())
1817 self.assertEqual(1,pfls[0].getNumberOfTuples());
1818 self.assertEqual(0,pfls[0].getIJ(0,0));
1819 self.assertEqual("sup",pfls[1].getName())
1820 self.assertEqual(2,pfls[1].getNumberOfTuples());
1821 self.assertEqual(0,pfls[1].getIJ(0,0));
1822 self.assertEqual(1,pfls[1].getIJ(1,0));
1825 d=DataArrayInt.New();
1826 d.setValues(val2,4,1);
1827 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1828 self.assertEqual(2,len(code));
1829 self.assertEqual(2,len(idsInPflPerType));
1830 expected2=[[3,1,0], [4,3,-1]]
1831 self.assertEqual(expected2,code);
1832 self.assertEqual(2,len(idsInPflPerType));
1833 self.assertEqual(1,idsInPflPerType[0].getNumberOfTuples());
1834 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1835 self.assertEqual(3,idsInPflPerType[1].getNumberOfTuples());
1836 self.assertEqual(1,idsInPflPerType[1].getIJ(0,0));
1837 self.assertEqual(2,idsInPflPerType[1].getIJ(1,0));
1838 self.assertEqual(3,idsInPflPerType[1].getIJ(2,0));
1840 self.assertEqual(1,len(pfls));
1841 self.assertEqual(1,pfls[0].getNumberOfTuples());
1842 self.assertEqual(0,pfls[0].getIJ(0,0));
1845 d=DataArrayInt.New();
1846 d.setValues(val3,3,1);
1847 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1848 self.assertEqual(2,len(code));
1849 self.assertEqual(2,len(idsInPflPerType));
1850 expected3=[[3,2,0], [4,1,1]]
1851 self.assertEqual(expected3,code);
1852 self.assertEqual(2,len(idsInPflPerType));
1853 self.assertEqual(2,idsInPflPerType[0].getNumberOfTuples());
1854 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1855 self.assertEqual(1,idsInPflPerType[0].getIJ(1,0));
1856 self.assertEqual(1,idsInPflPerType[1].getNumberOfTuples());
1857 self.assertEqual(2,idsInPflPerType[1].getIJ(0,0));
1859 self.assertEqual(2,len(pfls));
1860 self.assertEqual(2,pfls[0].getNumberOfTuples());
1861 self.assertEqual(1,pfls[0].getIJ(0,0));
1862 self.assertEqual(0,pfls[0].getIJ(1,0));
1863 self.assertEqual(0,pfls[1].getIJ(0,0));
1866 d=DataArrayInt.New();
1867 d.setValues(val4,2,1);
1868 code,idsInPflPerType,pfls=m.splitProfilePerType(d);
1869 self.assertEqual(1,len(code));
1870 self.assertEqual(1,len(idsInPflPerType));
1872 self.assertEqual(expected4,code);
1873 self.assertEqual(1,len(idsInPflPerType));
1874 self.assertEqual(2,idsInPflPerType[0].getNumberOfTuples());
1875 self.assertEqual(0,idsInPflPerType[0].getIJ(0,0));
1876 self.assertEqual(1,idsInPflPerType[0].getIJ(1,0));
1878 self.assertEqual(1,len(pfls));
1879 self.assertEqual(2,pfls[0].getNumberOfTuples());
1880 self.assertEqual(1,pfls[0].getIJ(0,0));
1881 self.assertEqual(2,pfls[0].getIJ(1,0));
1884 def testDAIBuildExplicitArrByRanges1(self):
1885 d=DataArrayInt.New();
1887 d.setValues(vals1,3,1);
1888 e=DataArrayInt.New();
1889 vals2=[0,3,6,10,14,20]
1890 e.setValues(vals2,6,1);
1892 f=d.buildExplicitArrByRanges(e);
1893 self.assertEqual(11,f.getNumberOfTuples());
1894 self.assertEqual(1,f.getNumberOfComponents());
1895 expected1=[0,1,2,6,7,8,9,10,11,12,13]
1897 self.assertEqual(expected1[i],f.getIJ(i,0));
1901 def testDAIComputeOffsets2(self):
1902 d=DataArrayInt.New();
1904 expected1=[0,3,8,9,11,11,19]
1905 d.setValues(vals1,6,1);
1906 d.computeOffsetsFull();
1907 self.assertEqual(7,d.getNumberOfTuples());
1908 self.assertEqual(1,d.getNumberOfComponents());
1910 self.assertEqual(expected1[i],d.getIJ(0,i));
1914 def testMergeField3(self):
1915 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1916 m.getCoords().setInfoOnComponent(0,"x [m]");
1917 m.getCoords().setInfoOnComponent(1,"z [km]");
1919 m.setDescription("desc");
1920 f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
1923 arr=DataArrayDouble.New();
1925 arr.setInfoOnComponent(0,"X [m]");
1926 arr.setInfoOnComponent(1,"YY [mm]");
1927 arr.fillWithValue(2.);
1930 f2=MEDCouplingFieldDouble.MergeFields([f1]);
1931 self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
1935 def testGetDistributionOfTypes1(self):
1936 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1938 self.assertRaises(InterpKernelException,m.getDistributionOfTypes);
1939 m.renumberCells(tab1,False);
1940 code=m.getDistributionOfTypes();
1941 self.assertEqual(2,len(code));
1942 self.assertEqual(3,code[0][0]);
1943 self.assertEqual(2,code[0][1]);
1944 self.assertEqual(-1,code[0][2]);
1945 self.assertEqual(4,code[1][0]);
1946 self.assertEqual(3,code[1][1]);
1947 self.assertEqual(-1,code[1][2]);
1950 def testNorm2_1(self):
1951 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1952 f=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
1955 d=DataArrayDouble.New();
1956 tab=[1.2,1.3,2.2,2.3,3.2,3.3,4.2,4.3,5.2,5.3]
1957 d.setValues(tab,5,2);
1959 f.checkConsistencyLight();
1961 self.assertAlmostEqual(11.209371079592289,f.norm2(),14);
1965 def testNormMax1(self):
1966 m=MEDCouplingDataForTest.build2DTargetMesh_1();
1967 f=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
1970 d=DataArrayDouble.New(); d.alloc(0,2)
1971 res = d.normMaxPerComponent()
1972 self.assertAlmostEqual(-1.0, res[0],14)
1973 self.assertAlmostEqual(-1.0, res[1],14)
1975 tab=[2.3,-1.2,6.3,-7.8,2.9,7.7,2.1,0.,3.6,-7.6]
1976 d.setValues(tab,5,2);
1978 res = d.normMaxPerComponent()
1979 self.assertAlmostEqual(6.3, res[0],14)
1980 self.assertAlmostEqual(7.8, res[1],14)
1983 f.checkConsistencyLight();
1986 self.assertAlmostEqual(6.3,res[0],14);
1987 self.assertAlmostEqual(7.8,res[1],14);
1989 self.assertAlmostEqual(6.3,f.normMax(0),14);
1990 self.assertAlmostEqual(7.8,f.normMax(1),14);
1994 def testFindAndCorrectBadOriented3DExtrudedCells1(self):
1995 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];
1996 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]
1997 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]
1998 invalidCells=[2,3,6,7]
1999 m=MEDCouplingUMesh.New("Example",3);
2000 coo=DataArrayDouble.New();
2001 coo.setValues(coords,38,3);
2004 m.insertNextCell(NORM_PENTA6,6,conn[0:6])
2005 m.insertNextCell(NORM_PENTA6,6,conn[6:12])
2006 m.insertNextCell(NORM_PENTA6,6,conn[12:18])
2007 m.insertNextCell(NORM_PENTA6,6,conn[18:24])
2008 m.insertNextCell(NORM_HEXA8,8,conn[24:32])
2009 m.insertNextCell(NORM_HEXA8,8,conn[32:40])
2010 m.insertNextCell(NORM_HEXA8,8,conn[40:48])
2011 m.insertNextCell(NORM_HEXA8,8,conn[48:56])
2012 m.finishInsertingCells();
2014 v=m.findAndCorrectBadOriented3DExtrudedCells();
2015 self.assertEqual(4,len(v));
2016 self.assertEqual(v.getValues(),invalidCells);
2017 self.assertEqual(connExp,m.getNodalConnectivity().getValues());
2018 self.assertTrue(m.findAndCorrectBadOriented3DExtrudedCells().empty())
2022 def testConvertExtrudedPolyhedra1(self):
2023 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]
2024 m=MEDCouplingUMesh.New("Example",3);
2025 coo=DataArrayDouble.New();
2027 coo.rearrange(1); coo.iota(0); coo.rearrange(3);
2030 m.insertNextCell(NORM_TETRA4,4,conn[0:4])
2031 m.insertNextCell(NORM_HEXA8,8,conn[4:12])
2032 m.insertNextCell(NORM_TETRA4,4,conn[12:16])
2033 m.insertNextCell(NORM_POLYHED,6,conn[16:22])
2034 m.insertNextCell(NORM_PENTA6,6,conn[22:28])
2035 m.insertNextCell(NORM_POLYHED,10,conn[28:38])
2036 m.insertNextCell(NORM_HEXA8,8,conn[38:46])
2037 m.insertNextCell(NORM_HEXGP12,12,conn[46:58])
2038 m.insertNextCell(NORM_POLYHED,14,conn[58:72])
2039 m.finishInsertingCells();
2041 m.convertExtrudedPolyhedra();
2042 da=m.getNodalConnectivity();
2043 dai=m.getNodalConnectivityIndex();
2044 self.assertEqual(10,dai.getNbOfElems());
2045 self.assertEqual(159,da.getNbOfElems());
2047 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];
2048 expected2=[0,5,14,19,42,49,86,95,108,159]
2049 self.assertEqual(expected1,da.getValues());
2050 self.assertEqual(expected2,dai.getValues());
2051 m.checkConsistency()
2054 def testNonRegressionCopyTinyStrings(self):
2055 m=MEDCouplingDataForTest.build2DTargetMesh_1()
2056 f1=m.getMeasureField(True)
2057 f1.getArray().setInfoOnComponent(0,"P [N/m^2]")
2058 bary=m.computeCellCenterOfMass()
2059 f2=f1.buildNewTimeReprFromThis(NO_TIME,False)
2061 self.assertRaises(InterpKernelException,f1.copyTinyAttrFrom,f2)
2064 def testDaDSetPartOfValuesAdv1(self):
2065 tab1=[3.,4.,5., 13.,14.,15., 23.,24.,25., 33.,34.,35., 43.,44.,45., 53.,54.,55.]
2066 tab2=[6.,7.,8., 16.,17.,18., 26.,27.,28.]
2067 tab3=[4,1, 2,2, 3,0]
2068 a=DataArrayDouble.New();
2069 a.setValues(tab1,6,3);
2070 b=DataArrayDouble.New();
2071 b.setValues(tab2,3,3);
2072 c=DataArrayInt.New();
2073 c.setValues(tab3,3,2);
2075 a.setPartOfValuesAdv(b,c);
2076 expected1=[3.,4.,5., 13.,14.,15., 26.,27.,28., 6.,7.,8., 16.,17.,18., 53.,54.,55.]
2077 self.assertEqual(expected1,a.getValues());
2080 def testUMeshBuildSetInstanceFromThis1(self):
2081 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
2082 m2=m.buildSetInstanceFromThis(3);
2083 self.assertTrue(m.isEqual(m2,1e-12));
2085 m=MEDCouplingUMesh.New("toto",2);
2086 m2=m.buildSetInstanceFromThis(3);
2087 self.assertEqual(0,m2.getNumberOfNodes());
2088 self.assertEqual(0,m2.getNumberOfCells());
2091 def testUMeshMergeMeshesCVW1(self):
2092 m=MEDCouplingDataForTest.build3DSurfTargetMesh_1();
2093 m2=MEDCouplingUMesh.New("toto",2);
2094 m3=MEDCouplingUMesh.MergeUMeshes([m,m2]);
2095 m3.setName(m.getName());
2096 self.assertTrue(m.isEqual(m3,1e-12));
2099 def testChangeUnderlyingMeshWithCMesh1(self):
2100 mesh=MEDCouplingCMesh.New();
2101 coordsX=DataArrayDouble.New();
2102 arrX=[ -1., 1., 2., 4. ]
2103 coordsX.setValues(arrX,4,1);
2104 coordsY=DataArrayDouble.New();
2105 arrY=[ -2., 2., 4., 8. ]
2106 coordsY.setValues(arrY,4,1);
2107 coordsZ=DataArrayDouble.New();
2108 arrZ=[ -3., 3., 6., 12. ]
2109 coordsZ.setValues(arrZ,4,1);
2110 mesh.setCoords(coordsX,coordsY,coordsZ);
2111 f=mesh.getMeasureField(True)
2112 mesh2=mesh.deepCopy()
2113 for myId in [0,1,2,10,11,12,20,21,22]:
2114 f=mesh.getMeasureField(True)
2115 f.changeUnderlyingMesh(mesh2,myId,1e-12);
2117 mesh2.setName("uuuu")
2118 for myId in [1,2,10,11,12,20,21,22]:
2119 f=mesh.getMeasureField(True)
2120 f.changeUnderlyingMesh(mesh2,myId,1e-12);
2124 def testDADFindCommonTuples1(self):
2125 da=DataArrayDouble.New();
2127 array1=[2.3,1.2,1.3,2.3,2.301,0.8]
2128 da.setValues(array1,6,1)
2129 c,cI=da.findCommonTuples(1e-2);
2132 self.assertEqual(3,c.getNbOfElems());
2133 self.assertEqual(2,cI.getNbOfElems());
2134 self.assertEqual(expected1,c.getValues())
2135 self.assertEqual(expected2,cI.getValues())
2136 c,cI=da.findCommonTuples(2e-1)
2137 expected3=[0,3,4,1,2]
2139 self.assertEqual(5,c.getNbOfElems());
2140 self.assertEqual(3,cI.getNbOfElems());
2141 self.assertEqual(expected3,c.getValues())
2142 self.assertEqual(expected4,cI.getValues())
2144 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]
2145 da.setValues(array2,6,2)
2146 c,cI=da.findCommonTuples(1e-2);
2147 self.assertEqual(3,c.getNbOfElems());
2148 self.assertEqual(2,cI.getNbOfElems());
2149 self.assertEqual(expected1,c.getValues())
2150 self.assertEqual(expected2,cI.getValues())
2151 c,cI=da.findCommonTuples(2e-1)
2152 self.assertEqual(5,c.getNbOfElems());
2153 self.assertEqual(3,cI.getNbOfElems());
2154 self.assertEqual(expected3,c.getValues())
2155 self.assertEqual(expected4,cI.getValues())
2157 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]
2158 da.setValues(array3,6,3)
2159 c,cI=da.findCommonTuples(1e-2);
2160 self.assertEqual(3,c.getNbOfElems());
2161 self.assertEqual(2,cI.getNbOfElems());
2162 self.assertEqual(expected1,c.getValues())
2163 self.assertEqual(expected2,cI.getValues())
2164 c,cI=da.findCommonTuples(2e-1)
2165 self.assertEqual(5,c.getNbOfElems());
2166 self.assertEqual(3,cI.getNbOfElems());
2167 self.assertEqual(expected3,c.getValues())
2168 self.assertEqual(expected4,cI.getValues())
2169 # nbOftuples=1, no common groups
2170 array11=[2.3,1.2,1.3,2.4,2.5,0.8]
2171 da.setValues(array11,6,1)
2172 c,cI=da.findCommonTuples(1e-2);
2173 self.assertEqual(0,c.getNbOfElems());
2174 self.assertEqual(1,cI.getNbOfElems());
2175 self.assertEqual([0],cI.getValues())
2178 da.setValues(array12,6,5) #bad NumberOfComponents
2179 self.assertRaises(InterpKernelException, da.findCommonTuples, 1e-2);
2182 def testDABack1(self):
2183 da=DataArrayDouble.New();
2184 array1=[2.3,1.2,1.3,2.3,2.301,0.8]
2185 da.setValues(array1,6,1);
2186 self.assertAlmostEqual(0.8,da.back(),14);
2188 self.assertRaises(InterpKernelException,da.back);
2190 self.assertRaises(InterpKernelException,da.back);
2192 da=DataArrayInt.New();
2194 da.setValues(array2,4,1);
2195 self.assertEqual(2,da.back());
2197 self.assertRaises(InterpKernelException,da.back);
2199 self.assertRaises(InterpKernelException,da.back);
2202 def testDADGetDifferentValues1(self):
2203 da=DataArrayDouble.New();
2204 array1=[2.3,1.2,1.3,2.3,2.301,0.8]
2205 da.setValues(array1,6,1)
2207 expected1=[2.301,1.2,1.3,0.8]
2208 dv=da.getDifferentValues(1e-2);
2209 self.assertEqual(4,dv.getNbOfElems());
2211 self.assertAlmostEqual(expected1[i],dv.getIJ(i,0),14);
2214 dv=da.getDifferentValues(2e-1);
2215 expected2=[2.301,1.3,0.8]
2216 self.assertEqual(3,dv.getNbOfElems());
2218 self.assertAlmostEqual(expected2[i],dv.getIJ(i,0),14);
2222 def testDAIBuildOld2NewArrayFromSurjectiveFormat2(self):
2225 a=DataArrayInt.New();
2226 a.setValues(arr,5,1);
2227 b=DataArrayInt.New();
2228 b.setValues(arrI,3,1);
2229 ret,newNbTuple=DataArrayInt.ConvertIndexArrayToO2N(10,a,b);
2230 expected=[0,1,2,0,3,4,5,4,6,4]
2231 self.assertEqual(10,ret.getNbOfElems());
2232 self.assertEqual(7,newNbTuple);
2233 self.assertEqual(1,ret.getNumberOfComponents());
2234 self.assertEqual(expected,ret.getValues());
2235 self.assertRaises(InterpKernelException,DataArrayInt.ConvertIndexArrayToO2N,9,a,b);
2238 def testDADIReverse1(self):
2240 a=DataArrayInt.New();
2241 a.setValues(arr,6,1);
2242 self.assertEqual(2,a.back());
2245 self.assertEqual(arr[5-i],a.getIJ(i,0));
2247 a.setValues(arr[:-1],5,1);
2250 self.assertEqual(arr[4-i],a.getIJ(i,0));
2253 arr2=[0.,3.,5.,7.,9.,2.]
2254 b=DataArrayDouble.New();
2255 b.setValues(arr2,6,1);
2258 self.assertAlmostEqual(arr2[5-i],b.getIJ(i,0),14);
2260 b.setValues(arr2[:5],5,1);
2261 self.assertAlmostEqual(9.,b.back(),14)
2264 self.assertAlmostEqual(arr2[4-i],b.getIJ(i,0),14);
2268 def testGetNodeIdsInUse1(self):
2269 m0=MEDCouplingDataForTest.build2DTargetMesh_1();
2271 m1=m0.buildPartOfMySelf(CellIds,True);
2272 arr,newNbOfNodes=m1.getNodeIdsInUse();
2273 expected=[-1,0,1,-1,2,3,-1,-1,-1]
2274 self.assertEqual(4,newNbOfNodes);
2275 self.assertEqual(9,arr.getNbOfElems());
2276 self.assertEqual(expected,arr.getValues());
2277 arr2=arr.invertArrayO2N2N2O(newNbOfNodes);
2278 self.assertEqual(4,arr2.getNbOfElems());
2280 self.assertEqual(expected2,arr2.getValues());
2283 def testBuildDescendingConnec2(self):
2284 mesh=MEDCouplingDataForTest.build2DTargetMesh_1();
2286 mesh2,desc,descIndx,revDesc,revDescIndx=mesh.buildDescendingConnectivity2();
2287 mesh2.checkConsistencyLight();
2288 self.assertEqual(1,mesh2.getMeshDimension());
2289 self.assertEqual(13,mesh2.getNumberOfCells());
2290 self.assertEqual(14,revDescIndx.getNbOfElems()); self.assertEqual(14,revDescIndx.getNumberOfTuples());
2291 self.assertEqual(6,descIndx.getNbOfElems()); self.assertEqual(6,descIndx.getNumberOfTuples());
2292 self.assertEqual(18,desc.getNbOfElems()); self.assertEqual(18,desc.getNumberOfTuples());
2293 self.assertEqual(18,revDesc.getNbOfElems()); self.assertEqual(18,revDesc.getNumberOfTuples());
2294 expected1=[1,2,3,4,-3,5,6, 7,8,-5,9,10,-2,11, 12,13,-7,-10]
2295 self.assertEqual(expected1,desc.getValues());
2296 expected2=[0,4,7,10,14,18]
2297 self.assertEqual(expected2,descIndx.getValues());
2298 expected3=[0,1,3,5,6,8,9,11,12,13,15,16,17,18]
2299 self.assertEqual(expected3,revDescIndx.getValues());
2300 expected4=[0, 0,3, 0,1, 0, 1,2, 1, 2,4, 2, 3, 3,4, 3, 4, 4]
2301 self.assertEqual(expected4,revDesc.getValues());
2302 conn=mesh2.getNodalConnectivity();
2303 connIndex=mesh2.getNodalConnectivityIndex();
2304 expected5=[0,3,6,9,12,15,18,21,24,27,30,33,36,39]
2305 self.assertEqual(expected5,connIndex.getValues());
2306 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]
2307 self.assertEqual(expected6,conn.getValues());
2310 def testIntersect2DMeshesTmp1(self):
2311 m1c=MEDCouplingCMesh.New();
2312 coordsX=DataArrayDouble.New();
2313 arrX=[ -1., 1., 2., 4. ]
2314 coordsX.setValues(arrX,4,1);
2315 m1c.setCoordsAt(0,coordsX);
2316 coordsY=DataArrayDouble.New();
2317 arrY=[ -2., 2., 4., 8. ]
2318 coordsY.setValues(arrY,4,1);
2319 m1c.setCoordsAt(1,coordsY);
2320 m1=m1c.buildUnstructured()
2321 m1bis=m1.buildPartOfMySelf([3,4,5],False)
2323 m2=m2.buildPartOfMySelf([0,1,2],False)
2324 m2.translate([0.5,0.5])
2326 m3,d1,d2=MEDCouplingUMesh.Intersect2DMeshes(m1bis,m2,1e-10)
2327 expected1=[0,0,1,1,1,2,2,2]
2328 expected2=[0,-1,0,1,-1,1,2,-1]
2329 self.assertEqual(8,d1.getNumberOfTuples());
2330 self.assertEqual(8,d2.getNumberOfTuples());
2331 self.assertEqual(8,m3.getNumberOfCells());
2332 self.assertEqual(22,m3.getNumberOfNodes());
2333 self.assertEqual(2,m3.getSpaceDimension());
2334 self.assertEqual(expected1,d1.getValues());
2335 self.assertEqual(expected2,d2.getValues());
2336 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]
2337 expected4=[0,5,12,17,22,28,33,38,44]
2338 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]
2339 self.assertEqual(44,m3.getNodalConnectivity().getNumberOfTuples());
2340 self.assertEqual(9,m3.getNodalConnectivityIndex().getNumberOfTuples());
2341 self.assertEqual(expected3,m3.getNodalConnectivity().getValues());
2342 self.assertEqual(expected4,m3.getNodalConnectivityIndex().getValues());
2344 self.assertAlmostEqual(expected5[i],m3.getCoords().getIJ(0,i),12);
2348 def testFindNodesOnLine1(self):
2349 mesh=MEDCouplingDataForTest.build2DTargetMesh_1();
2357 res=mesh.findNodesOnLine(pt,vec,1e-12);
2358 self.assertEqual(3,len(res));
2359 self.assertEqual(expected1,res.getValues());
2361 mesh.changeSpaceDimension(3);
2362 mesh.rotate(pt2,vec2,pi/4.);
2363 res=mesh.findNodesOnLine(pt3,vec3,1e-12);
2364 self.assertEqual(3,len(res));
2365 self.assertEqual(expected1,res.getValues());
2368 def testIntersect2DMeshesTmp2(self):
2369 m1c=MEDCouplingCMesh.New();
2370 coordsX1=DataArrayDouble.New();
2371 arrX1=[ 0., 1., 1.5, 2. ]
2372 coordsX1.setValues(arrX1,4,1);
2373 m1c.setCoordsAt(0,coordsX1);
2374 coordsY1=DataArrayDouble.New();
2375 arrY1=[ 0., 1.5, 3.]
2376 coordsY1.setValues(arrY1,3,1);
2377 m1c.setCoordsAt(1,coordsY1);
2378 m1=m1c.buildUnstructured();
2379 m2c=MEDCouplingCMesh.New();
2380 coordsX2=DataArrayDouble.New();
2381 arrX2=[ 0., 1., 2. ]
2382 coordsX2.setValues(arrX2,3,1);
2383 m2c.setCoordsAt(0,coordsX2);
2384 coordsY2=DataArrayDouble.New();
2386 coordsY2.setValues(arrY2,3,1);
2387 m2c.setCoordsAt(1,coordsY2);
2388 m2=m2c.buildUnstructured();
2390 m3,d1,d2=MEDCouplingUMesh.Intersect2DMeshes(m1,m2,1e-10)
2392 expected1=[0,0,1,1,2,2,3,4,5]
2393 expected2=[0,2,1,3,1,3,2,3,3]
2394 self.assertEqual(9,d1.getNumberOfTuples());
2395 self.assertEqual(9,d2.getNumberOfTuples());
2396 self.assertEqual(9,m3.getNumberOfCells());
2397 self.assertEqual(22,m3.getNumberOfNodes());
2398 self.assertEqual(2,m3.getSpaceDimension());
2399 self.assertEqual(expected1,d1.getValues());
2400 self.assertEqual(expected2,d2.getValues());
2401 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]
2402 expected4=[0,5,10,15,20,25,30,35,40,45]
2403 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]
2404 self.assertEqual(45,m3.getNodalConnectivity().getNumberOfTuples());
2405 self.assertEqual(10,m3.getNodalConnectivityIndex().getNumberOfTuples());
2406 self.assertEqual(expected3,m3.getNodalConnectivity().getValues());
2407 self.assertEqual(expected4,m3.getNodalConnectivityIndex().getValues());
2409 self.assertAlmostEqual(expected5[i],m3.getCoords().getIJ(0,i),12);
2413 def testBuildPartOfMySelfSafe1(self):
2414 mesh=MEDCouplingDataForTest.build2DTargetMesh_1()
2415 self.assertRaises(InterpKernelException,mesh.buildPartOfMySelf,[0,-1,4,2],True)
2416 self.assertRaises(InterpKernelException,mesh.buildPartOfMySelf,[0,4,5,4],True)
2419 def testIntersect2DMeshesTmp3(self):
2420 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];
2421 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];
2422 m1=MEDCouplingUMesh.New();
2423 m1.setMeshDimension(2);
2424 m1.allocateCells(8);
2425 m1.insertNextCell(NORM_TRI6,6,m1Conn[0:6]);
2426 m1.insertNextCell(NORM_QUAD8,8,m1Conn[6:14]);
2427 m1.insertNextCell(NORM_TRI6,6,m1Conn[14:20]);
2428 m1.insertNextCell(NORM_QUAD8,8,m1Conn[20:28]);
2429 m1.insertNextCell(NORM_TRI6,6,m1Conn[28:34]);
2430 m1.insertNextCell(NORM_QUAD8,8,m1Conn[34:42]);
2431 m1.insertNextCell(NORM_TRI6,6,m1Conn[42:48]);
2432 m1.insertNextCell(NORM_QUAD8,8,m1Conn[48:56]);
2433 m1.finishInsertingCells();
2434 myCoords1=DataArrayDouble.New();
2435 myCoords1.setValues(m1Coords,25,2);
2436 m1.setCoords(myCoords1);
2438 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.]
2439 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]
2440 m2=MEDCouplingUMesh.New();
2441 m2.setMeshDimension(2);
2442 m2.allocateCells(8);
2444 m2.insertNextCell(NORM_QUAD4,4,m2Conn[4*i:4*(i+1)])
2446 m2.finishInsertingCells();
2447 myCoords2=DataArrayDouble.New();
2448 myCoords2.setValues(m2Coords,15,2);
2449 m2.setCoords(myCoords2);
2451 m3,d1,d2=MEDCouplingUMesh.Intersect2DMeshes(m1,m2,1e-10)
2454 expected1=[0,1,1,1,2,3,3,3,4,5,5,5,6,7,7,7]
2455 expected2=[0,0,1,-1,2,2,3,-1,4,4,5,-1,6,6,7,-1]
2456 self.assertEqual(16,d1.getNumberOfTuples());
2457 self.assertEqual(16,d2.getNumberOfTuples());
2458 self.assertEqual(16,m3.getNumberOfCells());
2459 self.assertEqual(104,m3.getNumberOfNodes());
2460 self.assertEqual(2,m3.getSpaceDimension());
2461 self.assertEqual(expected1,d1.getValues());
2462 self.assertEqual(expected2,d2.getValues());
2463 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]
2464 expected4=[0,7,16,25,34,41,50,59,68,75,84,93,102,109,118,127,136]
2465 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]
2466 self.assertEqual(136,m3.getNodalConnectivity().getNumberOfTuples());
2467 self.assertEqual(17,m3.getNodalConnectivityIndex().getNumberOfTuples());
2468 self.assertEqual(expected3,m3.getNodalConnectivity().getValues());
2469 self.assertEqual(expected4,m3.getNodalConnectivityIndex().getValues());
2470 for i in range(208):
2471 self.assertAlmostEqual(expected5[i],m3.getCoords().getIJ(0,i),12);
2475 def testUMeshTessellate2D1(self):
2476 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];
2477 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];
2478 m1=MEDCouplingUMesh.New();
2479 m1.setMeshDimension(2);
2480 m1.allocateCells(8);
2481 m1.insertNextCell(NORM_TRI6,6,m1Conn[0:6]);
2482 m1.insertNextCell(NORM_QUAD8,8,m1Conn[6:14]);
2483 m1.insertNextCell(NORM_TRI6,6,m1Conn[14:20]);
2484 m1.insertNextCell(NORM_QUAD8,8,m1Conn[20:28]);
2485 m1.insertNextCell(NORM_TRI6,6,m1Conn[28:34]);
2486 m1.insertNextCell(NORM_QUAD8,8,m1Conn[34:42]);
2487 m1.insertNextCell(NORM_TRI6,6,m1Conn[42:48]);
2488 m1.insertNextCell(NORM_QUAD8,8,m1Conn[48:56]);
2489 m1.finishInsertingCells();
2490 myCoords1=DataArrayDouble.New();
2491 myCoords1.setValues(m1Coords,25,2);
2492 m1.setCoords(myCoords1);
2495 m11.tessellate2D(1.);
2496 self.assertTrue(m11.getCoords().isEqual(m11.getCoords(),1e-12));
2497 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]
2498 expected2=[0,5,12,17,24,29,36,41,48]
2499 self.assertEqual(48,m11.getNodalConnectivity().getNumberOfTuples());
2500 self.assertEqual(9,m11.getNodalConnectivityIndex().getNumberOfTuples());
2501 self.assertEqual(expected1,m11.getNodalConnectivity().getValues());
2502 self.assertEqual(expected2,m11.getNodalConnectivityIndex().getValues());
2505 m12.tessellate2D(0.5);
2506 self.assertEqual(41,m12.getNumberOfNodes());
2507 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]
2508 expected4=[0,6,15,21,30,36,45,51,60]
2509 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]
2511 self.assertAlmostEqual(expected5[i],m12.getCoords().getIJ(0,i),12);
2513 self.assertEqual(60,m12.getNodalConnectivity().getNumberOfTuples());
2514 self.assertEqual(9,m12.getNodalConnectivityIndex().getNumberOfTuples());
2515 self.assertEqual(expected3,m12.getNodalConnectivity().getValues());
2516 self.assertEqual(expected4,m12.getNodalConnectivityIndex().getValues());
2519 def testUMeshTessellate2DCurve1(self):
2520 # A quarter of circle:
2521 mcoords = [0.4,0.0, 0.0,-0.4, 0.283,-0.283]
2524 m1 = MEDCouplingUMesh.New()
2525 m1.setMeshDimension(1)
2527 m1.insertNextCell(NORM_SEG3, mconnec)
2529 myCoords = DataArrayDouble.New(mcoords, 3, 2)
2530 m1.setCoords(myCoords)
2533 m2.tessellate2D(0.1)
2534 # If the following raises, the test will fail automatically:
2535 m2.checkConsistency(0.0) # eps param not used
2537 def testGetCellIdsCrossingPlane1(self):
2538 mesh3D,mesh2D=MEDCouplingDataForTest.build3DExtrudedUMesh_1();
2540 origin=[1.524,1.4552,1.74768]
2541 ids1=mesh3D.getCellIdsCrossingPlane(origin,vec,1e-10)
2542 self.assertEqual([1,3,4,7,9,10,13,15,16],ids1.getValues())
2544 ids2=mesh3D.getCellIdsCrossingPlane(origin,vec2,1e-10)
2545 self.assertEqual([6,7,8,9,10,11],ids2.getValues())
2548 def testBuildSlice3D1(self):
2549 mesh3D,mesh2D=MEDCouplingDataForTest.build3DExtrudedUMesh_1();
2550 vec1=[-0.07,1.,0.07]
2551 origin1=[1.524,1.4552,1.74768]
2552 slice1,ids=mesh3D.buildSlice3D(origin1,vec1,1e-10);
2553 expected1=[1,3,4,7,9,10,13,15,16]
2554 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]
2555 expected3=[0,6,11,17,22,27,32,37,42,47]
2556 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.]
2557 self.assertEqual(2,slice1.getMeshDimension());
2558 self.assertEqual(3,slice1.getSpaceDimension());
2559 self.assertEqual(57,slice1.getNumberOfNodes());
2560 self.assertEqual(9,slice1.getNumberOfCells());
2561 self.assertEqual(9,ids.getNumberOfTuples());
2562 self.assertEqual(47,slice1.getNodalConnectivity().getNumberOfTuples());
2563 self.assertEqual(10,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2564 self.assertEqual(expected1,ids.getValues());
2565 self.assertEqual(expected2,slice1.getNodalConnectivity().getValues());
2566 self.assertEqual(expected3,slice1.getNodalConnectivityIndex().getValues());
2567 for i in range(171):
2568 self.assertAlmostEqual(expected4[i],slice1.getCoords().getIJ(0,i),12);
2570 # 2nd slice based on already existing nodes of mesh3D.
2573 slice1,ids=mesh3D.buildSlice3D(origin2,vec2,1e-10);
2574 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]
2575 expected6=[0,5,10,15,21,26,32,38,43,49]
2576 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]
2577 self.assertEqual(2,slice1.getMeshDimension());
2578 self.assertEqual(3,slice1.getSpaceDimension());
2579 self.assertEqual(60,slice1.getNumberOfNodes());
2580 self.assertEqual(9,slice1.getNumberOfCells());
2581 self.assertEqual(9,ids.getNumberOfTuples());
2582 self.assertEqual(49,slice1.getNodalConnectivity().getNumberOfTuples());
2583 self.assertEqual(10,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2584 self.assertEqual(expected1,ids.getValues());
2585 self.assertEqual(expected5,slice1.getNodalConnectivity().getValues());
2586 self.assertEqual(expected6,slice1.getNodalConnectivityIndex().getValues());
2587 for i in range(180):
2588 self.assertAlmostEqual(expected7[i],slice1.getCoords().getIJ(0,i),12);
2590 # 3rd slice based on shared face of mesh3D.
2593 slice1,ids=mesh3D.buildSlice3D(origin3,vec3,1e-10);
2594 expected8=[6,7,8,9,10,11,12,13,14,15,16,17]
2595 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]
2596 expected10=[0,5,12,17,22,29,34,39,46,51,56,63,68]
2597 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.]
2598 self.assertEqual(2,slice1.getMeshDimension());
2599 self.assertEqual(3,slice1.getSpaceDimension());
2600 self.assertEqual(45,slice1.getNumberOfNodes());
2601 self.assertEqual(12,slice1.getNumberOfCells());
2602 self.assertEqual(12,ids.getNumberOfTuples());
2603 self.assertEqual(68,slice1.getNodalConnectivity().getNumberOfTuples());
2604 self.assertEqual(13,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2605 self.assertEqual(expected8,ids.getValues());
2606 self.assertEqual(expected9,slice1.getNodalConnectivity().getValues());
2607 self.assertEqual(expected10,slice1.getNodalConnectivityIndex().getValues());
2608 for i in range(135):
2609 self.assertAlmostEqual(expected11[i],slice1.getCoords().getIJ(0,i),12);
2612 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)
2613 conn = DataArrayInt([18,0,2,4,6,1,3,5,7])
2614 connI = DataArrayInt([0,9])
2616 mesh3D_2=MEDCouplingUMesh.New();
2617 mesh3D_2.setName("3DMesh_2");
2618 mesh3D_2.setMeshDimension(3);
2619 mesh3D_2.setCoords(coo);
2620 mesh3D_2.setConnectivity(conn,connI,True);
2623 expected13=[5,5,9,8,4]
2625 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]
2627 y_cut = 0.8328692711876031
2628 ori, vec = [0.0, y_cut, 0.0], [0.0,1.0,0.0]
2629 slice1, ids = mesh3D_2.buildSlice3D(DataArrayDouble(ori,1,3), DataArrayDouble(vec,1,3), 1.0e-8)
2630 self.assertEqual(2,slice1.getMeshDimension());
2631 self.assertEqual(3,slice1.getSpaceDimension());
2632 self.assertEqual(10,slice1.getNumberOfNodes());
2633 self.assertEqual(1,slice1.getNumberOfCells());
2634 self.assertEqual(1,ids.getNumberOfTuples());
2635 self.assertEqual(5,slice1.getNodalConnectivity().getNumberOfTuples());
2636 self.assertEqual(2,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2637 self.assertEqual(expected12,ids.getValues());
2638 self.assertEqual(expected13,slice1.getNodalConnectivity().getValues());
2639 self.assertEqual(expected14,slice1.getNodalConnectivityIndex().getValues());
2641 self.assertAlmostEqual(expected15[i],slice1.getCoords().getIJ(0,i),12);
2644 def testBuildSlice3DSurf1(self):
2645 mesh3D,mesh2D=MEDCouplingDataForTest.build3DExtrudedUMesh_1();
2646 mesh2D=mesh3D.buildDescendingConnectivity()[0];
2647 vec1=[-0.07,1.,0.07]
2648 origin1=[1.524,1.4552,1.74768]
2649 slice1,ids=mesh2D.buildSlice3DSurf(origin1,vec1,1e-10);
2650 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]
2651 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]
2652 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];
2653 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.]
2654 self.assertEqual(1,slice1.getMeshDimension());
2655 self.assertEqual(3,slice1.getSpaceDimension());
2656 self.assertEqual(57,slice1.getNumberOfNodes());
2657 self.assertEqual(25,slice1.getNumberOfCells());
2658 self.assertEqual(25,ids.getNumberOfTuples());
2659 self.assertEqual(75,slice1.getNodalConnectivity().getNumberOfTuples());
2660 self.assertEqual(26,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2661 self.assertEqual(expected1,ids.getValues());
2662 self.assertEqual(expected2,slice1.getNodalConnectivity().getValues());
2663 self.assertEqual(expected3,slice1.getNodalConnectivityIndex().getValues());
2664 for i in range(171):
2665 self.assertAlmostEqual(expected4[i],slice1.getCoords().getIJ(0,i),12);
2670 slice1,ids=mesh2D.buildSlice3DSurf(origin2,vec2,1e-10);
2671 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]
2672 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]
2673 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];
2674 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.]
2675 self.assertEqual(1,slice1.getMeshDimension());
2676 self.assertEqual(3,slice1.getSpaceDimension());
2677 self.assertEqual(45,slice1.getNumberOfNodes());
2678 self.assertEqual(68,slice1.getNumberOfCells());
2679 self.assertEqual(68,ids.getNumberOfTuples());
2680 self.assertEqual(204,slice1.getNodalConnectivity().getNumberOfTuples());
2681 self.assertEqual(69,slice1.getNodalConnectivityIndex().getNumberOfTuples());
2682 self.assertEqual(expected5,ids.getValues());
2683 self.assertEqual(expected6,slice1.getNodalConnectivity().getValues());
2684 self.assertEqual(expected7,slice1.getNodalConnectivityIndex().getValues());
2685 for i in range(135):
2686 self.assertAlmostEqual(expected8[i],slice1.getCoords().getIJ(0,i),12);
2690 def testDataArrayDoubleAdvSetting1(self):
2691 data1=[1.,11.,2.,12.,3.,13.,4.,14.,5.,15.,6.,16.,7.,17.]
2692 data2=[8.,38.,9.,39.,0.,30.,11.,41.,12.,42.]
2693 compsCpp=["comp1","comp2"]
2694 da=DataArrayDouble.New();
2695 da.setInfoAndChangeNbOfCompo(compsCpp);
2698 compsCpp=compsCpp[:-1]
2699 self.assertRaises(InterpKernelException,da.setInfoAndChangeNbOfCompo,compsCpp);
2700 da.setValues(data1,7,2)
2702 p=[(0,3),(3,5),(5,7)]
2703 tmp=da.selectByTupleRanges(p);
2704 self.assertTrue(tmp.isEqual(da,1e-14));
2705 p=[(0,2),(3,4),(5,7)]
2706 tmp=da.selectByTupleRanges(p);
2707 expected1=[1.,11.,2.,12.,4.,14.,6.,16.,7.,17.]
2708 self.assertEqual(5,tmp.getNumberOfTuples());
2709 self.assertEqual(2,tmp.getNumberOfComponents());
2711 self.assertAlmostEqual(expected1[i],tmp.getIJ(0,i),14);
2713 p=[(0,2),(0,2),(5,6)]
2714 tmp=da.selectByTupleRanges(p);
2715 expected2=[1.,11.,2.,12.,1.,11.,2.,12.,6.,16.]
2716 self.assertEqual(5,tmp.getNumberOfTuples());
2717 self.assertEqual(2,tmp.getNumberOfComponents());
2719 self.assertAlmostEqual(expected2[i],tmp.getIJ(0,i),14);
2721 p=[(0,2),(-1,2),(5,6)]
2722 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2723 p=[(0,2),(0,2),(5,8)]
2724 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2726 da2=DataArrayDouble.New();
2727 da2.setValues(data2,5,2);
2730 dac.setContigPartOfSelectedValuesSlice(1,da2,2,4,1);
2731 expected3=[1.,11.,0.,30.,11.,41.,4.,14.,5.,15.,6.,16.,7.,17.]
2733 self.assertAlmostEqual(expected3[i],dac.getIJ(0,i),14);
2737 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,0,5,1);
2738 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,0,da2,4,6,1);
2739 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,5,0,1);
2740 dac.setContigPartOfSelectedValuesSlice(3,da2,1,5,1);
2741 expected4=[1.,11.,2.,12.,3.,13.,9.,39.,0.,30.,11.,41.,12.,42.]
2743 self.assertAlmostEqual(expected4[i],dac.getIJ(0,i),14);
2746 ids=DataArrayInt.New();
2749 ids.setIJ(0,0,2); ids.setIJ(1,0,0); ids.setIJ(2,0,4);
2750 dac.setContigPartOfSelectedValues(2,da2,ids);
2751 expected5=[1.,11.,2.,12.,0.,30.,8.,38.,12.,42.,6.,16.,7.,17.]
2753 self.assertAlmostEqual(expected5[i],dac.getIJ(0,i),14);
2757 ids.setIJ(0,0,2); ids.setIJ(1,0,5); ids.setIJ(2,0,4);
2758 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2759 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,-1);
2760 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2761 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2762 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,5,da2,ids);
2764 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2766 dac.setContigPartOfSelectedValues(4,da2,ids);
2767 expected6=[1.,11.,2.,12.,3.,13.,4.,14.,0.,30.,0.,30.,9.,39.]
2769 self.assertAlmostEqual(expected6[i],dac.getIJ(0,i),14);
2773 def testDataArrayIntAdvSetting1(self):
2774 data1=[1,11,2,12,3,13,4,14,5,15,6,16,7,17]
2775 data2=[8,38,9,39,0,30,11,41,12,42]
2776 compsCpp=["comp1","comp2"]
2777 da=DataArrayInt.New();
2778 da.setInfoAndChangeNbOfCompo(compsCpp);
2781 compsCpp=compsCpp[:-1]
2782 self.assertRaises(InterpKernelException,da.setInfoAndChangeNbOfCompo,compsCpp);
2783 da.setValues(data1,7,2)
2785 p=[(0,3),(3,5),(5,7)]
2786 tmp=da.selectByTupleRanges(p);
2787 self.assertTrue(tmp.isEqual(da));
2788 p=[(0,2),(3,4),(5,7)]
2789 tmp=da.selectByTupleRanges(p);
2790 expected1=[1,11,2,12,4,14,6,16,7,17]
2791 self.assertEqual(5,tmp.getNumberOfTuples());
2792 self.assertEqual(2,tmp.getNumberOfComponents());
2794 self.assertEqual(expected1[i],tmp.getIJ(0,i));
2796 p=[(0,2),(0,2),(5,6)]
2797 tmp=da.selectByTupleRanges(p);
2798 expected2=[1,11,2,12,1,11,2,12,6,16]
2799 self.assertEqual(5,tmp.getNumberOfTuples());
2800 self.assertEqual(2,tmp.getNumberOfComponents());
2802 self.assertEqual(expected2[i],tmp.getIJ(0,i));
2804 p=[(0,2),(-1,2),(5,6)]
2805 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2806 p=[(0,2),(0,2),(5,8)]
2807 self.assertRaises(InterpKernelException,da.selectByTupleRanges,p);
2809 da2=DataArrayInt.New();
2810 da2.setValues(data2,5,2);
2813 dac.setContigPartOfSelectedValuesSlice(1,da2,2,4,1);
2814 expected3=[1,11,0,30,11,41,4,14,5,15,6,16,7,17]
2816 self.assertEqual(expected3[i],dac.getIJ(0,i));
2820 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,0,5,1);
2821 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,0,da2,4,6,1);
2822 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValuesSlice,3,da2,5,0,1);
2823 dac.setContigPartOfSelectedValuesSlice(3,da2,1,5,1);
2824 expected4=[1,11,2,12,3,13,9,39,0,30,11,41,12,42]
2826 self.assertEqual(expected4[i],dac.getIJ(0,i));
2829 ids=DataArrayInt.New();
2832 ids.setIJ(0,0,2); ids.setIJ(1,0,0); ids.setIJ(2,0,4);
2833 dac.setContigPartOfSelectedValues(2,da2,ids);
2834 expected5=[1,11,2,12,0,30,8,38,12,42,6,16,7,17]
2836 self.assertEqual(expected5[i],dac.getIJ(0,i));
2840 ids.setIJ(0,0,2); ids.setIJ(1,0,5); ids.setIJ(2,0,4);
2841 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2842 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,-1);
2843 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,1,da2,ids);
2844 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2845 self.assertRaises(InterpKernelException,dac.setContigPartOfSelectedValues,5,da2,ids);
2847 ids.setIJ(0,0,2); ids.setIJ(1,0,2); ids.setIJ(2,0,1);
2849 dac.setContigPartOfSelectedValues(4,da2,ids);
2850 expected6=[1,11,2,12,3,13,4,14,0,30,0,30,9,39]
2852 self.assertEqual(expected6[i],dac.getIJ(0,i));
2856 def testBuildDescendingConnec2Of3DMesh1(self):
2857 mesh=MEDCouplingDataForTest.build3DSourceMesh_1();
2859 mesh2,desc,descIndx,revDesc,revDescIndx=mesh.buildDescendingConnectivity2();
2860 mesh2.checkConsistencyLight();
2861 self.assertEqual(2,mesh2.getMeshDimension());
2862 self.assertEqual(30,mesh2.getNumberOfCells());
2863 self.assertEqual(31,revDescIndx.getNbOfElems()); self.assertEqual(31,revDescIndx.getNumberOfTuples());
2864 self.assertEqual(13,descIndx.getNbOfElems()); self.assertEqual(13,descIndx.getNumberOfTuples());
2865 self.assertEqual(48,desc.getNbOfElems()); self.assertEqual(48,desc.getNumberOfTuples());
2866 self.assertEqual(48,revDesc.getNbOfElems()); self.assertEqual(48,revDesc.getNumberOfTuples());
2867 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]
2868 self.assertEqual(expected1,desc.getValues());
2869 expected2=[0,4,8,12,16,20,24,28,32,36,40,44,48]
2870 self.assertEqual(expected2,descIndx.getValues());
2871 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]
2872 self.assertEqual(expected3,revDescIndx.getValues());
2873 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]
2874 self.assertEqual(expected4,revDesc.getValues());
2875 conn=mesh2.getNodalConnectivity();
2876 connIndex=mesh2.getNodalConnectivityIndex();
2877 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]
2878 self.assertEqual(expected5,connIndex.getValues());
2879 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]
2880 self.assertEqual(expected6,conn.getValues());
2883 def testAre2DCellsNotCorrectlyOriented1(self):
2884 m1Coords=[1.,1.,-1.,-1.,-1.,-1.,1.,-1.]
2886 m1=MEDCouplingUMesh.New();
2887 m1.setMeshDimension(2);
2888 m1.allocateCells(1);
2889 m1.insertNextCell(NORM_QUAD4,4,m1Conn[0:4])
2890 m1.finishInsertingCells();
2891 myCoords1=DataArrayDouble.New();
2892 myCoords1.setValues(m1Coords,4,2);
2893 m1.setCoords(myCoords1);
2897 vec2=[3.*cos(pi/9.*i),3.*sin(pi/9.*i)];
2898 m1Cpy=m1.deepCopy();
2899 m1Cpy.translate(vec2);
2900 self.assertRaises(InterpKernelException,m1Cpy.are2DCellsNotCorrectlyOriented,vec1,False);
2901 m1Cpy.changeSpaceDimension(3);
2902 res=m1Cpy.are2DCellsNotCorrectlyOriented(vec1,False)
2903 self.assertEqual([0],res.getValues());
2907 def testDataArrayAbs1(self):
2908 d1=DataArrayDouble.New();
2909 val1=[2.,-3.,-5.,6.,-7.,-8.,9.,10.,-11.,-12.,-13.,-15.]
2910 expected1=[2.,3.,5.,6.,7.,8.,9.,10.,11.,12.,13.,15.]
2911 d1.setValues(val1,6,2);
2912 d2=d1.convertToIntArr();
2916 self.assertAlmostEqual(expected1[i],d1.getIJ(0,i),14);
2919 expected2=[2,3,5,6,7,8,9,10,11,12,13,15]
2922 self.assertEqual(expected2[i],d2.getIJ(0,i));
2928 def testGetValueOn3(self):
2930 v2=[0.7,1.25,0.,2.,1.5]
2931 disp=[5.,50.,500.,6.,60.,600.,7.,70.,700.,8.,80.,800.]
2932 m=MEDCouplingUMesh.New("myMesh",1)
2935 m.allocateCells(nbCells)
2936 coords=DataArrayDouble.New() ; coords.setValues(v,nbNodes,1)
2938 m.insertNextCell(NORM_SEG2,2,[0,1])
2939 m.insertNextCell(NORM_SEG2,2,[2,1])
2940 m.insertNextCell(NORM_SEG2,2,[2,3])
2941 m.finishInsertingCells()
2942 f=MEDCouplingFieldDouble.New(ON_NODES)
2944 array=DataArrayDouble.New(); array.setValues(disp,m.getNumberOfNodes(),3)
2946 arr1=f.getValueOnMulti(v2)
2947 self.assertEqual(5,arr1.getNumberOfTuples());
2948 self.assertEqual(3,arr1.getNumberOfComponents());
2949 expected1=[5.7,57.,570.,6.5,65.,650.,5.,50.,500.,8.,80.,800.,7.,70.,700.]
2951 self.assertAlmostEqual(expected1[i],arr1.getIJ(0,i),14);
2955 def testGetNodeIdsOfCell2(self):
2956 m1c=MEDCouplingCMesh.New();
2957 coordsX=DataArrayDouble.New();
2958 arrX=[ -1., 1., 2., 4., 4.5 ]
2959 coordsX.setValues(arrX,5,1);
2960 coordsY=DataArrayDouble.New();
2961 arrY=[ -2., 2., 4., 8.]
2962 coordsY.setValues(arrY,4,1);
2963 coordsZ=DataArrayDouble.New();
2965 coordsZ.setValues(arrZ,3,1);
2967 m1c.setCoordsAt(0,coordsX);
2968 expected1=[[0,1],[1,2],[2,3],[3,4]]
2969 self.assertEqual(4,m1c.getNumberOfCells())
2970 for i in range(m1c.getNumberOfCells()):
2971 self.assertEqual(expected1[i],m1c.getNodeIdsOfCell(i))
2974 m1c.setCoordsAt(1,coordsY);
2975 self.assertEqual(12,m1c.getNumberOfCells())
2976 self.assertEqual(20,m1c.getNumberOfNodes())
2977 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]]
2978 for i in range(m1c.getNumberOfCells()):
2979 self.assertEqual(expected2[i],m1c.getNodeIdsOfCell(i))
2982 m1c.setCoordsAt(2,coordsZ);
2983 self.assertEqual(24,m1c.getNumberOfCells())
2984 self.assertEqual(60,m1c.getNumberOfNodes())
2985 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]]
2986 self.assertEqual(24,m1c.getNumberOfCells())
2987 for i in range(m1c.getNumberOfCells()):
2988 self.assertEqual(expected3[i],m1c.getNodeIdsOfCell(i))
2994 if __name__ == '__main__':