Salome HOME
Updated copyright comment
[tools/medcoupling.git] / src / MEDLoader / Swig / MEDLoaderExamplesTest.py
index 24ee31a8cf8b7bacc8c54dc1f654554ff6253ca2..ec6c77e75d2b5c6629b4aeff2f719c34044d7a47 100644 (file)
@@ -1,10 +1,10 @@
 #  -*- coding: iso-8859-1 -*-
-# Copyright (C) 2007-2012  CEA/DEN, EDF R&D
+# Copyright (C) 2007-2024  CEA, EDF
 #
 # This library is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
 # License as published by the Free Software Foundation; either
-# version 2.1 of the License.
+# version 2.1 of the License, or (at your option) any later version.
 #
 # This library is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 
 from MEDLoader import *
 import unittest
+import os
+from MEDLoaderDataForTest import WriteInTmpDir
 
 class MEDLoaderBasicsTest(unittest.TestCase):
+    @WriteInTmpDir
     def testExampleReadFieldOnAllEntity1(self):
         from MEDLoaderDataForTest import MEDLoaderDataForTest
 #! [PySnippetReadFieldOnAllEntity1_1]
@@ -36,7 +39,7 @@ class MEDLoaderBasicsTest(unittest.TestCase):
         m=MEDLoaderDataForTest.build2DMesh_3()
         m=m[:10]
         m.setName(meshName)
-        f=m.getMeasureField(ON_CELLS)
+        f=m.getMeasureField(False)
         f=f.buildNewTimeReprFromThis(ONE_TIME,False)
         f.setTime(5.5,iteration,order)
         f.setName(fieldName)
@@ -77,6 +80,7 @@ class MEDLoaderBasicsTest(unittest.TestCase):
 #! [PySnippetReadFieldOnAllEntity1_5]
         pass
 
+    @WriteInTmpDir
     def testExampleReadFieldPartial1(self):
         from MEDLoaderDataForTest import MEDLoaderDataForTest
 #! [PySnippetReadFieldPartial1_1]
@@ -91,7 +95,7 @@ class MEDLoaderBasicsTest(unittest.TestCase):
         m.sortCellsInMEDFileFrmt()
         m.setName(meshName)
         # end of generation of a mesh -> let's create a field on that mesh
-        f=m.getMeasureField(ON_CELLS)
+        f=m.getMeasureField(False)
         f=f.buildNewTimeReprFromThis(ONE_TIME,False)
         f.setTime(5.5,iteration,order)
         f.setName(fieldName)
@@ -154,6 +158,7 @@ class MEDLoaderBasicsTest(unittest.TestCase):
 #! [PySnippetReadFieldPartial1_6]
         pass
 
+    @WriteInTmpDir
     def testExampleMeshAdvAPI1(self):
         da=DataArrayDouble.New([0.,1.1,2.3,3.6])
         meshName="Example2"
@@ -162,30 +167,31 @@ class MEDLoaderBasicsTest(unittest.TestCase):
 #! [PySnippetMeshAdvAPI1_1]
         self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
         myMesh.setName(meshName)
-        MEDLoader.WriteUMesh("wFile1.med",myMesh,True)
+        WriteUMesh("wFile1.med",myMesh,True)
 #! [PySnippetMeshAdvAPI1_1]
+        os.remove("wFile1.med")
 #! [PySnippetMeshAdvAPI1_2]
         self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
         myMesh.setName(meshName)
-        MEDLoader.WriteUMesh("wFile1.med",myMesh,True)
+        WriteUMesh("wFile1.med",myMesh,False)
 #! [PySnippetMeshAdvAPI1_2]
-        f=myMesh.getMeasureField(ON_CELLS)
+        f=myMesh.getMeasureField(False)
         f=f.buildNewTimeReprFromThis(ONE_TIME,False)
         f.setName("myField")
 #! [PySnippetMeshAdvAPI1_3]
-        MEDLoader.WriteUMesh("file3.med",f.getMesh(),True)
+        WriteUMesh("file3.med",f.getMesh(),True)
         f.setTime(1.2,1,0)
         fileNameMultiTimeStep="file3.med"
-        MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
+        WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
         f.setTime(1.3,2,0)
         f.applyFunc("sqrt(x)");
         #Writing second time step with iteration==2 and order==0
-        MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file3.med",f);
+        WriteFieldUsingAlreadyWrittenMesh("file3.med",f);
 #! [PySnippetMeshAdvAPI1_3]
 #! [PySnippetMeshAdvAPI1_11]
