src/Plugins/MEDReader/plugin/Test/MEDReaderHelper.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2019-2021  CEA/DEN, EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License, or (at your option) any later version.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 # Author : Anthony Geay (EDF R&D)
  21
  22 def WriteInTmpDir(func):
  23     def decoratedFunc(*args,**kwargs):
  24         import tempfile,os
  25         ret = None
  26         with tempfile.TemporaryDirectory() as tmpdirname:
  27             os.chdir(tmpdirname)
  28             ret = func(*args,**kwargs)
  29             pass
  30         return ret
  31     return decoratedFunc
  32
  33 def RetriveBaseLine(imgFile):
  34     import os,sys
  35     try:
  36         baselineIndex = sys.argv.index('-B')+1
  37         baselinePath = sys.argv[baselineIndex]
  38     except:
  39         print("Could not get baseline directory. Test failed.")
  40         exit(1)
  41     baseline_file = os.path.join(baselinePath, imgFile)
  42     return os.path.abspath(baseline_file)
  43