Salome HOME
[EDF19821] : tests are writing tmp file into /tmp
[modules/paravis.git] / src / Plugins / MEDReader / Test / testMEDReader9.py
index 11de06553cfc63724cad1562f0e47bc7eaa753d2..4945528f8e537ce145efee7fcb839bbd52b2f63b 100644 (file)
 #
 # Author : Anthony Geay
 
-from MEDLoader import *
-
-""" This test is a non regression test. Test that ExtractGroup adapt to the current mesh of MEDReader. A MED file containing 2 fields on a 2 different meshes.
-This test checks that the MeshName follows.
-"""
-
-fname="testMEDReader9.med"
-#########
-m1=MEDCouplingUMesh("m1",2) ; m1.setCoords(DataArrayDouble([(0.,0.),(1.,0.),(0.,1.)])) ; m1.allocateCells() ; m1.insertNextCell(NORM_TRI3,[0,1,2])
-m2=MEDCouplingUMesh("m2",2) ; m2.setCoords(DataArrayDouble([(0.,0.),(1.,0.),(0.,1.),(1.,1.)])) ; m2.allocateCells() ; m2.insertNextCell(NORM_QUAD4,[0,1,3,2])
-mm1=MEDFileUMesh() ; mm2=MEDFileUMesh()
-mm1.setMeshAtLevel(0,m1) ; mm2.setMeshAtLevel(0,m2)
-ms=MEDFileMeshes() ; ms.pushMesh(mm1) ; ms.pushMesh(mm2)
-f1=MEDCouplingFieldDouble(ON_CELLS) ; f1.setName("f1") ; f1.setMesh(m1) ; f1.setArray(DataArrayDouble([1.111]))
-f2=MEDCouplingFieldDouble(ON_CELLS) ; f2.setName("f2") ; f2.setMesh(m2) ; f2.setArray(DataArrayDouble([2.2222]))
-ff1=MEDFileField1TS() ; ff2=MEDFileField1TS()
-ff1.setFieldNoProfileSBT(f1) ; ff2.setFieldNoProfileSBT(f2)
-ms.write(fname,2)
-ff1.write(fname,0)
-ff2.write(fname,0)
-################### MED write is done -> Go to MEDReader
+from medcoupling import *
 from paraview.simple import *
+from MEDReaderHelper import WriteInTmpDir,RetriveBaseLine
+
+def GenerateCase():
+    """ This test is a non regression test. Test that ExtractGroup adapt to the current mesh of MEDReader. A MED file containing 2 fields on a 2 different meshes.
+    This test checks that the MeshName follows.
+    """
+    fname="testMEDReader9.med"
+    #########
+    m1=MEDCouplingUMesh("m1",2) ; m1.setCoords(DataArrayDouble([(0.,0.),(1.,0.),(0.,1.)])) ; m1.allocateCells() ; m1.insertNextCell(NORM_TRI3,[0,1,2])
+    m2=MEDCouplingUMesh("m2",2) ; m2.setCoords(DataArrayDouble([(0.,0.),(1.,0.),(0.,1.),(1.,1.)])) ; m2.allocateCells() ; m2.insertNextCell(NORM_QUAD4,[0,1,3,2])
+    mm1=MEDFileUMesh() ; mm2=MEDFileUMesh()
+    mm1.setMeshAtLevel(0,m1) ; mm2.setMeshAtLevel(0,m2)
+    ms=MEDFileMeshes() ; ms.pushMesh(mm1) ; ms.pushMesh(mm2)
+    f1=MEDCouplingFieldDouble(ON_CELLS) ; f1.setName("f1") ; f1.setMesh(m1) ; f1.setArray(DataArrayDouble([1.111]))
+    f2=MEDCouplingFieldDouble(ON_CELLS) ; f2.setName("f2") ; f2.setMesh(m2) ; f2.setArray(DataArrayDouble([2.2222]))
+    ff1=MEDFileField1TS() ; ff2=MEDFileField1TS()
+    ff1.setFieldNoProfileSBT(f1) ; ff2.setFieldNoProfileSBT(f2)
+    ms.write(fname,2)
+    ff1.write(fname,0)
+    ff2.write(fname,0)
+    return fname
 
-myMedReader=MEDReader(FileName=fname)
-myMedReader.AllArrays=['TS0/m1/ComSup0/f1@@][@@P0']
-#myMedReader.UpdatePipelineInformation()
-ExtractGroup1=ExtractGroup(Input=myMedReader)
-#ExtractGroup1.UpdatePipelineInformation()
-assert(ExtractGroup1.GetProperty("MeshName")[0]=="m1")
-myMedReader.AllArrays=['TS0/m2/ComSup0/f2@@][@@P0']
-#myMedReader.UpdatePipeline()
-ExtractGroup2=ExtractGroup(Input=myMedReader)
-#ExtractGroup2.UpdatePipelineInformation()
-assert(ExtractGroup2.GetProperty("MeshName")[0]=="m2")
+@WriteInTmpDir
+def test():
+    fname = GenerateCase()
+    ################### MED write is done -> Go to MEDReader
+    myMedReader=MEDReader(FileName=fname)
+    myMedReader.AllArrays=['TS0/m1/ComSup0/f1@@][@@P0']
+    #myMedReader.UpdatePipelineInformation()
+    ExtractGroup1=ExtractGroup(Input=myMedReader)
+    #ExtractGroup1.UpdatePipelineInformation()
+    assert(ExtractGroup1.GetProperty("MeshName")[0]=="m1")
+    myMedReader.AllArrays=['TS0/m2/ComSup0/f2@@][@@P0']
+    #myMedReader.UpdatePipeline()
+    ExtractGroup2=ExtractGroup(Input=myMedReader)
+    #ExtractGroup2.UpdatePipelineInformation()
+    assert(ExtractGroup2.GetProperty("MeshName")[0]=="m2")
+    
+if __name__ == "__main__":
+    test()