-        timeStepsIds=MEDLoader.GetCellFieldIterations("file3.med","Example2","myField")
+        timeStepsIds=GetCellFieldIterations("file3.med","Example2","myField")
         self.assertEqual([(1, 0),(2, 0)],timeStepsIds)
-        fs=MEDLoader.ReadFieldsOnSameMesh(ON_CELLS,"file3.med","Example2",0,"myField",timeStepsIds);
+        fs=ReadFieldsOnSameMesh(ON_CELLS,"file3.med","Example2",0,"myField",timeStepsIds);
 #! [PySnippetMeshAdvAPI1_11]
         ###
         myMesh0=myMesh[:] ; myMesh0.setName("Example2")
@@ -203,34 +209,34 @@ class MEDLoaderBasicsTest(unittest.TestCase):
         F1Cell.setArray(myMesh0.getCoords()[:myMesh0.getNumberOfCells()])
         F1Cell.setTime(1000.,2,3)
         F1Cell.setName("F1Cell")
-        MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell)
-        F1Cell1=F1Cell.deepCpy()
+        WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell)
+        F1Cell1=F1Cell.deepCopy()
         F1Cell1.setMesh(myMesh1)
-        F1Cell1.setArray(myMesh1.getBarycenterAndOwner())
-        MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell1)
+        F1Cell1.setArray(myMesh1.computeCellCenterOfMass())
+        WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell1)
 #! [PySnippetMeshAdvAPI1_12]
-        f1Cell_3D=MEDLoader.ReadFieldCell("file2.med","Example2",0,"F1Cell",2,3)
+        f1Cell_3D=ReadFieldCell("file2.med","Example2",0,"F1Cell",2,3)
 #! [PySnippetMeshAdvAPI1_12]
 #! [PySnippetMeshAdvAPI1_13]
-        f1Cell_2D=MEDLoader.ReadFieldCell("file2.med","Example2",-1,"F1Cell",2,3)
+        f1Cell_2D=ReadFieldCell("file2.med","Example2",-1,"F1Cell",2,3)
 #! [PySnippetMeshAdvAPI1_13]
         self.assertTrue(F1Cell.isEqual(f1Cell_3D,1e-12,1e-12))
 #! [PySnippetMeshAdvAPI1_8]
-        self.assertEqual(3,MEDLoader.ReadUMeshDimFromFile("file2.med","Example2"))
+        self.assertEqual(3,ReadUMeshDimFromFile("file2.med","Example2"))
 #! [PySnippetMeshAdvAPI1_8]
 #! [PySnippetMeshAdvAPI1_7]
-        m2D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",0)
+        m2D=ReadUMeshFromFile("file2.med","Example2",0)
 #! [PySnippetMeshAdvAPI1_7]
 #! [PySnippetMeshAdvAPI1_4]
-        m2D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-1)
+        m2D=ReadUMeshFromFile("file2.med","Example2",-1)
 #! [PySnippetMeshAdvAPI1_4]
 #! [PySnippetMeshAdvAPI1_5]
-        m1D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-2)
+        m1D=ReadUMeshFromFile("file2.med","Example2",-2)
 #! [PySnippetMeshAdvAPI1_5]
 #! [PySnippetMeshAdvAPI1_6]
-        m0D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-3)
+        m0D=ReadUMeshFromFile("file2.med","Example2",-3)
 #! [PySnippetMeshAdvAPI1_6]
-        for i in xrange(4):
+        for i in range(4):
             mm.removeMeshAtLevel(-i)
             pass
         mm.setMeshAtLevel(0,myMesh1)
@@ -238,10 +244,10 @@ class MEDLoaderBasicsTest(unittest.TestCase):
         mm.setName("MyMesh")
         mm.write("file1.med",2)
 #! [PySnippetMeshAdvAPI1_9]
-        m2D=MEDLoader.ReadUMeshFromFile("file1.med","MyMesh",0)
+        m2D=ReadUMeshFromFile("file1.med","MyMesh",0)
 #! [PySnippetMeshAdvAPI1_9]
 #! [PySnippetMeshAdvAPI1_10]
-        m1D=MEDLoader.ReadUMeshFromFile("file1.med","MyMesh",-1)
+        m1D=ReadUMeshFromFile("file1.med","MyMesh",-1)
 #! [PySnippetMeshAdvAPI1_10]
         pass