Update tests to run on Windows platform.

[modules/smesh.git] / doc / salome / examples / testme.py

diff --git a/doc/salome/examples/testme.py b/doc/salome/examples/testme.py
index 510a408f8d5022a28898ee218f664512fbd90834..1e0dfb19b113b59a0ffe4e060506fd48a128ff80 100755 (executable)
--- a/doc/salome/examples/testme.py
+++ b/doc/salome/examples/testme.py
@@ -24,11 +24,21 @@ import unittest, sys, os
 class SalomeSession(object):
     def __init__(self, script):
         import runSalome
-        sys.argv  = ["runSalome.py"]
+        run_script = "runSalome.py"
+        if sys.platform == 'win32':
+            module_variable = "KERNEL_ROOT_DIR"
+            if os.environ.has_key(module_variable):
+                run_script = os.environ[module_variable] + "\\bin\\salome\\" + run_script
+        sys.argv  = [run_script]
         sys.argv += ["--terminal"]
         sys.argv += ["--modules=GEOM,MED,SMESH"]
         sys.argv += ["%s" % script]
+        if sys.platform == 'win32':
+            main_module_path = sys.modules['__main__'].__file__
+            sys.modules['__main__'].__file__ = ''
         clt, d = runSalome.main()
+        if sys.platform == 'win32':
+            sys.modules['__main__'].__file__ = main_module_path
         return
 
     def __del__(self):