Porting 4.2. Tests OK \!

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

diff --git a/test/demo12.py b/test/demo12.py
index 920b9a27e553406a7573ea4434214ea32bbc1790..620822cb5251757076ed059a1ce6948944231bf9 100755 (executable)
--- a/test/demo12.py
+++ b/test/demo12.py
@@ -1,9 +1,9 @@
-# Copyright (C) 2010-2013  CEA/DEN, EDF R&D
+# Copyright (C) 2010-2014  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
 # 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
@@ -21,7 +21,7 @@
 if not ('servermanager' in dir()):
   from pvsimple import *
 
-import os
+import os, inspect
 
 def demo12(fname, impth):
     """This method demonstrates the Stream Tracer."""
@@ -49,8 +49,10 @@ def demo12(fname, impth):
     Render()
     WriteImage(filename = (impth + "streamTracer_1.png"), view=v, Magnification=2)
 
-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__":
- demo12(fname=pvdata+"/Data/disk_out_ref.ex2", impth=testdir+"/Pic/")
+    demo12(fname=pvdata+"/Data/disk_out_ref.ex2", impth=testdir+"/Pic/")