1 # Copyright (C) 2006-2017 CEA/DEN, EDF R&D
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.
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.
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
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 class TestPlayGround0(unittest.TestCase):
25 pg=pilot.PlayGround([("a0",28),("a1",28),("a2",28)])
26 pd=pilot.ContigPartDefinition(pg,4,0,3*28)
27 res=pg.partition([(pd,1.),(pd,1.)])
29 assert(isinstance(res[0],pilot.ContigPartDefinition))
30 assert(isinstance(res[1],pilot.ContigPartDefinition))
31 assert(res[0].getStart()==0 and res[0].getStop()==40)
32 assert(res[1].getStart()==40 and res[1].getStop()==84)
33 assert(sum([elt.getNumberOfCoresConsumed() for elt in res])==pg.getNumberOfCoresAvailable())
34 pd2=pilot.AllPartDefinition(pg,4)
35 assert(pd2.getNumberOfCoresConsumed()==84)
36 res=pg.partition([(pd2,1.),(pd2,1.),(pd2,1.)])
38 assert(isinstance(res[0],pilot.ContigPartDefinition))
39 assert(isinstance(res[1],pilot.ContigPartDefinition))
40 assert(isinstance(res[2],pilot.ContigPartDefinition))
41 assert(res[0].getStart()==0 and res[0].getStop()==28)
42 assert(res[1].getStart()==28 and res[1].getStop()==56)
43 assert(res[2].getStart()==56 and res[2].getStop()==84)
45 pg.setData([("a0",2),("a1",8),("a2",8)])
46 res=pg.partition([(pilot.AllPartDefinition(pg,4),1.),(pilot.AllPartDefinition(pg,1),1.)])
48 assert(isinstance(res[0],pilot.ContigPartDefinition))
49 assert(isinstance(res[1],pilot.NonContigPartDefinition))
50 assert(res[0].getStart()==2 and res[0].getStop()==10)
51 assert(res[1].getIDs()==(0,1,10,11,12,13,14,15,16,17))
55 """ test focused on complicated cut due to lack of cores"""
56 pg=pilot.PlayGround([("a0",13)])
57 pd=pilot.ContigPartDefinition(pg,4,0,13)
58 res=pg.partition([(pd,1.),(pd,1.)])
60 assert(isinstance(res[0],pilot.ContigPartDefinition) and isinstance(res[1],pilot.ContigPartDefinition))
61 assert(res[0].getStart()==0 and res[0].getStop()==4)
62 assert(res[1].getStart()==4 and res[1].getStop()==12)# 1 core lost
64 pg=pilot.PlayGround([("a0",2),("a1",11)])
65 pd=pilot.ContigPartDefinition(pg,4,0,13)
66 res=pg.partition([(pd,1.),(pd,1.)])
68 assert(isinstance(res[0],pilot.ContigPartDefinition) and isinstance(res[1],pilot.ContigPartDefinition))
69 assert(res[0].getStart()==2 and res[0].getStop()==6)
70 assert(res[1].getStart()==6 and res[1].getStop()==10)# 5 cores lost
75 if __name__ == '__main__':