code: def test_calibration(self):
      before_calib_param_init = list(
          map(
              lambda x: x.clone(),
              _get_first_module(
                  self.calibrator.wrapper.model
              ).parameters(),
          )
      )
      before_calib_param = list(
          map(
              lambda x: x.clone(),
              self.calibrator.calibrated_model.parameters(),
          )
      )
  
      self.calibrator.calibrate(
          self.dataset,
          self.dataset,
          batch_size=10,
          epoch=5,
          use_cuda=False,
          double_fit=False,
          workers=0,
      )
      after_calib_param_init = list(
          map(
              lambda x: x.clone(),
              _get_first_module(
                  self.calibrator.wrapper.model
              ).parameters(),
          )
      )
      after_calib_param = list(
          map(
              lambda x: x.clone(),
              self.calibrator.calibrated_model.parameters(),
          )
      )
  
      assert all(
          [
              np.allclose(i.detach(), j.detach())
              for i, j in zip(
                  before_calib_param_init, after_calib_param_init
              )
          ]
      )
  
      assert not all(
          [
              np.allclose(i.detach(), j.detach())
              for i, j in zip(before_calib_param, after_calib_param)
          ]
      )
  
imports: import unittest
  from unittest.mock import Mock
  import numpy
  import pytest
  import torch
  from torch import nn
  from torch.utils.data import Dataset
  from baal.calibration import DirichletCalibrator
  from baal.modelwrapper import ModelWrapper
label_desc: Use image node_id 3 for calling the CalibrationTest obj's underlying member method code with example usage: obj.test_calibration() without return types