class Tutorial1Island(object):
"""Manage the tutorial example in an island mode.
to launch 4 islands in local, type:
ipcluster local -n 4
It's up to you to launch the engine befoir launching gp.
"""
def __init__(self):
"""Initialize the island computation.
Each island must have the code in its classpath
"""
#Launch cluster
self._process = subprocess.Popen("ipcluster start --n=4", shell=True, stdout=subprocess.PIPE)
time.sleep(10)
init_script = """from pystepx.tutorials.functions_tutorial_island import *;"""
self._pystepx = PySTEPXIsland(nb_islands=4, init_script=init_script)
def run(self):
self._pystepx.evolve()
class TestPystepIsland(unittest.TestCase):
def __init__(self, a):
"""Start the processes."""
super(TestPystepIsland, self).__init__(a)
def setUp(self):
"""Initilize each thing."""
return
print 'Launch 4 clusters'
subprocess.Popen("ipcluster start --n=4", shell=True, stdout=subprocess.PIPE)
time.sleep(6)
def tearDown(self):
print 'Close 4 clusters'
return
subprocess.Popen("ipcluster stop", shell=True, stdout=subprocess.PIPE)
time.sleep(3)
def test_initialisation(self):
"""Test if we create correctly the objects"""
self._pystepx = PySTEPXIsland(nb_islands=4, init_script=init_script)
self._pystepx.__parametrize__()
# Test if functions are equal
f1 = pystepx.tutorials.functions_tutorial_island.treeRules
for f2 in self._pystepx._rc[:]['treeRules']:
self.assertEqual(f1, f2)
def test_population_movements_without_compilation(self):
"""Verify if we are able to move population while not compiling objects"""
self._pystepx = PySTEPXIsland(nb_islands=4, init_script=init_script)
self._pystepx._rc[0].execute('elems = gp_engine.get_evolver().select_and_remove_individuals(0.01, False)',
block=True)
print self._pystepx._rc[0]['elems']
def test_population_movements_with_compilation(self):
"""Verify if we are able to move population while compiling objects"""
self._pystepx = PySTEPXIsland(nb_islands=4, init_script=init_script)
print self._pystepx._rc[0]['gp_engine']
self._pystepx._rc[0].execute('elems = gp_engine.get_evolver().select_and_remove_individuals(0.01)',
block=True)
print self._pystepx._rc[0]['elems']
def test_evolution(self):
"""Test if we are able to evolve"""
self._pystepx = PySTEPXIsland(nb_islands=4, init_script=init_script)
self._pystepx.evolve()