code: def test_out_of_mem_raises_error(self):
      self.wrapper.eval()
      input = torch.stack((self.dataset[0][0], self.dataset[1][0]))
      with pytest.raises(RuntimeError) as e_info:
          self.wrapper.predict_on_batch(input, 0, False)
      assert (
          "CUDA ran out of memory while BaaL tried to replicate data"
          in str(e_info.value)
      )
  
imports: import math
  import unittest
  from unittest.mock import Mock
  import numpy
  import pytest
  import torch
  import torchmetrics
  from torch import nn
  from torch.utils.data import Dataset, DataLoader
  from baal.modelwrapper import ModelWrapper, mc_inference
  from baal.utils.metrics import ClassificationReport
label_desc: Use image node_id 5 for calling the ModelWrapperMultiOutMultiInTest obj's underlying member method code with example usage: obj.test_out_of_mem_raises_error() without return types