def test_reduce(self): def concat(a, b): return a + b pickled_function = cloudpickle.dumps(concat) store_locator = ErStoreLocator(store_type="levelDb", namespace="ns", name='name') job = ErJob(id="1", name="reduce", inputs=[ErStore(store_locator=store_locator)], functors=[ErFunctor(name="reduce", body=pickled_function)]) channel = grpc.insecure_channel( target='localhost:20000', options=[('grpc.max_send_message_length', -1), ('grpc.max_receive_message_length', -1)]) roll_pair_stub = command_pb2_grpc.CommandServiceStub(channel) request = ErCommandRequest( seq=1, uri='com.webank.eggroll.rollpair.component.RollPair.reduce', args=[job.to_proto().SerializeToString()]) result = roll_pair_stub.call(request.to_proto()) time.sleep(1200)
def test_scalar_mul_raw(self): def scalar_mul(v): pub_key, private_key = rpt_engine.keygen() return rpt_engine.slcmul(rpt_engine.load(v, 1, 1, 1), 2.0, pub_key, private_key) pickled_function = cloudpickle.dumps(scalar_mul) store_locator = ErStoreLocator(store_type="levelDb", namespace="ns", name='mat_a') functor = ErFunctor(name="mapValues", body=pickled_function) job = ErJob(id="1", name="mapValues", inputs=[ErStore(store_locator=store_locator)], functors=[functor]) channel = grpc.insecure_channel( target='localhost:20000', options=[('grpc.max_send_message_length', -1), ('grpc.max_receive_message_length', -1)]) roll_pair_stub = command_pb2_grpc.CommandServiceStub(channel) request = ErCommandRequest( seq=1, uri='com.webank.eggroll.rollpair.component.RollPair.mapValues', args=[job.to_proto().SerializeToString()]) # print(f"ready to call") result = roll_pair_stub.call(request.to_proto()) time.sleep(1200)