def testLocalCluster(self):
endpoint = gen_endpoint('0.0.0.0')
with LocalDistributedCluster(endpoint,
scheduler_n_process=2,
worker_n_process=3) as cluster:
pool = cluster.pool
self.assertTrue(
pool.has_actor(pool.actor_ref(
ClusterInfoActor.default_name())))
self.assertTrue(
pool.has_actor(
pool.actor_ref(SessionManagerActor.default_name())))
self.assertTrue(
pool.has_actor(pool.actor_ref(DispatchActor.default_name())))
with new_session(endpoint) as session:
api = session._api
t = mt.ones((3, 3), chunk_size=2)
result = session.run(t)
np.testing.assert_array_equal(result, np.ones((3, 3)))
self.assertNotIn(session._session_id,
api.session_manager.get_sessions())
def testDistribute(self):
distributor = gen_distributor(2, 4)
idx = distributor.distribute(SessionActor.default_name())
self.assertEqual(idx, 1)
idx = distributor.distribute(
GraphActor.gen_uid('fake_session_id', 'fake_graph_key'))
self.assertEqual(idx, 0)
idx = distributor.distribute(DispatchActor.default_name())
self.assertEqual(idx, 2)
idx = distributor.distribute('w:1:mars-sender')
self.assertEqual(idx, 3)
