def SetCardinalities( self, request: executor_pb2.SetCardinalitiesRequest, context: grpc.ServicerContext, ) -> executor_pb2.SetCardinalitiesResponse: """Sets the cartinality for the executor service.""" py_typecheck.check_type(request, executor_pb2.SetCardinalitiesRequest) try: cardinalities_dict = executor_serialization.deserialize_cardinalities( request.cardinalities) self._executor = self._ex_factory.create_executor(cardinalities_dict) return executor_pb2.SetCardinalitiesResponse() except (ValueError, TypeError) as err: _set_invalid_arg_err(context, err) return executor_pb2.SetCardinalitiesResponse()
def test_set_cardinalities_returns_none(self, mock_stub): response = executor_pb2.SetCardinalitiesResponse() instance = mock_stub.return_value instance.SetCardinalities = mock.Mock(side_effect=[response]) executor = create_remote_executor() result = executor.set_cardinalities({placements.CLIENTS: 3}) self.assertIsNone(result)
def test_set_cardinalities_returns_none(self, mock_stub): response = executor_pb2.ExecuteResponse( set_cardinalities=executor_pb2.SetCardinalitiesResponse()) executor = _setup_mock_streaming_executor(mock_stub, response) loop = asyncio.get_event_loop() result = loop.run_until_complete( executor.set_cardinalities({placement_literals.CLIENTS: 3})) self.assertIsNone(result)
def test_set_cardinalities_returns_none(self, mock_stub): response = executor_pb2.SetCardinalitiesResponse() instance = mock_stub.return_value instance.SetCardinalities = mock.Mock(side_effect=[response]) executor = create_remote_executor() loop = asyncio.get_event_loop() result = loop.run_until_complete( executor.set_cardinalities({placement_literals.CLIENTS: 3})) self.assertIsNone(result)