A forgotten C++ test

[tools/medcoupling.git] / src / MEDLoader / Swig / MEDLoaderCouplingTrainingSession.py

diff --git a/src/MEDLoader/Swig/MEDLoaderCouplingTrainingSession.py b/src/MEDLoader/Swig/MEDLoaderCouplingTrainingSession.py
index 668b56260484b614556016d1399b4938da34c181..730772680b27ae39f79e1275d27503e62de6978c 100644 (file)
--- a/src/MEDLoader/Swig/MEDLoaderCouplingTrainingSession.py
+++ b/src/MEDLoader/Swig/MEDLoaderCouplingTrainingSession.py
@@ -480,12 +480,12 @@ print "IntegralGlobConstraint %lf == %lf"%(srcField.getArray().accumulate()[0],t
 from numpy import *
 from math import acos
 
-med_root_dir=os.getenv("MEDTOOL_ROOT_DIR")
+med_root_dir=os.getenv("MEDCOUPLING_ROOT_DIR")
 if med_root_dir:
-  agitateur_file=os.path.join(os.getenv("MEDTOOL_ROOT_DIR"),"share","resources","med","agitateur.med")
-else:
-  current_dir = os.path.dirname(os.path.realpath(__file__))
-  agitateur_file=os.path.join(current_dir, "..", "..", "..", "resources","agitateur.med")
+  agitateur_file = os.path.join(os.getenv("MEDCOUPLING_ROOT_DIR"),"share","resources","med","agitateur.med")
+  if not os.path.exists(agitateur_file):
+    current_dir = os.path.dirname(os.path.realpath(__file__))
+    agitateur_file=os.path.join(current_dir, "..", "..", "..", "resources","agitateur.med")
   pass
 data=MEDFileData(agitateur_file)
 ts=data.getFields()[0].getTimeSteps()