src/yacsloader_swig/Test/testSaveLoadRun.py

   1 # Copyright (C) 2006-2016  CEA/DEN, EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19
  20 import unittest
  21 import pilot
  22 import SALOMERuntime
  23 import loader
  24
  25 import datetime
  26
  27 class TestSaveLoadRun(unittest.TestCase):
  28   def setUp(self):
  29     SALOMERuntime.RuntimeSALOME.setRuntime()
  30     self.r=SALOMERuntime.getSALOMERuntime()
  31     pass
  32
  33   def test0(self):
  34     """First test of HP Container no loop here only the 3 sorts of python nodes (the Distributed is it still used and useful ?) """
  35     fname="TestSaveLoadRun0.xml"
  36     nbOfNodes=8
  37     sqrtOfNumberOfTurn=1000 # 3000 -> 3.2s/Node, 1000 -> 0.1s/Node
  38     l=loader.YACSLoader()
  39     p=self.r.createProc("prTest0")
  40     td=p.createType("double","double")
  41     ti=p.createType("int","int")
  42     cont=p.createContainer("gg","HPSalome")
  43     cont.setSizeOfPool(4)
  44     cont.setProperty("name","localhost")
  45     cont.setProperty("hostname","localhost")
  46     script0="""
  47 def ff(nb,dbg):
  48     from math import cos
  49     import datetime
  50
  51     ref=datetime.datetime.now()
  52     t=0. ; pas=1./float(nb)
  53     for i in xrange(nb):
  54         for j in xrange(nb):
  55             x=j*pas
  56             t+=1.+cos(1.*(x*3.14159))
  57             pass
  58         pass
  59     print "coucou from script0-%i  -> %s"%(dbg,str(datetime.datetime.now()-ref))
  60     return t
  61 """
  62     script1="""
  63 from math import cos
  64 import datetime
  65 ref=datetime.datetime.now()
  66 o2=0. ; pas=1./float(i1)
  67 for i in xrange(i1):
  68   for j in xrange(i1):
  69     x=j*pas
  70     o2+=1.+cos(1.*(x*3.14159))
  71     pass
  72 print "coucou from script1-%i  -> %s"%(dbg,str(datetime.datetime.now()-ref))
  73 """
  74     for i in xrange(nbOfNodes):
  75       node0=self.r.createFuncNode("DistPython","node%i"%(i))
  76       p.edAddChild(node0)
  77       node0.setFname("ff")
  78       node0.setContainer(cont)
  79       node0.setScript(script0)
  80       nb=node0.edAddInputPort("nb",ti) ; nb.edInitInt(sqrtOfNumberOfTurn)
  81       dbg=node0.edAddInputPort("dbg",ti) ; dbg.edInitInt(i+1)
  82       out0=node0.edAddOutputPort("s",td)
  83       #
  84       nodeMiddle=self.r.createFuncNode("Salome","node%i_1"%(i))
  85       p.edAddChild(nodeMiddle)
  86       p.edAddCFLink(node0,nodeMiddle)
  87       nodeMiddle.setFname("ff")
  88       nodeMiddle.setContainer(cont)
  89       nodeMiddle.setScript(script0)
  90       nb=nodeMiddle.edAddInputPort("nb",ti) ; nb.edInitInt(sqrtOfNumberOfTurn)
  91       dbg=nodeMiddle.edAddInputPort("dbg",ti) ; dbg.edInitInt(i+1)
  92       out0=nodeMiddle.edAddOutputPort("s",td)
  93       nodeMiddle.setExecutionMode("remote")
  94       #
  95       nodeEnd=self.r.createScriptNode("Salome","node%i_2"%(i+1))
  96       p.edAddChild(nodeEnd)
  97       p.edAddCFLink(nodeMiddle,nodeEnd)
  98       nodeEnd.setContainer(cont)
  99       nodeEnd.setScript(script1)
 100       i1=nodeEnd.edAddInputPort("i1",ti) ; i1.edInitInt(sqrtOfNumberOfTurn)
 101       dbg=nodeEnd.edAddInputPort("dbg",ti) ; dbg.edInitInt(i)
 102       o2=nodeEnd.edAddOutputPort("o2",td)
 103       nodeEnd.setExecutionMode("remote")
 104       pass
 105     p.saveSchema(fname)
 106     p=l.load(fname)
 107     ex=pilot.ExecutorSwig()
 108     self.assertEqual(p.getState(),pilot.READY)
 109     st=datetime.datetime.now()
 110     # 1st exec
 111     ex.RunW(p,0)
 112     print "Time spend of test0 to run 1st %s"%(str(datetime.datetime.now()-st))
 113     self.assertEqual(p.getState(),pilot.DONE)
 114     # 2nd exec using the same already launched remote python interpreters
 115     st=datetime.datetime.now()
 116     ex.RunW(p,0)
 117     print "Time spend of test0 to run 2nd %s"%(str(datetime.datetime.now()-st))
 118     self.assertEqual(p.getState(),pilot.DONE)
 119     # 3rd exec using the same already launched remote python interpreters
 120     st=datetime.datetime.now()
 121     ex.RunW(p,0)
 122     print "Time spend of test0 to run 3rd %s"%(str(datetime.datetime.now()-st))
 123     self.assertEqual(p.getState(),pilot.DONE)
 124     pass
 125
 126   def test1(self):
 127     """ HP Container again like test0 but the initialization key of HPContainer is used here."""
 128     fname="TestSaveLoadRun1.xml"
 129     nbOfNodes=8
 130     sqrtOfNumberOfTurn=1000 # 3000 -> 3.2s/Node, 1000 -> 0.1s/Node
 131     l=loader.YACSLoader()
 132     p=self.r.createProc("prTest1")
 133     td=p.createType("double","double")
 134     ti=p.createType("int","int")
 135     cont=p.createContainer("gg","HPSalome")
 136     cont.setSizeOfPool(4)
 137     cont.setProperty("InitializeScriptKey","aa=123.456")
 138     cont.setProperty("name","localhost")
 139     cont.setProperty("hostname","localhost")
 140     script0="""
 141 def ff(nb,dbg):
 142     from math import cos
 143     import datetime
 144
 145     ref=datetime.datetime.now()
 146     t=0. ; pas=1./float(nb)
 147     for i in xrange(nb):
 148         for j in xrange(nb):
 149             x=j*pas
 150             t+=1.+cos(1.*(x*3.14159))
 151             pass
 152         pass
 153     print "coucou from script0-%i  -> %s"%(dbg,str(datetime.datetime.now()-ref))
 154     return t
 155 """
 156     # here in script1 aa is refered ! aa will exist thanks to HPCont Init Script
 157     script1="""
 158 from math import cos
 159 import datetime
 160 ref=datetime.datetime.now()
 161 o2=0. ; pas=1./float(i1)
 162 for i in xrange(i1):
 163   for j in xrange(i1):
 164     x=j*pas
 165     o2+=1.+cos(1.*(x*3.14159))
 166     pass
 167 print "coucou %lf from script1-%i  -> %s"%(aa,dbg,str(datetime.datetime.now()-ref))
 168 aa+=1.
 169 """
 170     #
 171     for i in xrange(nbOfNodes):
 172       nodeMiddle=self.r.createFuncNode("Salome","node%i_1"%(i)) # PyFuncNode remote
 173       p.edAddChild(nodeMiddle)
 174       nodeMiddle.setFname("ff")
 175       nodeMiddle.setContainer(cont)
 176       nodeMiddle.setScript(script0)
 177       nb=nodeMiddle.edAddInputPort("nb",ti) ; nb.edInitInt(sqrtOfNumberOfTurn)
 178       dbg=nodeMiddle.edAddInputPort("dbg",ti) ; dbg.edInitInt(i+1)
 179       out0=nodeMiddle.edAddOutputPort("s",td)
 180       nodeMiddle.setExecutionMode("remote")
 181       #
 182       nodeEnd=self.r.createScriptNode("Salome","node%i_2"%(i+1)) # PythonNode remote
 183       p.edAddChild(nodeEnd)
 184       p.edAddCFLink(nodeMiddle,nodeEnd)
 185       nodeEnd.setContainer(cont)
 186       nodeEnd.setScript(script1)
 187       i1=nodeEnd.edAddInputPort("i1",ti) ; i1.edInitInt(sqrtOfNumberOfTurn)
 188       dbg=nodeEnd.edAddInputPort("dbg",ti) ; dbg.edInitInt(i)
 189       o2=nodeEnd.edAddOutputPort("o2",td)
 190       nodeEnd.setExecutionMode("remote")
 191       pass
 192     #
 193     p.saveSchema(fname)
 194     p=l.load(fname)
 195     self.assertEqual(p.edGetDirectDescendants()[0].getContainer().getProperty("InitializeScriptKey"),"aa=123.456")
 196     # 1st exec
 197     ex=pilot.ExecutorSwig()
 198     self.assertEqual(p.getState(),pilot.READY)
 199     st=datetime.datetime.now()
 200     ex.RunW(p,0)
 201     print "Time spend of test1 to 1st run %s"%(str(datetime.datetime.now()-st))
 202     self.assertEqual(p.getState(),pilot.DONE)
 203     # 2nd exec
 204     st=datetime.datetime.now()
 205     ex.RunW(p,0)
 206     print "Time spend of test1 to 2nd run %s"%(str(datetime.datetime.now()-st))
 207     self.assertEqual(p.getState(),pilot.DONE)
 208     # 3rd exec
 209     st=datetime.datetime.now()
 210     ex.RunW(p,0)
 211     print "Time spend of test1 to 3rd run %s"%(str(datetime.datetime.now()-st))
 212     self.assertEqual(p.getState(),pilot.DONE)
 213     pass
 214
 215   def test2(self):
 216     """ Test on HP Containers in foreach context."""
 217     script0="""def ff():
 218     global aa
 219     print "%%lf - %%s"%%(aa,str(my_container))
 220     return 16*[%i],0
 221 """
 222     script1="""from math import cos
 223 import datetime
 224 ref=datetime.datetime.now()
 225 o2=0. ; pas=1./float(i1)
 226 for i in xrange(i1):
 227   for j in xrange(i1):
 228     x=j*pas
 229     o2+=1.+cos(1.*(x*3.14159))
 230     pass
 231 print "coucou %lf from script  -> %s"%(aa,str(datetime.datetime.now()-ref))
 232 aa+=1.
 233 o3=0
 234 """
 235     script2="""o9=sum(i8)
 236 """
 237     fname="TestSaveLoadRun2.xml"
 238     nbOfNodes=8
 239     sqrtOfNumberOfTurn=1000 # 3000 -> 3.2s/Node, 1000 -> 0.1s/Node
 240     l=loader.YACSLoader()
 241     p=self.r.createProc("prTest1")
 242     td=p.createType("double","double")
 243     ti=p.createType("int","int")
 244     tdi=p.createSequenceTc("seqint","seqint",ti)
 245     tdd=p.createSequenceTc("seqdouble","seqdouble",td)
 246     cont=p.createContainer("gg","HPSalome")
 247     cont.setSizeOfPool(4)
 248     cont.setProperty("InitializeScriptKey","aa=123.456")
 249     cont.setProperty("name","localhost")
 250     cont.setProperty("hostname","localhost")
 251     #
 252     node0=self.r.createFuncNode("Salome","PyFunction0") # PyFuncNode remote
 253     p.edAddChild(node0)
 254     node0.setFname("ff")
 255     node0.setContainer(cont)
 256     node0.setScript(script0%(sqrtOfNumberOfTurn))
 257     out0_0=node0.edAddOutputPort("o1",tdi)
 258     out1_0=node0.edAddOutputPort("o2",ti)
 259     node0.setExecutionMode("remote")
 260     #
 261     node1=self.r.createForEachLoop("node1",ti)
 262     p.edAddChild(node1)
 263     p.edAddCFLink(node0,node1)
 264     p.edAddLink(out0_0,node1.edGetSeqOfSamplesPort())
 265     node1.edGetNbOfBranchesPort().edInitInt(8)
 266     #
 267     node2=self.r.createScriptNode("Salome","PyScript3")
 268     node1.edAddChild(node2)
 269     node2.setContainer(cont)
 270     node2.setScript(script1)
 271     i1=node2.edAddInputPort("i1",ti)
 272     p.edAddLink(node1.edGetSamplePort(),i1)
 273     out0_2=node2.edAddOutputPort("o2",td)
 274     out1_2=node2.edAddOutputPort("o3",ti)
 275     node2.setExecutionMode("remote")
 276     #
 277     node3=self.r.createScriptNode("Salome","PyScript7")
 278     p.edAddChild(node3)
 279     node3.setScript(script2)
 280     p.edAddCFLink(node1,node3)
 281     i8=node3.edAddInputPort("i8",tdd)
 282     o9=node3.edAddOutputPort("o9",td)
 283     p.edAddLink(out0_2,i8)
 284     #
 285     p.saveSchema(fname)
 286     p=l.load(fname)
 287     o9=p.getChildByName("PyScript7").getOutputPort("o9")
 288     self.assertTrue(len(p.edGetDirectDescendants()[1].getChildByName("PyScript3").getContainer().getProperty("InitializeScriptKey"))!=0)
 289     # 1st exec
 290     refExpected=16016013.514623128
 291     ex=pilot.ExecutorSwig()
 292     self.assertEqual(p.getState(),pilot.READY)
 293     st=datetime.datetime.now()
 294     ex.RunW(p,0)
 295     print "Time spend of test2 to 1st run %s"%(str(datetime.datetime.now()-st))
 296     self.assertEqual(p.getState(),pilot.DONE)
 297     self.assertAlmostEqual(refExpected,o9.getPyObj(),5)
 298     # 2nd exec
 299     st=datetime.datetime.now()
 300     ex.RunW(p,0)
 301     print "Time spend of test2 to 2nd run %s"%(str(datetime.datetime.now()-st))
 302     self.assertEqual(p.getState(),pilot.DONE)
 303     self.assertAlmostEqual(refExpected,o9.getPyObj(),5)
 304     # 3rd exec
 305     st=datetime.datetime.now()
 306     ex.RunW(p,0)
 307     print "Time spend of test2 to 3rd run %s"%(str(datetime.datetime.now()-st))
 308     self.assertEqual(p.getState(),pilot.DONE)
 309     self.assertAlmostEqual(refExpected,o9.getPyObj(),5)
 310     pass
 311
 312   def test3(self):
 313     """ Test that focuses on parallel load of containers."""
 314     script0="""def ff():
 315     global aa
 316     print "%%lf - %%s"%%(aa,str(my_container))
 317     return 100*[%i],0
 318 """
 319     script1="""from math import cos
 320 import datetime
 321 ref=datetime.datetime.now()
 322 o2=0. ; pas=1./float(i1)
 323 for i in xrange(i1):
 324   for j in xrange(i1):
 325     x=j*pas
 326     o2+=1.+cos(1.*(x*3.14159))
 327     pass
 328 print "coucou %lf from script  -> %s"%(aa,str(datetime.datetime.now()-ref))
 329 aa+=1.
 330 o3=0
 331 """
 332     script2="""o9=sum(i8)
 333 """
 334     fname="TestSaveLoadRun3.xml"
 335     nbOfNodes=8
 336     sqrtOfNumberOfTurn=10
 337     l=loader.YACSLoader()
 338     p=self.r.createProc("prTest1")
 339     td=p.createType("double","double")
 340     ti=p.createType("int","int")
 341     tdi=p.createSequenceTc("seqint","seqint",ti)
 342     tdd=p.createSequenceTc("seqdouble","seqdouble",td)
 343     cont=p.createContainer("gg","HPSalome")
 344     cont.setSizeOfPool(8)
 345     cont.setProperty("InitializeScriptKey","aa=123.456")
 346     cont.setProperty("name","localhost")
 347     cont.setProperty("hostname","localhost")
 348     #
 349     node0=self.r.createFuncNode("Salome","PyFunction0") # PyFuncNode remote
 350     p.edAddChild(node0)
 351     node0.setFname("ff")
 352     node0.setContainer(cont)
 353     node0.setScript(script0%(sqrtOfNumberOfTurn))
 354     out0_0=node0.edAddOutputPort("o1",tdi)
 355     out1_0=node0.edAddOutputPort("o2",ti)
 356     node0.setExecutionMode("remote")
 357     #
 358     node1=self.r.createForEachLoop("node1",ti)
 359     p.edAddChild(node1)
 360     p.edAddCFLink(node0,node1)
 361     p.edAddLink(out0_0,node1.edGetSeqOfSamplesPort())
 362     node1.edGetNbOfBranchesPort().edInitInt(16)
 363     #
 364     node2=self.r.createScriptNode("Salome","PyScript3")
 365     node1.edAddChild(node2)
 366     node2.setContainer(cont)
 367     node2.setScript(script1)
 368     i1=node2.edAddInputPort("i1",ti)
 369     p.edAddLink(node1.edGetSamplePort(),i1)
 370     out0_2=node2.edAddOutputPort("o2",td)
 371     out1_2=node2.edAddOutputPort("o3",ti)
 372     node2.setExecutionMode("remote")
 373     #
 374     node3=self.r.createScriptNode("Salome","PyScript7")
 375     p.edAddChild(node3)
 376     node3.setScript(script2)
 377     p.edAddCFLink(node1,node3)
 378     i8=node3.edAddInputPort("i8",tdd)
 379     o9=node3.edAddOutputPort("o9",td)
 380     p.edAddLink(out0_2,i8)
 381     #
 382     p.saveSchema(fname)
 383     p=l.load(fname)
 384     o9=p.getChildByName("PyScript7").getOutputPort("o9")
 385     self.assertTrue(len(p.edGetDirectDescendants()[1].getChildByName("PyScript3").getContainer().getProperty("InitializeScriptKey"))!=0)
 386     # 1st exec
 387     refExpected=11000.008377058712
 388     ex=pilot.ExecutorSwig()
 389     self.assertEqual(p.getState(),pilot.READY)
 390     st=datetime.datetime.now()
 391     ex.RunW(p,0)
 392     print "Time spend of test3 to 1st run %s"%(str(datetime.datetime.now()-st))
 393     self.assertEqual(p.getState(),pilot.DONE)
 394     self.assertAlmostEqual(refExpected,o9.getPyObj(),5)
 395     # 2nd exec
 396     st=datetime.datetime.now()
 397     ex.RunW(p,0)
 398     print "Time spend of test3 to 2nd run %s"%(str(datetime.datetime.now()-st))
 399     self.assertEqual(p.getState(),pilot.DONE)
 400     self.assertAlmostEqual(refExpected,o9.getPyObj(),5)
 401     # 3rd exec
 402     st=datetime.datetime.now()
 403     ex.RunW(p,0)
 404     print "Time spend of test3 to 3rd run %s"%(str(datetime.datetime.now()-st))
 405     self.assertEqual(p.getState(),pilot.DONE)
 406     self.assertAlmostEqual(refExpected,o9.getPyObj(),5)
 407     pass
 408
 409   def test4(self):
 410     """Non regression test of multi pyScriptNode, pyFuncNode sharing the same HPContainer instance."""
 411     fname="TestSaveLoadRun4.xml"
 412     script1="""nb=7
 413 ii=0
 414 o1=nb*[None]
 415 for i in xrange(nb):
 416     tmp=(i+10)*[None]
 417     for j in xrange(i+10):
 418         tmp[j]=ii
 419         ii+=1
 420         pass
 421     o1[i]=tmp
 422     pass
 423 """
 424     l=loader.YACSLoader()
 425     ex=pilot.ExecutorSwig()
 426     p=self.r.createProc("pr")
 427     cont=p.createContainer("gg","HPSalome")
 428     cont.setSizeOfPool(10)
 429     td=p.createType("int","int")
 430     td2=p.createSequenceTc("seqint","seqint",td)
 431     td3=p.createSequenceTc("seqintvec","seqintvec",td2)
 432     node1=self.r.createScriptNode("","node1")
 433     node1.setScript(script1)
 434     o1=node1.edAddOutputPort("o1",td3)
 435     p.edAddChild(node1)
 436     #
 437     node2=self.r.createForEachLoop("node2",td2)
 438     p.edAddChild(node2)
 439     p.edAddCFLink(node1,node2)
 440     p.edAddLink(o1,node2.edGetSeqOfSamplesPort())
 441     node2.edGetNbOfBranchesPort().edInitInt(2)
 442     node20=self.r.createBloc("node20")
 443     node2.edAddChild(node20)
 444     node200=self.r.createForEachLoop("node200",td)
 445     node20.edAddChild(node200)
 446     node200.edGetNbOfBranchesPort().edInitInt(10)
 447     p.edAddLink(node2.edGetSamplePort(),node200.edGetSeqOfSamplesPort())
 448     node2000=self.r.createScriptNode("","node2000")
 449     node2000.setContainer(cont)
 450     node2000.setExecutionMode("remote")
 451     node200.edAddChild(node2000)
 452     i5=node2000.edAddInputPort("i5",td)
 453     o6=node2000.edAddOutputPort("o6",td)
 454     node2000.setScript("o6=2+i5")
 455     p.edAddLink(node200.edGetSamplePort(),i5)
 456     #
 457     node3=self.r.createForEachLoop("node3",td2)
 458     p.edAddChild(node3)
 459     p.edAddCFLink(node2,node3)
 460     p.edAddLink(o6,node3.edGetSeqOfSamplesPort())
 461     node3.edGetNbOfBranchesPort().edInitInt(2)
 462     node30=self.r.createBloc("node30")
 463     node3.edAddChild(node30)
 464     node300=self.r.createForEachLoop("node300",td)
 465     node30.edAddChild(node300)
 466     node300.edGetNbOfBranchesPort().edInitInt(10)
 467     p.edAddLink(node3.edGetSamplePort(),node300.edGetSeqOfSamplesPort())
 468     node3000=self.r.createScriptNode("","node3000")
 469     node3000.setContainer(cont)
 470     node3000.setExecutionMode("remote")
 471     node300.edAddChild(node3000)
 472     i14=node3000.edAddInputPort("i14",td)
 473     o15=node3000.edAddOutputPort("o15",td)
 474     node3000.setScript("o15=3+i14")
 475     p.edAddLink(node300.edGetSamplePort(),i14)
 476     #
 477     node4=self.r.createScriptNode("","node4")
 478     node4.setScript("o9=i8")
 479     p.edAddChild(node4)
 480     i8=node4.edAddInputPort("i8",td3)
 481     o9=node4.edAddOutputPort("o9",td3)
 482     p.edAddCFLink(node3,node4)
 483     p.edAddLink(o15,i8)
 484     p.saveSchema(fname)
 485     p=l.load(fname)
 486     ex = pilot.ExecutorSwig()
 487     self.assertEqual(p.getState(),pilot.READY)
 488     ex.RunW(p,0)
 489     self.assertEqual(p.getState(),pilot.DONE)
 490     zeResu=p.getChildByName("node4").getOutputPort("o9").get()
 491     self.assertEqual(zeResu,[[5,6,7,8,9,10,11,12,13,14],[15,16,17,18,19,20,21,22,23,24,25],[26,27,28,29,30,31,32,33,34,35,36,37],[38,39,40,41,42,43,44,45,46,47,48,49,50],[51,52,53,54,55,56,57,58,59,60,61,62,63,64],[65,66,67,68,69,70,71,72,73,74,75,76,77,78,79], [80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95]])
 492     pass
 493
 494   def test5(self):
 495     """Non regression test 2 of multi pyNode, pyFuncNode sharing the same HPContainer instance."""
 496     fname="TestSaveLoadRun5.xml"
 497     script1="""nb=7
 498 ii=0
 499 o1=nb*[None]
 500 for i in xrange(nb):
 501     tmp=(i+10)*[None]
 502     for j in xrange(i+10):
 503         tmp[j]=ii
 504         ii+=1
 505         pass
 506     o1[i]=tmp
 507     pass
 508 """
 509     l=loader.YACSLoader()
 510     ex=pilot.ExecutorSwig()
 511     p=self.r.createProc("pr")
 512     cont=p.createContainer("gg","HPSalome")
 513     cont.setSizeOfPool(10)
 514     td=p.createType("int","int")
 515     td2=p.createSequenceTc("seqint","seqint",td)
 516     td3=p.createSequenceTc("seqintvec","seqintvec",td2)
 517     node1=self.r.createScriptNode("","node1")
 518     node1.setScript(script1)
 519     o1=node1.edAddOutputPort("o1",td3)
 520     p.edAddChild(node1)
 521     #
 522     node2=self.r.createForEachLoop("node2",td2)
 523     p.edAddChild(node2)
 524     p.edAddCFLink(node1,node2)
 525     p.edAddLink(o1,node2.edGetSeqOfSamplesPort())
 526     node2.edGetNbOfBranchesPort().edInitInt(2)
 527     node20=self.r.createBloc("node20")
 528     node2.edAddChild(node20)
 529     node200=self.r.createForEachLoop("node200",td)
 530     node20.edAddChild(node200)
 531     node200.edGetNbOfBranchesPort().edInitInt(10)
 532     p.edAddLink(node2.edGetSamplePort(),node200.edGetSeqOfSamplesPort())
 533     node2000=self.r.createFuncNode("Salome","node2000")
 534     node2000.setFname("ff")
 535     node2000.setContainer(cont)
 536     node2000.setExecutionMode("remote")
 537     node200.edAddChild(node2000)
 538     i5=node2000.edAddInputPort("i5",td)
 539     o6=node2000.edAddOutputPort("o6",td)
 540     node2000.setScript("def ff(x):\n  return 2+x")
 541     p.edAddLink(node200.edGetSamplePort(),i5)
 542     #
 543     node3=self.r.createForEachLoop("node3",td2)
 544     p.edAddChild(node3)
 545     p.edAddCFLink(node2,node3)
 546     p.edAddLink(o6,node3.edGetSeqOfSamplesPort())
 547     node3.edGetNbOfBranchesPort().edInitInt(2)
 548     node30=self.r.createBloc("node30")
 549     node3.edAddChild(node30)
 550     node300=self.r.createForEachLoop("node300",td)
 551     node30.edAddChild(node300)
 552     node300.edGetNbOfBranchesPort().edInitInt(10)
 553     p.edAddLink(node3.edGetSamplePort(),node300.edGetSeqOfSamplesPort())
 554     node3000=self.r.createFuncNode("Salome","node3000")
 555     node3000.setFname("ff")
 556     node3000.setContainer(cont)
 557     node3000.setExecutionMode("remote")
 558     node300.edAddChild(node3000)
 559     i14=node3000.edAddInputPort("i14",td)
 560     o15=node3000.edAddOutputPort("o15",td)
 561     node3000.setScript("def ff(x):\n  return 3+x")
 562     p.edAddLink(node300.edGetSamplePort(),i14)
 563     #
 564     node4=self.r.createScriptNode("","node4")
 565     node4.setScript("o9=i8")
 566     p.edAddChild(node4)
 567     i8=node4.edAddInputPort("i8",td3)
 568     o9=node4.edAddOutputPort("o9",td3)
 569     p.edAddCFLink(node3,node4)
 570     p.edAddLink(o15,i8)
 571     p.saveSchema(fname)
 572     p=l.load(fname)
 573     ex = pilot.ExecutorSwig()
 574     self.assertEqual(p.getState(),pilot.READY)
 575     ex.RunW(p,0)
 576     self.assertEqual(p.getState(),pilot.DONE)
 577     zeResu=p.getChildByName("node4").getOutputPort("o9").get()
 578     self.assertEqual(zeResu,[[5,6,7,8,9,10,11,12,13,14],[15,16,17,18,19,20,21,22,23,24,25],[26,27,28,29,30,31,32,33,34,35,36,37],[38,39,40,41,42,43,44,45,46,47,48,49,50],[51,52,53,54,55,56,57,58,59,60,61,62,63,64],[65,66,67,68,69,70,71,72,73,74,75,76,77,78,79], [80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95]])
 579     pass
 580
 581   def test6(self):
 582     fname="test6.xml"
 583     p=self.r.createProc("prTest0")
 584     td=p.createType("double","double")
 585     ti=p.createType("int","int")
 586     tsi=p.createSequenceTc("seqint","seqint",ti)
 587     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
 588     n0=self.r.createScriptNode("","n0")
 589     o0=n0.edAddOutputPort("o0",tsi)
 590     n0.setScript("o0=[3,6,8,9,-2,5]")
 591     p.edAddChild(n0)
 592     n1=self.r.createForEachLoop("n1",ti)
 593     n10=self.r.createScriptNode("","n10")
 594     n1.edAddChild(n10)
 595     n10.setScript("o2=2*i1")
 596     i1=n10.edAddInputPort("i1",ti)
 597     o2=n10.edAddOutputPort("o2",ti)
 598     p.edAddChild(n1)
 599     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
 600     p.edAddLink(n1.edGetSamplePort(),i1)
 601     p.edAddCFLink(n0,n1)
 602     n1.edGetNbOfBranchesPort().edInitPy(1)
 603     n2=self.r.createScriptNode("","n2")
 604     n2.setScript("o4=i3")
 605     i3=n2.edAddInputPort("i3",tsi)
 606     o4=n2.edAddOutputPort("o4",tsi)
 607     n2.setScript("o4=i3")
 608     p.edAddChild(n2)
 609     p.edAddCFLink(n1,n2)
 610     p.edAddLink(o2,i3)
 611     p.saveSchema(fname)
 612     #
 613     l=loader.YACSLoader()
 614     p=l.load(fname)
 615     n1=p.getChildByName("n1")
 616     ex=pilot.ExecutorSwig()
 617     #
 618     ex.RunW(p,0)
 619     #
 620     self.assertEqual(n1.getState(),pilot.DONE)
 621     n1.edGetSeqOfSamplesPort().getPyObj()
 622     a,b,c=n1.getPassedResults(ex)
 623     self.assertEqual(a,range(6))
 624     self.assertEqual([elt.getPyObj() for elt in b],[[6L, 12L, 16L, 18L, -4L, 10L]])
 625     self.assertEqual(c,['n10_o2_interceptor'])
 626     pass
 627
 628   def test7(self):
 629     fname="test7.xml"
 630     p=self.r.createProc("prTest1")
 631     cont=p.createContainer("gg","Salome")
 632     cont.setProperty("name","localhost")
 633     cont.setProperty("hostname","localhost")
 634     cont.setProperty("type","multi")
 635     td=p.createType("double","double")
 636     ti=p.createType("int","int")
 637     tsi=p.createSequenceTc("seqint","seqint",ti)
 638     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
 639     n0=self.r.createScriptNode("","n0")
 640     o0=n0.edAddOutputPort("o0",tsi)
 641     n0.setScript("o0=[3,6,8,9,-2,5]")
 642     p.edAddChild(n0)
 643     n1=self.r.createForEachLoop("n1",ti)
 644     n10=self.r.createScriptNode("","n10")
 645     n10.setExecutionMode("remote")
 646     n10.setContainer(cont)
 647     n1.edAddChild(n10)
 648     n10.setScript("""
 649 import time
 650 if i1==9:
 651   time.sleep(2)
 652   raise Exception("Simulated error !")
 653 else:
 654   o2=2*i1
 655 """)
 656     i1=n10.edAddInputPort("i1",ti)
 657     o2=n10.edAddOutputPort("o2",ti)
 658     p.edAddChild(n1)
 659     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
 660     p.edAddLink(n1.edGetSamplePort(),i1)
 661     p.edAddCFLink(n0,n1)
 662     n1.edGetNbOfBranchesPort().edInitPy(1)
 663     n2=self.r.createScriptNode("","n2")
 664     n2.setScript("o4=i3")
 665     i3=n2.edAddInputPort("i3",tsi)
 666     o4=n2.edAddOutputPort("o4",tsi)
 667     n2.setScript("o4=i3")
 668     p.edAddChild(n2)
 669     p.edAddCFLink(n1,n2)
 670     p.edAddLink(o2,i3)
 671     p.saveSchema(fname)
 672     #
 673     l=loader.YACSLoader()
 674     p=l.load(fname)
 675     n1=p.getChildByName("n1")
 676     ex=pilot.ExecutorSwig()
 677     #
 678     ex.RunW(p,0)
 679     #
 680     self.assertEqual(n1.getState(),pilot.FAILED)
 681     n1.edGetSeqOfSamplesPort().getPyObj()
 682     a,b,c=n1.getPassedResults(ex)
 683     self.assertEqual(a,range(3))
 684     self.assertEqual([elt.getPyObj() for elt in b],[[6L,12L,16L]])
 685     self.assertEqual(c,['n10_o2_interceptor'])
 686     pass
 687
 688   def test8(self):
 689     from datetime import datetime
 690     fname="test8.xml"
 691     p=self.r.createProc("prTest2")
 692     cont=p.createContainer("gg","Salome")
 693     cont.setProperty("name","localhost")
 694     cont.setProperty("hostname","localhost")
 695     cont.setProperty("type","multi")
 696     td=p.createType("double","double")
 697     ti=p.createType("int","int")
 698     tsi=p.createSequenceTc("seqint","seqint",ti)
 699     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
 700     n0=self.r.createScriptNode("","n0")
 701     o0=n0.edAddOutputPort("o0",tsi)
 702     n0.setScript("o0=[3,6,8,9,-2,5]")
 703     p.edAddChild(n0)
 704     n1=self.r.createForEachLoop("n1",ti)
 705     n10=self.r.createScriptNode("","n10")
 706     n10.setExecutionMode("remote")
 707     n10.setContainer(cont)
 708     n1.edAddChild(n10)
 709     n10.setScript("""
 710 import time
 711 if i1==9:
 712   raise Exception("Simulated error !")
 713 else:
 714   time.sleep(0.1)
 715   o2=2*i1
 716 """)
 717     i1=n10.edAddInputPort("i1",ti)
 718     o2=n10.edAddOutputPort("o2",ti)
 719     p.edAddChild(n1)
 720     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
 721     p.edAddLink(n1.edGetSamplePort(),i1)
 722     p.edAddCFLink(n0,n1)
 723     n1.edGetNbOfBranchesPort().edInitPy(2)
 724     n2=self.r.createScriptNode("","n2")
 725     n2.setScript("o4=i3")
 726     i3=n2.edAddInputPort("i3",tsi)
 727     o4=n2.edAddOutputPort("o4",tsi)
 728     n2.setScript("o4=i3")
 729     p.edAddChild(n2)
 730     p.edAddCFLink(n1,n2)
 731     p.edAddLink(o2,i3)
 732     p.saveSchema(fname)
 733     #
 734     l=loader.YACSLoader()
 735     p=l.load(fname)
 736     n1=p.getChildByName("n1")
 737     ex=pilot.ExecutorSwig()
 738     ex.setKeepGoingProperty(True)
 739     #
 740     startt=datetime.now()
 741     ex.RunW(p,0)
 742     t0=datetime.now()-startt
 743     #
 744     self.assertEqual(n1.getState(),pilot.FAILED)
 745     n1.edGetSeqOfSamplesPort().getPyObj()
 746     a,b,c=n1.getPassedResults(ex)
 747     self.assertEqual(a,[0,1,2,4,5])
 748     self.assertEqual([elt.getPyObj() for elt in b],[[6L,12L,16L,-4L,10L]])
 749     self.assertEqual(c,['n10_o2_interceptor'])
 750
 751     p.getChildByName("n1").getChildByName("n10").setScript("""
 752 import time
 753 if i1==3:
 754   time.sleep(2)
 755   raise Exception("Simulated error !")
 756 else:
 757   o2=2*i1
 758 """)
 759     ex=pilot.ExecutorSwig()
 760     ex.setKeepGoingProperty(True)
 761     #
 762     startt=datetime.now()
 763     ex.RunW(p,0)
 764     t1=datetime.now()-startt
 765     #
 766     self.assertEqual(n1.getState(),pilot.FAILED)
 767     n1.edGetSeqOfSamplesPort().getPyObj()
 768     a,b,c=n1.getPassedResults(ex)
 769     self.assertEqual(a,[1,2,3,4,5])
 770     self.assertEqual([elt.getPyObj() for elt in b],[[12L,16L,18L,-4L,10L]])
 771     self.assertEqual(c,['n10_o2_interceptor'])
 772     pass
 773
 774   def test9(self):
 775     """ Test of assignation of already computed values for foreach node."""
 776     fname="test9.xml"
 777     from datetime import datetime
 778     p=self.r.createProc("prTest2")
 779     cont=p.createContainer("gg","Salome")
 780     cont.setProperty("name","localhost")
 781     cont.setProperty("hostname","localhost")
 782     cont.setProperty("type","multi")
 783     td=p.createType("double","double")
 784     ti=p.createType("int","int")
 785     tsi=p.createSequenceTc("seqint","seqint",ti)
 786     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
 787     n0=self.r.createScriptNode("","n0")
 788     o0=n0.edAddOutputPort("o0",tsi)
 789     n0.setScript("o0=[3,6,8,9,-2,5]")
 790     p.edAddChild(n0)
 791     n1=self.r.createForEachLoop("n1",ti)
 792     n10=self.r.createScriptNode("","n10")
 793     n10.setExecutionMode("remote")
 794     n10.setContainer(cont)
 795     n1.edAddChild(n10)
 796     n10.setScript("""
 797 import time
 798 if i1==9:
 799   raise Exception("Simulated error !")
 800 else:
 801   time.sleep(0.1)
 802   o2=2*i1
 803 """)
 804     i1=n10.edAddInputPort("i1",ti)
 805     o2=n10.edAddOutputPort("o2",ti)
 806     p.edAddChild(n1)
 807     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
 808     p.edAddLink(n1.edGetSamplePort(),i1)
 809     p.edAddCFLink(n0,n1)
 810     n1.edGetNbOfBranchesPort().edInitPy(2)
 811     n2=self.r.createScriptNode("","n2")
 812     n2.setScript("o4=i3")
 813     i3=n2.edAddInputPort("i3",tsi)
 814     o4=n2.edAddOutputPort("o4",tsi)
 815     n2.setScript("o4=i3")
 816     p.edAddChild(n2)
 817     p.edAddCFLink(n1,n2)
 818     p.edAddLink(o2,i3)
 819     p.saveSchema(fname)
 820     #
 821     l=loader.YACSLoader()
 822     p=l.load(fname)
 823     n1=p.getChildByName("n1")
 824     ex=pilot.ExecutorSwig()
 825     ex.setKeepGoingProperty(True)
 826     #
 827     startt=datetime.now()
 828     ex.RunW(p,0)
 829     t0=datetime.now()-startt
 830     #
 831     self.assertEqual(p.getState(),pilot.FAILED)
 832     self.assertEqual(n1.getState(),pilot.FAILED)
 833     n1.edGetSeqOfSamplesPort().getPyObj()
 834     a,b,c=n1.getPassedResults(ex)
 835     self.assertEqual(a,[0,1,2,4,5])
 836     self.assertEqual([elt.getPyObj() for elt in b],[[6L,12L,16L,-4L,10L]])
 837     self.assertEqual(c,['n10_o2_interceptor'])
 838
 839     p.getChildByName("n1").getChildByName("n10").setScript("""
 840 import time
 841 time.sleep(2)
 842 o2=7*i1
 843 """)
 844     ex=pilot.ExecutorSwig()
 845     ex.setKeepGoingProperty(True)
 846     p.getChildByName("n1").assignPassedResults(a,b,c)
 847     #
 848     startt=datetime.now()
 849     ex.RunW(p,0)
 850     t1=datetime.now()-startt
 851     #
 852     self.assertEqual(n1.getState(),pilot.DONE)
 853     self.assertEqual(p.getState(),pilot.DONE)
 854     self.assertEqual(p.getChildByName("n2").getOutputPort("o4").getPyObj(),[6L,12L,16L,63L,-4L,10L])
 855     pass
 856
 857   def test10(self):
 858     fname="test10.xml"
 859     from datetime import datetime
 860     p=self.r.createProc("prTest2")
 861     cont=p.createContainer("gg","Salome")
 862     cont.setProperty("name","localhost")
 863     cont.setProperty("hostname","localhost")
 864     cont.setProperty("type","multi")
 865     td=p.createType("double","double")
 866     ti=p.createType("int","int")
 867     tsi=p.createSequenceTc("seqint","seqint",ti)
 868     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
 869     n0=self.r.createScriptNode("","n0")
 870     o0=n0.edAddOutputPort("o0",tsi)
 871     n0.setScript("o0=[ 3*elt for elt in range(12) ]")
 872     p.edAddChild(n0)
 873     n1=self.r.createForEachLoop("n1",ti)
 874     n10=self.r.createScriptNode("","n10")
 875     n10.setExecutionMode("remote")
 876     n10.setContainer(cont)
 877     n1.edAddChild(n10)
 878     n10.setScript("""
 879 import time
 880 if i1%2==0:
 881   raise Exception("Simulated error !")
 882 else:
 883   time.sleep(0.1)
 884   o2=4*i1
 885 """)
 886     i1=n10.edAddInputPort("i1",ti)
 887     o2=n10.edAddOutputPort("o2",ti)
 888     p.edAddChild(n1)
 889     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
 890     p.edAddLink(n1.edGetSamplePort(),i1)
 891     p.edAddCFLink(n0,n1)
 892     n1.edGetNbOfBranchesPort().edInitPy(2)
 893     n2=self.r.createScriptNode("","n2")
 894     n2.setScript("o4=i3")
 895     i3=n2.edAddInputPort("i3",tsi)
 896     o4=n2.edAddOutputPort("o4",tsi)
 897     n2.setScript("o4=i3")
 898     p.edAddChild(n2)
 899     p.edAddCFLink(n1,n2)
 900     p.edAddLink(o2,i3)
 901     p.saveSchema(fname)
 902     #
 903     l=loader.YACSLoader()
 904     p=l.load(fname)
 905     n1=p.getChildByName("n1")
 906     ex=pilot.ExecutorSwig()
 907     ex.setKeepGoingProperty(True)
 908     #
 909     startt=datetime.now()
 910     ex.RunW(p,0)
 911     t0=datetime.now()-startt
 912     #
 913     self.assertEqual(p.getState(),pilot.FAILED)
 914     self.assertEqual(n1.getState(),pilot.FAILED)
 915     n1.edGetSeqOfSamplesPort().getPyObj()
 916     a,b,c=n1.getPassedResults(ex)
 917     self.assertEqual(a,[1,3,5,7,9,11])
 918     self.assertEqual([elt.getPyObj() for elt in b],[[12L,36L,60L,84L,108L,132L]])
 919     self.assertEqual(c,['n10_o2_interceptor'])
 920
 921     p.getChildByName("n1").getChildByName("n10").setScript("""
 922 import time
 923 if i1%2==1:
 924   raise Exception("Simulated error !")
 925 else:
 926   time.sleep(1)
 927   o2=5*i1
 928 """)
 929     ex=pilot.ExecutorSwig()
 930     ex.setKeepGoingProperty(True)
 931     p.getChildByName("n1").assignPassedResults(a,b,c)
 932     #
 933     startt=datetime.now()
 934     ex.RunW(p,0)
 935     t1=datetime.now()-startt
 936     #assert(t1.total_seconds()<6.+1.)# normally 6/2+1 s (6 remaining elts in 2 // branches + 1s to launch container)
 937     #
 938     self.assertEqual(n1.getState(),pilot.DONE)
 939     self.assertEqual(p.getState(),pilot.DONE)
 940     self.assertEqual(p.getChildByName("n2").getOutputPort("o4").getPyObj(),[0L,12L,30L,36L,60L,60L,90L,84L,120L,108L,150L,132L])
 941     pass
 942
 943   pass
 944
 945
 946   def test11(self):
 947     "test if we do not restart from the begining of the schema after an error in a foreach"
 948     fname="test11.xml"
 949     from datetime import datetime
 950     p=self.r.createProc("prTest2")
 951     cont=p.createContainer("gg","Salome")
 952     cont.setProperty("name","localhost")
 953     cont.setProperty("hostname","localhost")
 954     cont.setProperty("type","multi")
 955     td=p.createType("double","double")
 956     ti=p.createType("int","int")
 957     tsi=p.createSequenceTc("seqint","seqint",ti)
 958     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
 959     n0=self.r.createScriptNode("","n0")
 960     o0=n0.edAddOutputPort("o0",tsi)
 961     n0.setScript("o0=[ elt for elt in range(12) ]")
 962     p.edAddChild(n0)
 963     n1=self.r.createForEachLoop("n1",ti)
 964     n10=self.r.createScriptNode("","n10")
 965     n10.setExecutionMode("remote")
 966     n10.setContainer(cont)
 967     n1.edAddChild(n10)
 968     n10.setScript("""
 969 import time
 970 if i1%2==1:
 971   raise Exception("Simulated error !")
 972 else:
 973   time.sleep(0.1)
 974   o2=2*i1
 975 """)
 976     i1=n10.edAddInputPort("i1",ti)
 977     o2=n10.edAddOutputPort("o2",ti)
 978     p.edAddChild(n1)
 979     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
 980     p.edAddLink(n1.edGetSamplePort(),i1)
 981     p.edAddCFLink(n0,n1)
 982     n1.edGetNbOfBranchesPort().edInitPy(2)
 983     n2=self.r.createScriptNode("","n2")
 984     n2.setScript("o4=i3")
 985     i3=n2.edAddInputPort("i3",tsi)
 986     o4=n2.edAddOutputPort("o4",tsi)
 987     n2.setScript("o4=i3")
 988     p.edAddChild(n2)
 989     p.edAddCFLink(n1,n2)
 990     p.edAddLink(o2,i3)
 991     p.saveSchema(fname)
 992     #
 993     l=loader.YACSLoader()
 994     p=l.load(fname)
 995     n1=p.getChildByName("n1")
 996     ex=pilot.ExecutorSwig()
 997     ex.setKeepGoingProperty(True)
 998     #
 999     startt=datetime.now()
