pydefx 9.14.0__py3-none-any.whl

salome/bin/salome/test/pyexample/test_insitu.py

@@ -0,0 +1,60 @@
+import unittest
+import insitu.insitumanager
+import os
+import time
+
+
+class TestYdefx(unittest.TestCase):
+  def test_insitu(self):
+    """
+    This test shows how to use insitu processing.
+    """
+    import pydefx
+
+    myParams = pydefx.Parameters()
+    myParams.configureResource("localhost")
+    mywd = os.path.join(myParams.salome_parameters.work_directory,
+                        "insitu_test" +
+                        time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()))
+    myParams.salome_parameters.work_directory = mywd
+    myParams.createResultDirectory("/tmp")
+
+    pyScript = """
+def _exec(x):
+  with open("mydata.txt") as f:
+    mydata = f.read()
+  intdata = int(mydata)
+  y = x + intdata
+  return y"""
+
+    myScript = pydefx.PyScript()
+    myScript.loadString(pyScript)
+
+    mySample = myScript.CreateEmptySample()
+    mydata = {"x":list(range(10))}
+    mySample.setInputValues(mydata)
+
+    # pre-processing script called before the first evaluation
+    myPrescript = """
+with open("mydata.txt", "w") as f:
+  f.write("1")
+"""
+    mySchemaBuilder = pydefx.DefaultSchemaBuilder(myPrescript)
+
+    mySampleManager = insitu.insitumanager.InsituManager()
+
+    myStudy = pydefx.PyStudy(sampleManager=mySampleManager,
+                             schemaBuilder=mySchemaBuilder)
+    myStudy.createNewJob(myScript, mySample, myParams)
+
+    myStudy.launch()
+    myStudy.wait()
+    myStudy.getResult()
+    expected = """Exit code : 0
+Error message : None
+Result:
+55.0"""
+    self.assertEqual(str(myStudy.global_result),expected)
+
+if __name__ == '__main__':
+    unittest.main()

salome/bin/salome/test/pyexample/test_mpmcn.py

@@ -0,0 +1,40 @@
+
+#!/usr/bin/env python3
+#  -*- coding: utf-8 -*-
+# 
+
+# multiple processes on multiple compute nodes
+
+def f(x):
+    return x*x*x
+    
+def f2(x):
+    if x==3:
+        raise RuntimeError("lllll")
+    return x*x*x
+
+import mpmcn
+
+def gg():
+    # case 0 : normal behavior with no raise
+    params = mpmcn.init("localhost")
+    with mpmcn.Pool(params) as p:
+        res = p.map(f,list(range(10)))
+        # getResultDirectory : for advanced users
+        p.getResultDirectory()
+    if res != [0.0, 1.0, 8.0, 27.0, 64.0, 125.0, 216.0, 343.0, 512.0, 729.0]:
+        raise RuntimeError("Test Failed !")
+    # case 1 : behavior with raise
+    params = mpmcn.init("localhost")
+    with mpmcn.Pool(params) as p2:
+        try:
+            res = p2.map(f2,list(range(10)))
+            raise RuntimeError("Exception not thrown -> Error !")
+        except RuntimeError as e:
+            strExpected = "Error for sample # 3 : \'lllll\'"
+            # tmp_dir attr of e returns the ResultDirectory to dive into
+            if str(e)[:len(strExpected)] != strExpected or not hasattr(e,"tmp_dir"):
+                raise RuntimeError("Test Failed 2 !")
+
+if __name__ == "__main__":
+    gg()

salome/bin/salome/test/pyexample/test_prescript.py

@@ -0,0 +1,52 @@
+import unittest
+import os
+import time
+
+class TestYdefx(unittest.TestCase):
+  def test_prescript(self):
+    """
+    This test shows how to use an initialization script which is called one time
+    before any evaluation of the study function.
+    """
+    import pydefx
+
+    myParams = pydefx.Parameters()
+    myParams.configureResource("localhost")
+    mywd = os.path.join(myParams.salome_parameters.work_directory,
+                        "prescript_test" +
+                        time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()))
+    myParams.salome_parameters.work_directory = mywd
+    myParams.createResultDirectory("/tmp")
+
+    pyScript = """
+def _exec(name):
+  with open("mydata.txt") as f:
+    mydata = f.read()
+  message = mydata + name
+  return message"""
+
+    myScript = pydefx.PyScript()
+    myScript.loadString(pyScript)
+
+    mySample = myScript.CreateEmptySample()
+    mydata = {"name":["Jean", "Toto", "Titi", "Zizi"]}
+    mySample.setInputValues(mydata)
+
+    myPrescript = """
+with open("mydata.txt", "w") as f:
+  f.write("Hello ")
+"""
+
+    mySchemaBuilder = pydefx.DefaultSchemaBuilder(myPrescript)
+
+    myStudy = pydefx.PyStudy(schemaBuilder=mySchemaBuilder)
+    myStudy.createNewJob(myScript, mySample, myParams)
+
+    myStudy.launch()
+    myStudy.wait()
+    myStudy.getResult()
+    expected = "name,message,messages\n'Jean','Hello Jean',\n'Toto','Hello Toto',\n'Titi','Hello Titi',\n'Zizi','Hello Zizi',\n"
+    self.assertEqual(str(myStudy.sample),expected)
+
+if __name__ == '__main__':
+    unittest.main()

