def testOperandsWithoutPrepareInputs(self):
self.start_processes(etcd=False, modules=['mars.scheduler.tests.integrated.no_prepare_op'])
session_id = uuid.uuid1()
actor_client = new_client()
session_ref = actor_client.actor_ref(self.session_manager_ref.create_session(session_id))
actor_address = self.cluster_info.get_scheduler(ResourceActor.default_uid())
resource_ref = actor_client.actor_ref(ResourceActor.default_uid(), address=actor_address)
worker_endpoints = resource_ref.get_worker_endpoints()
t1 = mt.random.rand(10)
t1.op._expect_worker = worker_endpoints[0]
t2 = mt.random.rand(10)
t2.op._expect_worker = worker_endpoints[1]
t = NoPrepareOperand().new_tileable([t1, t2])
t.op._prepare_inputs = [False, False]
graph = t.build_graph()
targets = [t.key]
graph_key = uuid.uuid1()
session_ref.submit_tileable_graph(json.dumps(graph.to_json()),
graph_key, target_tileables=targets)
state = self.wait_for_termination(actor_client, session_ref, graph_key)
self.assertEqual(state, GraphState.SUCCEEDED)
def testOperandsWithoutPrepareInputs(self):
self.start_processes(etcd=False, modules=['mars.scheduler.tests.integrated.no_prepare_op'])
sess = new_session(self.session_manager_ref.address)
actor_address = self.cluster_info.get_scheduler(ResourceActor.default_uid())
resource_ref = sess._api.actor_client.actor_ref(ResourceActor.default_uid(), address=actor_address)
worker_endpoints = resource_ref.get_worker_endpoints()
t1 = mt.random.rand(10)
t1.op._expect_worker = worker_endpoints[0]
t2 = mt.random.rand(10)
t2.op._expect_worker = worker_endpoints[1]
t = NoPrepareOperand().new_tileable([t1, t2])
t.op._prepare_inputs = [False, False]
t.execute(session=sess, timeout=self.timeout)