Small refactor.

[modules/paravis.git] / test / demo5.py

diff --git a/test/demo5.py b/test/demo5.py
index 4f1da0a425a2101d84d2e8d322e7486cf945c2a9..d91d1f66fcb92647ad492eda9d790e8b3c42a2fd 100755 (executable)
--- a/test/demo5.py
+++ b/test/demo5.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2010-2014  CEA/DEN, EDF R&D
+# Copyright (C) 2010-2015  CEA/DEN, EDF R&D
 #
 # This library is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
@@ -21,7 +21,7 @@
 if not ('servermanager' in dir()):
   from pvsimple import *
 
-import os
+import os, inspect
 
 def LoadMultipleFiles(FilePath, FilePrefix):
 
@@ -65,8 +65,10 @@ def LoadMultipleFiles(FilePath, FilePrefix):
   Render()
 
 
-testdir = os.getenv("TESTDIR")
-pvdata = os.getenv("PVDATA")
+
+scriptdir = inspect.getframeinfo(inspect.currentframe())[0]
+testdir = os.path.dirname( os.path.abspath(scriptdir) )
+pvdata = os.getenv("SAMPLES_SRC_DIR")
 
 if __name__ == "__main__":
   LoadMultipleFiles(FilePath=pvdata+"/Data/dualSphereAnimation/", FilePrefix="dualSphereAnimation_P00T000")