salome/bin/salome/test/pyexample/test_ydefx_base.py

@@ -0,0 +1,79 @@
+#  -*- coding: utf-8 -*-
+# Copyright (C) 2024  CEA, EDF
+#
+# 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, 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
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
+
+from salome.kernel import salome_utils
+import unittest
+import os
+from pathlib import Path
+import pydefx
+
+class TestYdefxBase(unittest.TestCase):
+  def test0(self):
+    """
+    See EDF31521
+    """
+    myParams = pydefx.Parameters("localhost", 4)
+    myParams.salome_parameters.work_directory = "{}".format(Path( myParams.salome_parameters.work_directory ) / "my_base_case" )
+    myParams.salome_parameters.verbose_py_log_level = "ERROR" # ERROR - WARNING - INFO - DEBUG
+    myParams.createResultDirectory("/tmp")
+    myParams.salome_parameters.in_files = []
+    salome_utils.logger.info( f"ressource de calcul: {myParams.salome_parameters.resource_required.name}")
+    salome_utils.logger.info( f"nombre d'évaluations parallèles: {myParams.nb_branches}")
+    salome_utils.logger.info( f"nombre de coeurs demandés: {myParams.salome_parameters.resource_required.nb_proc}" )
+    salome_utils.logger.info( f"nombre de noeuds demandés: {myParams.salome_parameters.resource_required.nb_node}" )
+    salome_utils.logger.info( f"répertoire de travail: {myParams.salome_parameters.work_directory}" )
+    salome_utils.logger.info( f"répertoire local de gestion: {myParams.salome_parameters.result_directory}" )
+    myScript = pydefx.PyScript()
+    myScript.loadString("""from salome.kernel import salome_utils
+def _exec(x, y):
+  from salome.kernel import KernelBasis
+  cst = 1.28
+  salome_utils.logger.info("Je suis une info")
+  KernelBasis.HeatMarcel(cst*1.0,1)
+  d = y / x
+  t = "{} / {} = {}".format(x, y, d)
+  print(f"****** {x}")
+  return d, t
+""")
+    salome_utils.logger.info( f"Inputs : {myScript.getInputNames()}")
+    salome_utils.logger.info( f"Outputs : {myScript.getOutputNames()}")
+    mySample = myScript.CreateEmptySample()
+    mySample.setInputValues({ "x":[ 10, 20, 30, 40],
+                            "y":[ 20, 60, 120, 200]})
+    myStudy = pydefx.PyStudy()
+    myStudy.createNewJob(myScript, mySample, myParams)
+    myStudy.launch() 
+    salome_utils.logger.info( f"Avancement: {myStudy.getProgress()}")
+    salome_utils.logger.info( f"Etat: {myStudy.getJobState()}" )
+
+    # Attendre la fin des calculs
+    myStudy.wait()
+
+    salome_utils.logger.info( f"Etat: {myStudy.getJobState()}" )
+    salome_utils.logger.info( f"Etat: {myStudy.getJobState()}" )
+    res = myStudy.getResult()
+    self.assertEqual(res.result.getOutput("d"), [ 2.0, 3.0, 4.0, 5.0 ] )
+    self.assertEqual( res.result.getOutput("t"), [ '10.0 / 20.0 = 2.0', '20.0 / 60.0 = 3.0', '30.0 / 120.0 = 4.0', '40.0 / 200.0 = 5.0' ] )
+
+if __name__ == '__main__':
+  import KernelBasis
+  KernelBasis.SetVerbosityLevel("WARNING")
+  salome_utils.positionVerbosityOfLoggerRegardingState()
+  unittest.main()