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)