def test_rpc_communicator_initialize_timeout(mock_impl, mock_grpc_server): comm = RpcCommunicator(timeout_wait=0.25) comm.unity_to_external.parent_conn.poll.return_value = None input = UnityInputProto() # Expect a timeout with pytest.raises(UnityTimeOutException): comm.initialize(input) comm.unity_to_external.parent_conn.poll.assert_called()
def test_rpc_communicator_initialize_callback(mock_impl, mock_grpc_server): def callback(): raise UnityEnvironmentException comm = RpcCommunicator(timeout_wait=0.25) comm.unity_to_external.parent_conn.poll.return_value = None input = UnityInputProto() # Expect a timeout with pytest.raises(UnityEnvironmentException): comm.initialize(input, poll_callback=callback) comm.unity_to_external.parent_conn.poll.assert_called()
def _wrap_unity_input(rl_input: UnityRLInputProto) -> UnityInputProto: result = UnityInputProto() result.rl_input.CopyFrom(rl_input) return result
def _send_academy_parameters( self, init_parameters: UnityRLInitializationInputProto ) -> UnityOutputProto: inputs = UnityInputProto() inputs.rl_initialization_input.CopyFrom(init_parameters) return self._communicator.initialize(inputs)
def test_rpc_communicator_initialize_OK(mock_impl, mock_grpc_server): comm = RpcCommunicator(timeout_wait=0.25) comm.unity_to_external.parent_conn.poll.return_value = True input = UnityInputProto() comm.initialize(input) comm.unity_to_external.parent_conn.poll.assert_called()