1000     ex.RunW(p,0)
1001     t0=datetime.now()-startt
1002     #
1003     self.assertEqual(p.getState(),pilot.FAILED)
1004     self.assertEqual(n1.getState(),pilot.FAILED)
1005     n1.edGetSeqOfSamplesPort().getPyObj()
1006     a,b,c=n1.getPassedResults(ex)
1007
1008     self.assertEqual(a,[0,2,4,6,8,10])
1009     self.assertEqual([elt.getPyObj() for elt in b],[[0L,4L,8L,12L,16L,20L]])
1010
1011     p.getChildByName("n0").setScript("o0=[ 3*elt for elt in range(12) ]")
1012     p.getChildByName("n1").getChildByName("n10").setScript("""
1013 import time
1014 if i1%2==0:
1015   raise Exception("Simulated error !")
1016 else:
1017   time.sleep(1)
1018   o2=5*i1
1019 """)
1020     p.resetState(1)
1021
1022
1023     p.getChildByName("n1").assignPassedResults(a,b,c)
1024     p.exUpdateState();
1025     #
1026     startt=datetime.now()
1027     ex.RunW(p,0,False)
1028     t1=datetime.now()-startt
1029     #
1030     self.assertEqual(n1.getState(),pilot.DONE)
1031     self.assertEqual(p.getState(),pilot.DONE)
1032     self.assertEqual(p.getChildByName("n2").getOutputPort("o4").getPyObj(),[0L,5L,4L,15L,8L,25L,12L,35L,16L,45L,20L,55L])
1033     pass
1034
1035   def test12(self):
1036     """ Test of nested ForEachLoop with a port connected inside and outside the loop."""
1037     schema = self.r.createProc("schema")
1038     ti = schema.getTypeCode("int")
1039     tiset = schema.createSequenceTc("", "seqint", ti)
1040     tisetseq = schema.createSequenceTc("", "seqintvec", tiset)
1041
1042     n1 = self.r.createScriptNode("", "PyScript2")
1043     n1.edAddInputPort("i3", ti)
1044     n1.edAddInputPort("i4", ti)
1045     n1.edAddOutputPort("o5", ti)
1046     n1.setScript("o5=i3+i4")
1047
1048     n2 = self.r.createScriptNode("", "PyScript1")
1049     n2.edAddInputPort("i2", ti)
1050     n2.edAddOutputPort("o3", ti)
1051     n2.setScript("o3=i2")
1052
1053     b1 = self.r.createBloc("Bloc1")
1054     b1.edAddChild(n1)
1055     b1.edAddChild(n2)
1056
1057     fe1 = self.r.createForEachLoop("ForEach1", ti)
1058     fe1.getInputPort("nbBranches").edInitPy(2)
1059     fe1.getInputPort("SmplsCollection").edInitPy([1, 2, 3, 4])
1060     fe1.edSetNode(b1)
1061
1062     n3 = self.r.createScriptNode("", "PostProcessing")
1063     n3.edAddInputPort("i7", tiset)
1064     n3.edAddInputPort("i5", tiset)
1065     n3.edAddOutputPort("o4", ti)
1066     n3.setScript("""
1067 o4 = 0
1068 for i in i7:
1069     o4 = i + o4
1070
1071 for i in i5:
1072     o4 = i + o4
1073 """)
1074
1075     b0 = self.r.createBloc("Bloc0")
1076     b0.edAddChild(fe1)
1077     b0.edAddChild(n3)
1078
1079     fe0 = self.r.createForEachLoop("ForEach1", ti)
1080     fe0.getInputPort("nbBranches").edInitPy(2)
1081     fe0.getInputPort("SmplsCollection").edInitPy([1, 2, 3, 4])
1082     fe0.edSetNode(b0)
1083
1084     schema.edAddChild(fe0)
1085
1086     nx = self.r.createScriptNode("", "Result")
1087     nx.edAddInputPort("i8", tiset)
1088     nx.edAddOutputPort("o6", ti)
1089     nx.setScript("""
1090 o6 = 0
1091 for i in i8:
1092     o6 = i + o6
1093 """)
1094     schema.edAddChild(nx)
1095
1096     schema.edAddLink(fe1.getOutputPort("evalSamples"), n1.getInputPort("i3"))
1097     schema.edAddLink(fe0.getOutputPort("evalSamples"), n1.getInputPort("i4"))
1098
1099     schema.edAddDFLink(n1.getOutputPort("o5"), n3.getInputPort("i7"))
1100     schema.edAddDFLink(n2.getOutputPort("o3"), n3.getInputPort("i5"))
1101
1102     po5 = fe1.getOutputPort("Bloc1.PyScript2.o5")
1103     schema.edAddDFLink(po5, n2.getInputPort("i2"))
1104
1105     schema.edAddDFLink(n3.getOutputPort("o4"), nx.getInputPort("i8"))
1106 #    schema.saveSchema("foreach12.xml")
1107
1108     e = pilot.ExecutorSwig()
1109     e.RunW(schema)
1110     self.assertEqual(schema.getState(),pilot.DONE)
1111     resVal = schema.getChildByName("Result").getOutputPort("o6").getPyObj()
1112     self.assertEqual(resVal, 160)
1113     pass
1114
1115   def test13(self):
1116     """ Non regression test EDF11239. ForEach into ForEach. Problem on cloning linked to DeloymentTree.appendTask method that was too strong."""
1117     p=self.r.createProc("Bug11239")
1118     ti=p.createType("int","int")
1119     ti2=p.createSequenceTc("seqint","seqint",ti)
1120     #
1121     cont=p.createContainer("DefaultContainer","Salome")
1122     #
1123     node0=self.r.createForEachLoop("ForEachLoop_int0",ti)
1124     p.edAddChild(node0)
1125     node0.edGetSeqOfSamplesPort().edInitPy(range(4))
1126     node0.edGetNbOfBranchesPort().edInitInt(2)
1127     #
1128     node00=self.r.createBloc("Bloc0")
1129     node0.edAddChild(node00)
1130     node000_0=self.r.createForEachLoop("ForEachLoop_int1",ti)
1131     node00.edAddChild(node000_0)
1132     node000_0.edGetSeqOfSamplesPort().edInitPy(range(4))
1133     node000_0.edGetNbOfBranchesPort().edInitInt(3)
1134     #
1135     node0000=self.r.createBloc("Bloc1")
1136     node000_0.edAddChild(node0000)
1137     #
1138     node0000_0=self.r.createScriptNode("","PyScript2")
1139     node0000.edAddChild(node0000_0)
1140     i3=node0000_0.edAddInputPort("i3",ti)
1141     i4=node0000_0.edAddInputPort("i4",ti)
1142     o5=node0000_0.edAddOutputPort("o5",ti)
1143     node0000_0.setScript("o5 = i3 + i4")
1144     node0000_0.setContainer(cont)
1145     node0000_0.setExecutionMode("remote")
1146     p.edAddLink(node0.edGetSamplePort(),i3)
1147     p.edAddLink(node000_0.edGetSamplePort(),i4)
1148     #
1149     node0000_1=self.r.createScriptNode("","PyScript1")
1150     node0000.edAddChild(node0000_1)
1151     o3=node0000_1.edAddOutputPort("o3",ti)
1152     node0000_1.setScript("o3 = 7")
1153     node0000_1.setExecutionMode("local")
1154     p.edAddCFLink(node0000_0,node0000_1)
1155     #
1156     node000_1=self.r.createScriptNode("","PostTraitement")
1157     node00.edAddChild(node000_1)
1158     i7=node000_1.edAddInputPort("i7",ti2)
1159     i5=node000_1.edAddInputPort("i5",ti2)
1160     node000_1.setScript("for i in i7:\n    print i\nprint \"separation\"\nfor i in i5:\n    print i")
1161     node000_1.setContainer(cont)
1162     node000_1.setExecutionMode("remote")
1163     p.edAddLink(o5,i7)
1164     p.edAddLink(o3,i5)
1165     p.edAddCFLink(node000_0,node000_1)
1166     #
1167     #p.saveSchema("tmpp.xml")
1168     ex = pilot.ExecutorSwig()
1169     self.assertEqual(p.getState(),pilot.READY)
1170     ex.RunW(p,0)
1171     self.assertEqual(p.getState(),pilot.DONE)
1172     pass
1173
1174   def test14(self):
1175     """ Non regression EDF11027. Problem after Save/Load of a foreach node with type pyobj with input "SmplsCollection" manually set before. Correction in convertToYacsObjref from XML->Neutral. Objref can hide a string !"""
1176     xmlFileName="test14.xml"
1177     SALOMERuntime.RuntimeSALOME_setRuntime()
1178     r=pilot.getRuntime()
1179     n0=r.createProc("test23/zeRun")
1180     tp=n0.createInterfaceTc("python:obj:1.0","pyobj",[])
1181     tp2=n0.createSequenceTc("list[pyobj]","list[pyobj]",tp)
1182     n0bis=r.createBloc("test23/main") ; n0.edAddChild(n0bis)
1183     n00=r.createBloc("test23/run") ; n0bis.edAddChild(n00)
1184     #
1185     n000=r.createForEachLoop("test23/FE",tp) ; n00.edAddChild(n000)
1186     n0000=r.createScriptNode("Salome","test23/run_internal") ; n000.edSetNode(n0000)
1187     i0=n0000.edAddInputPort("i0",tp)
1188     i1=n0000.edAddInputPort("i1",tp) ; i1.edInitPy(3)
1189     o0=n0000.edAddOutputPort("o0",tp)
1190     n0000.setScript("o0=i0+i1")
1191     #
1192     n00.edAddLink(n000.edGetSamplePort(),i0)
1193     #
1194     n000.edGetSeqOfSamplesPort().edInitPy(range(10))
1195     n000.edGetNbOfBranchesPort().edInitInt(2)
1196     #
1197     n01=r.createScriptNode("Salome","test23/check") ; n0bis.edAddChild(n01)
1198     n0bis.edAddCFLink(n00,n01)
1199     i2=n01.edAddInputPort("i2",tp2)
1200     o1=n01.edAddOutputPort("o1",tp2)
1201     n01.setScript("o1=i2")
1202     n0bis.edAddLink(o0,i2)
1203     #
1204     n0.saveSchema(xmlFileName)
1205     #
1206     l=loader.YACSLoader()
1207     p=l.load(xmlFileName) # very import do not use n0 but use p instead !
1208     ex=pilot.ExecutorSwig()
1209     #
1210     self.assertEqual(p.getState(),pilot.READY)
1211     ex.RunW(p,0)
1212     self.assertEqual(p.getState(),pilot.DONE)
1213     self.assertEqual(p.getChildByName("test23/main.test23/check").getOutputPort("o1").getPyObj(),[3,4,5,6,7,8,9,10,11,12])
1214     pass
1215
1216   def test15(self):
1217     fname="BugInConcurrentLaunchDftCont.xml"
1218     p=self.r.createProc("pr")
1219     ti=p.createType("int","int")
1220     cont=p.createContainer("DefaultContainer","Salome")
1221     cont.setProperty("container_name","FactoryServer")
1222     b=self.r.createBloc("Bloc") ; p.edAddChild(b)
1223     #
1224     nb=4
1225     outs=[]
1226     for i in xrange(nb):
1227       node=self.r.createScriptNode("Salome","node%d"%i)
1228       node.setExecutionMode("remote")
1229       node.setContainer(cont)
1230       outs.append(node.edAddOutputPort("i",ti))
1231       node.setScript("i=%d"%i)
1232       b.edAddChild(node)
1233     #
1234     node=self.r.createScriptNode("Salome","nodeEnd")
1235     node.setExecutionMode("remote")
1236     node.setContainer(cont)
1237     res=node.edAddOutputPort("res",ti)
1238     p.edAddChild(node)
1239     l=[]
1240     for i in xrange(nb):
1241       elt="i%d"%i
1242       inp=node.edAddInputPort(elt,ti) ; l.append(elt)
1243       p.edAddChild(node)
1244       p.edAddLink(outs[i],inp)
1245     node.setScript("res="+"+".join(l))
1246     p.edAddCFLink(b,node)
1247     #
1248     for i in xrange(10):
1249       p.init()
1250       ex = pilot.ExecutorSwig()
1251       self.assertEqual(p.getState(),pilot.READY)
1252       ex.RunW(p,0)
1253       self.assertEqual(res.get(),6)
1254       self.assertEqual(p.getState(),pilot.DONE)
1255     pass
1256
1257   def test16(self):
1258     """ Test to check that a list[pyobj] outputport linked to pyobj inputport is OK."""
1259     SALOMERuntime.RuntimeSALOME_setRuntime()
1260     self.r=pilot.getRuntime()
1261     n0=self.r.createProc("test16/zeRun")
1262     tp=n0.createInterfaceTc("python:obj:1.0","pyobj",[])
1263     tp2=n0.createSequenceTc("list[pyobj]","list[pyobj]",tp)
1264
1265     n00=self.r.createScriptNode("Salome","n00") ; n0.edAddChild(n00)
1266     o0=n00.edAddOutputPort("o0",tp2)
1267     n00.setScript("o0=[[i+1] for i in xrange(8)]")
1268     n01=self.r.createScriptNode("Salome","n01") ; n0.edAddChild(n01)
1269     i1=n01.edAddInputPort("i1",tp)
1270     n01.setScript("assert(i1==[[1], [2], [3], [4], [5], [6], [7], [8]])")
1271     n0.edAddCFLink(n00,n01)
1272     n0.edAddLink(o0,i1)
1273     #
1274     ex=pilot.ExecutorSwig()
1275     self.assertEqual(n0.getState(),pilot.READY)
1276     ex.RunW(n0,0)
1277     self.assertEqual(n0.getState(),pilot.DONE)
1278     pass
1279
1280   def test17(self):
1281     """ Same as test16 except that tp2 is not list of tp but a list of copy of tp"""
1282     SALOMERuntime.RuntimeSALOME_setRuntime()
1283     self.r=pilot.getRuntime()
1284     n0=self.r.createProc("test17/zeRun")
1285     tp=n0.createInterfaceTc("python:obj:1.0","pyobj",[])
1286     tpp=n0.createInterfaceTc("python:obj:1.0","pyobj",[]) # diff is here
1287     tp2=n0.createSequenceTc("list[pyobj]","list[pyobj]",tpp)
1288
1289     n00=self.r.createScriptNode("Salome","n00") ; n0.edAddChild(n00)
1290     o0=n00.edAddOutputPort("o0",tp2)
1291     n00.setScript("o0=[[i+1] for i in xrange(8)]")
1292     n01=self.r.createScriptNode("Salome","n01") ; n0.edAddChild(n01)
1293     i1=n01.edAddInputPort("i1",tp)
1294     n01.setScript("assert(i1==[[1], [2], [3], [4], [5], [6], [7], [8]])")
1295     n0.edAddCFLink(n00,n01)
1296     n0.edAddLink(o0,i1)
1297     #
1298     ex=pilot.ExecutorSwig()
1299     self.assertEqual(n0.getState(),pilot.READY)
1300     ex.RunW(n0,0)
1301     self.assertEqual(n0.getState(),pilot.DONE)
1302     pass
1303
1304   def test18(self):
1305     p=self.r.createProc("prTest18")
1306     n00=self.r.createScriptNode("Salome","n00")
1307     self.assertEqual(n00.getMaxLevelOfParallelism(),1)
1308     n00.setExecutionMode("remote")
1309     self.assertEqual(n00.getMaxLevelOfParallelism(),1)
1310     cont=p.createContainer("gg","Salome")
1311     n00.setContainer(cont)
1312     self.assertEqual(n00.getMaxLevelOfParallelism(),1)
1313     cont.setProperty("nb_proc_per_nod","6")
1314     self.assertEqual(n00.getMaxLevelOfParallelism(),1)
1315     cont.setProperty("nb_proc_per_node","7")           # <- here
1316     self.assertEqual(n00.getMaxLevelOfParallelism(),7) # <- here
1317     pass
1318
1319   def test19(self):
1320     """This test checks the mechanism of YACS that allow PythonNodes to know their DynParaLoop context."""
1321     fname="test19.xml"
1322     l=loader.YACSLoader()
1323     #
1324     p=self.r.createProc("PROC")
1325     ti=p.createType("int","int")
1326     tdi=p.createSequenceTc("seqint","seqint",ti)
1327     # Level0
1328     fe0=self.r.createForEachLoop("FE0",ti) ; p.edAddChild(fe0)
1329     fe0.edGetNbOfBranchesPort().edInitInt(4)
1330     fe0_end=self.r.createScriptNode("Salome","fe0_end")
1331     fe0.edSetFinalizeNode(fe0_end)
1332     fe0_end.setScript("""assert([elt[0] for elt in my_dpl_localization]==["FE0"])
1333 assert(my_dpl_localization[0][1]>=0 and my_dpl_localization[0][1]<4)""")
1334     n0=self.r.createScriptNode("Salome","n0") ; p.edAddChild(n0)
1335     n0.setScript("o1=range(10)")
1336     a=n0.edAddOutputPort("o1",tdi)
1337     p.edAddLink(a,fe0.edGetSeqOfSamplesPort()) ; p.edAddCFLink(n0,fe0)
1338     # Level1
1339     b0=self.r.createBloc("b0") ; fe0.edAddChild(b0)
1340     n1=self.r.createScriptNode("Salome","n1") ; b0.edAddChild(n1)
1341     n1.setScript("""assert([elt[0] for elt in my_dpl_localization]==["FE0"])
1342 assert(my_dpl_localization[0][1]>=0 and my_dpl_localization[0][1]<4)
1343 o1=range(10)""")
1344     b=n1.edAddOutputPort("o1",tdi)
1345     fe1=self.r.createForEachLoop("FE1",ti) ; b0.edAddChild(fe1)
1346     fe1.edGetNbOfBranchesPort().edInitInt(3)
1347     fe1_end=self.r.createScriptNode("Salome","fe1_end")
1348     fe1_end.setScript("""assert([elt[0] for elt in my_dpl_localization]==["FE0.b0.FE1","FE0"])
1349 assert(my_dpl_localization[1][1]>=0 and my_dpl_localization[1][1]<4)
1350 assert(my_dpl_localization[0][1]>=0 and my_dpl_localization[0][1]<3)
1351 """)
1352     fe1.edSetFinalizeNode(fe1_end)
1353     p.edAddLink(b,fe1.edGetSeqOfSamplesPort()) ; p.edAddCFLink(n1,fe1)
1354     # Level2
1355     n2=self.r.createScriptNode("Salome","n2") ; fe1.edAddChild(n2)
1356     n2.setScript("""assert([elt[0] for elt in my_dpl_localization]==["FE0.b0.FE1","FE0"])
1357 assert(my_dpl_localization[1][1]>=0 and my_dpl_localization[1][1]<4)
1358 assert(my_dpl_localization[0][1]>=0 and my_dpl_localization[0][1]<3)
1359 """)
1360
1361     p.saveSchema(fname)
1362     ex=pilot.ExecutorSwig()
1363
1364     # local run of PythonNodes n1 and n2
1365     p=l.load(fname)
1366     p.init()
1367     self.assertEqual(p.getState(),pilot.READY)
1368     ex.setDPLScopeSensitive(True) # <- this line is the aim of the test
1369     ex.RunW(p,0)
1370     self.assertEqual(p.getState(),pilot.DONE)
1371
1372     # run remote
1373     p=l.load(fname)
1374     cont=p.createContainer("gg","HPSalome")
1375     cont.setSizeOfPool(2)
1376     n1=p.getChildByName("FE0.b0.n1") ; n1.setExecutionMode("remote") ; n1.setContainer(cont)
1377     n2=p.getChildByName("FE0.b0.FE1.n2") ; n2.setExecutionMode("remote") ; n2.setContainer(cont)
1378
1379     p.init()
1380     self.assertEqual(p.getState(),pilot.READY)
1381     ex.RunW(p,0)
1382     self.assertEqual(p.getState(),pilot.DONE)
1383     pass
1384
1385   def test20(self):
1386     """This test revealed a huge bug in ElementaryNode contained in a loop or foreach. The RECONNECT state generated invalid dependancies that only HPContainer can reveal the problem"""
1387     def assignCont(n,cont):
1388       n.setExecutionMode("remote") ; n.setContainer(cont)
1389       pass
1390     xmlFileName="test20.xml"
1391     p=self.r.createProc("test26")
1392     #
1393     cont=p.createContainer("gg","HPSalome") # very important ! HP Container needed for the test !
1394     cont.setSizeOfPool(8) # important make this figure >= 6
1395     #
1396     po=p.createInterfaceTc("python:obj:1.0","pyobj",[])
1397     sop=p.createSequenceTc("list[pyobj]","list[pyobj]",po)
1398     #
1399     b0=self.r.createBloc("test26/main") ; p.edAddChild(b0)
1400     n0=self.r.createScriptNode("Salome","test26/n0") ; assignCont(n0,cont) # 1
1401     n0.setScript("""import os
1402 dd=range(10)""")
1403     dd=n0.edAddOutputPort("dd",sop) ; b0.edAddChild(n0)
1404     fe0=self.r.createForEachLoop("test26/FE0",po) ; b0.edAddChild(fe0)
1405     fe0.edGetNbOfBranchesPort().edInitInt(1) # very important for the test : 1 !
1406     fe0i=self.r.createBloc("test26/FE0_internal") ; fe0.edSetNode(fe0i)
1407     zeArgInitNode2=self.r.createScriptNode("Salome","zeArgInitNode") ; assignCont(zeArgInitNode2,cont) # 2
1408     fe0i.edAddChild(zeArgInitNode2)
1409     c1=zeArgInitNode2.edAddInputPort("c",po)
1410     c2=zeArgInitNode2.edAddOutputPort("c",po)
1411     zeRun=self.r.createBloc("test26/zeRun") ; fe0i.edAddChild(zeRun)
1412     zeArgInitNode=self.r.createScriptNode("Salome","zeArgInitNode") ; assignCont(zeArgInitNode,cont) # 3
1413     zeRun.edAddChild(zeArgInitNode)
1414     ff1=zeArgInitNode.edAddInputPort("ff",po)
1415     ff2=zeArgInitNode.edAddOutputPort("ff",po)
1416     line01=self.r.createScriptNode("Salome","line01") ; zeRun.edAddChild(line01) ; assignCont(line01,cont) # 4
1417     line01.setScript("ee=3")
1418     ee0=line01.edAddOutputPort("ee",po)
1419     initt=self.r.createScriptNode("Salome","test26/initt") ; assignCont(initt,cont) # 5
1420     initt.setScript("pass") ; zeRun.edAddChild(initt)
1421     end=self.r.createScriptNode("Salome","test26/end") ; assignCont(end,cont) # 6
1422     end.setScript("import os") ; zeRun.edAddChild(end)
1423     retu=self.r.createScriptNode("Salome","return") ; assignCont(retu,cont) # 7
1424     retu.setScript("ee=i0") ; zeRun.edAddChild(retu)
1425     i0=retu.edAddInputPort("i0",po)
1426     ee=retu.edAddOutputPort("ee",po)
1427     zeRun.edAddCFLink(zeArgInitNode,line01)
1428     zeRun.edAddCFLink(line01,initt)
1429     zeRun.edAddCFLink(initt,end)
1430     zeRun.edAddCFLink(end,retu)
1431     p.edAddLink(ee0,i0)
1432     #
1433     returnn=self.r.createScriptNode("Salome","return") ; assignCont(returnn,cont) # 8
1434     returnn.setScript("elt=i0")
1435     i00=returnn.edAddInputPort("i0",po)
1436     elt=returnn.edAddOutputPort("elt",po)
1437     fe0i.edAddChild(returnn)
1438     fe0i.edAddCFLink(zeArgInitNode2,zeRun)
1439     fe0i.edAddCFLink(zeRun,returnn)
1440     p.edAddLink(c2,ff1)
1441     p.edAddLink(ee,i00)
1442     #
1443     finalize=self.r.createScriptNode("Salome","test26/finalize") ; b0.edAddChild(finalize) ; assignCont(finalize,cont) # 9
1444     finalize.setScript("pass")
1445     b0.edAddCFLink(n0,fe0)
1446     b0.edAddCFLink(fe0,finalize)
1447     #
1448     p.edAddLink(dd,fe0.edGetSeqOfSamplesPort())
1449     p.edAddLink(fe0.edGetSamplePort(),c1)
1450     #
1451     #p.saveSchema(xmlFileName)
1452     p.getChildByName("test26/main.test26/FE0").edGetNbOfBranchesPort().edInitInt(1) # very important 1 !
1453     #
1454     ex=pilot.ExecutorSwig()
1455     self.assertEqual(p.getState(),pilot.READY)
1456     ex.RunW(p,0)
1457     self.assertEqual(p.getState(),pilot.DONE)
1458     pass
1459
1460   pass
1461
1462   def test21(self):
1463     "test if we restart from a saved state in a foreach loop"
1464     fname="test21.xml"
1465     xmlStateFileName="saveState21.xml"
1466     from datetime import datetime
1467     p=self.r.createProc("prTest21")
1468     cont=p.createContainer("gg","Salome")
1469     cont.setProperty("name","localhost")
1470     cont.setProperty("hostname","localhost")
1471     cont.setProperty("type","multi")
1472     td=p.createType("double","double")
1473     ti=p.createType("int","int")
1474     tsi=p.createSequenceTc("seqint","seqint",ti)
1475     tsd=p.createSequenceTc("seqdbl","seqdbl",td)
1476     n0=self.r.createScriptNode("","n0")
1477     o0=n0.edAddOutputPort("o0",tsi)
1478     n0.setScript("o0=[ elt for elt in range(6) ]")
1479     p.edAddChild(n0)
1480     n1=self.r.createForEachLoop("n1",ti)
1481     n10=self.r.createScriptNode("","n10")
1482     n10.setExecutionMode("remote")
1483     n10.setContainer(cont)
1484     n1.edAddChild(n10)
1485     n10.setScript("""
1486 import time
1487 time.sleep(2)
1488 o2=2*i1
1489 """)
1490     i1=n10.edAddInputPort("i1",ti)
1491     o2=n10.edAddOutputPort("o2",ti)
1492     p.edAddChild(n1)
1493     p.edAddLink(o0,n1.edGetSeqOfSamplesPort())
1494     p.edAddLink(n1.edGetSamplePort(),i1)
1495     p.edAddCFLink(n0,n1)
1496     n1.edGetNbOfBranchesPort().edInitPy(2)
1497     n2=self.r.createScriptNode("","n2")
1498     n2.setScript("o4=i3")
1499     i3=n2.edAddInputPort("i3",tsi)
1500     o4=n2.edAddOutputPort("o4",tsi)
1501     n2.setScript("o4=i3")
1502     p.edAddChild(n2)
1503     p.edAddCFLink(n1,n2)
1504     p.edAddLink(o2,i3)
1505     p.saveSchema(fname)
1506     #
1507     l=loader.YACSLoader()
1508     p=l.load(fname)
1509     n1=p.getChildByName("n1")
1510     ex=pilot.ExecutorSwig()
1511     ex.setKeepGoingProperty(True)
1512     #
1513     startt=datetime.now()
1514     import threading
1515     myRun=threading.Thread(None, ex.RunW, None, (p,0))
1516     myRun.start()
1517     import time
1518     time.sleep(5)
1519     SALOMERuntime.schemaSaveState(p, ex, xmlStateFileName)
1520     a,b,c=n1.getPassedResults(ex)
1521     myRun.join()
1522     t0=datetime.now()-startt
1523     #
1524     self.assertEqual(p.getState(),pilot.DONE)
1525     self.assertEqual(n1.getState(),pilot.DONE)
1526     self.assertEqual(a,[0,1])
1527     self.assertEqual([elt.getPyObj() for elt in b],[[0L,2L]])
1528     #
1529     p.getChildByName("n0").setScript("o0=[ 3*elt for elt in range(6) ]")
1530     p.getChildByName("n1").getChildByName("n10").setScript("""
1531 import time
1532 time.sleep(0.1)
1533 o2=5*i1
1534 """)
1535     loader.loadState(p, xmlStateFileName)
1536     p.resetState(1)
1537     p.getChildByName("n1").assignPassedResults(a,b,c)
1538     p.exUpdateState();
1539     #
1540     startt=datetime.now()
1541     ex.RunW(p,0,False)
1542     t1=datetime.now()-startt
1543     #
1544     self.assertEqual(n1.getState(),pilot.DONE)
1545     self.assertEqual(p.getState(),pilot.DONE)
1546     self.assertEqual(p.getChildByName("n2").getOutputPort("o4").getPyObj(),[0L,2L,10L,15L,20L,25L])
1547     pass
1548   pass
1549
1550   def test22(self):
1551     """Restart from a saved state in a foreach loop without using assignPassedResults.
1552        This test uses the files test21.xml and saveState21.xml produced by test21.
1553     """
1554     fname="test21.xml"
1555     xmlStateFileName="saveState21.xml"
1556
1557     ex=pilot.ExecutorSwig()
1558     l=loader.YACSLoader()
1559     q=l.load(fname)
1560     q.getChildByName("n0").setScript("o0=[ 3*elt for elt in range(6) ]")
1561     q.getChildByName("n1").getChildByName("n10").setScript("""
1562 import time
1563 time.sleep(0.1)
1564 print "execution n10:", i1
1565 o2=5*i1
1566 """)
1567     q.getChildByName("n2").setScript("""
1568 print "execution n2:", i3
1569 o4=i3
1570 """)
1571     loader.loadState(q, xmlStateFileName)
1572     q.resetState(1)
1573     q.exUpdateState()
1574     #
1575     ex.RunW(q,0,False)
1576     #
1577     self.assertEqual(q.getChildByName("n1").getState(),pilot.DONE)
1578     self.assertEqual(q.getState(),pilot.DONE)
1579     self.assertEqual(q.getChildByName("n2").getOutputPort("o4").getPyObj(),[0L,2L,10L,15L,20L,25L])
1580     pass
1581   pass
1582
1583 if __name__ == '__main__':
1584   import os,sys
1585   U = os.getenv('USER')
1586   f=open("/tmp/" + U + "/UnitTestsResult", 'a')
1587   f.write("  --- TEST src/yacsloader: testSaveLoadRun.py\n")
1588   suite = unittest.makeSuite(TestSaveLoadRun)
1589   result=unittest.TextTestRunner(f, descriptions=1, verbosity=1).run(suite)
1590   f.close()
1591   sys.exit(not result.wasSuccessful())