Copyright update 2022

[modules/yacs.git] / src / yacsloader_swig / Test / testWorkloadManager.py

diff --git a/src/yacsloader_swig/Test/testWorkloadManager.py b/src/yacsloader_swig/Test/testWorkloadManager.py
index ab07a4089f91d0d48423b815a6ffd11928405694..d1f299b06f05ce1edffda9a228fa2139462dcd18 100755 (executable)
--- a/src/yacsloader_swig/Test/testWorkloadManager.py
+++ b/src/yacsloader_swig/Test/testWorkloadManager.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-# Copyright (C) 2006-2020  CEA/DEN, EDF R&D
+# Copyright (C) 2006-2022  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
@@ -27,6 +27,7 @@ import tempfile
 import os
 import salome
 
+NB_NODE=15
 class TestEdit(unittest.TestCase):
 
     def setUp(self):
@@ -35,12 +36,13 @@ class TestEdit(unittest.TestCase):
         self.l = loader.YACSLoader()
         self.e = pilot.ExecutorSwig()
         # We need a catalog which contains only one resource named "localhost"
-        # with 16 cores. The modifications made here are not saved to the
+        # with NB_NODE cores. The modifications made here are not saved to the
         # catalog file.
         salome.salome_init()
         resourceManager = salome.lcc.getResourcesManager()
         resource_definition = resourceManager.GetResourceDefinition("localhost")
-        resource_definition.nb_node = 16
+        resource_definition.nb_node = NB_NODE
+        resource_definition.nb_proc_per_node = 1
         resourceManager.AddResource(resource_definition, False, "")
         resource_required = salome.ResourceParameters()
         resource_required.can_run_containers = True
@@ -48,8 +50,9 @@ class TestEdit(unittest.TestCase):
         for r in res_list:
           if r != "localhost":
             resourceManager.RemoveResource(r, False, "")
-        #resource_definition = resourceManager.GetResourceDefinition("localhost")
-        #self.assertEqual(resource_definition.nb_node, 16)
+        resource_definition = resourceManager.GetResourceDefinition("localhost")
+        self.assertEqual(resource_definition.nb_node, NB_NODE)
+        self.assertEqual(resource_definition.nb_proc_per_node, 1)
 
     def tearDown(self):
         cm = salome.lcc.getContainerManager()
@@ -65,10 +68,12 @@ class TestEdit(unittest.TestCase):
         # theoretical time should be 15s
         execution_time = res_port.getPyObj()
         # lower time means some resources are overloaded
-        self.assertTrue(execution_time > 13)
+        msg = "Execution time is too short : {}s".format(execution_time)
+        self.assertTrue(execution_time > 13, msg)
         # The containers may need some time to be launched.
         # We need some delay to add to the 15s.
-        self.assertTrue(execution_time < 20)
+        msg = "Execution time is too long : {}s".format(execution_time)
+        self.assertTrue(execution_time < 20, msg)
 
     def test2(self):
         """ Two parallel foreach-s with different containers and python nodes
@@ -83,10 +88,17 @@ class TestEdit(unittest.TestCase):
         # theoretical time should be 16s
         execution_time = total_time.getPyObj()
         # lower time means some resources are overloaded
-        self.assertTrue(execution_time > 14)
+        msg = "Execution time is too short : {}s".format(execution_time)
+        self.assertTrue(execution_time > 14, msg)
         # The containers may need some time to be launched.
         # We need some delay to add to the 16s.
-        self.assertTrue(execution_time < 20)
+        msg = "Execution time is too long : {}s".format(execution_time)
+        self.assertTrue(execution_time < 20, msg)
+        coeff_cont = proc.getChildByName("End").getOutputPort("coeff_cont").getPyObj()
+        msg = "coeff_cont too low:"+str(coeff_cont)
+        self.assertTrue(coeff_cont >= NB_NODE, msg)
+        msg = "coeff_cont too high:"+str(coeff_cont)
+        self.assertTrue(coeff_cont <= 2*NB_NODE, msg)
 
     def test3(self):
         """ Launch 8 independent nodes in parallel.
@@ -96,9 +108,24 @@ class TestEdit(unittest.TestCase):
         self.assertEqual(proc.getState(),pilot.DONE)
         ok = proc.getChildByName("End").getOutputPort("ok")
         if not ok :
-          err_message = proc.getChildByName("End").getOutputPort("err_message")
+          err_message = proc.getChildByName("End").getOutputPort("err_message").getPyObj()
           self.fail(err_message)
 
+    def test4(self):
+        """ Verify the execution is stoped if no resource can run a task.
+        """
+        proc = self.l.load("samples/wlm_error.xml")
+        self.e.RunW(proc,0)
+        self.assertEqual(proc.getState(),pilot.FAILED)
+        self.assertEqual(proc.getChildByName("ErrorNode").getState(),pilot.ERROR)
+
+    def test5(self):
+        """ Foreach with 1000 points and several nodes in the block.
+        """
+        proc = self.l.load("samples/wlm_complex_foreach.xml")
+        self.e.RunW(proc,0)
+        self.assertEqual(proc.getState(),pilot.DONE)
+
 if __name__ == '__main__':
   dir_test = tempfile.mkdtemp(suffix=".yacstest")
   file_test = os.path.join(dir_test,"UnitTestsResult")