code: def test_InformationGainSegmentation(multivariate_mean_shift):
      """Test the InformationGainSegmentation."""
      igts = InformationGainSegmentation(k_max=3, step=1)
      assert igts.get_params() == {"k_max": 3, "step": 1}
      pred = igts.fit_predict(multivariate_mean_shift)
      assert np.array_equal(
          pred,
          np.array(
              [
                  0,
                  0,
                  0,
                  0,
                  0,
                  1,
                  1,
                  1,
                  1,
                  1,
                  2,
                  2,
                  2,
                  2,
                  2,
                  3,
                  3,
                  3,
                  3,
                  3,
              ],
              dtype=np.int32,
          ),
      )
      assert igts.change_points_ == [0, 5, 10, 15, 20]
      assert len(igts.intermediate_results_) == 3
  
file_path: sktime/sktime/annotation/tests/test_igts.py
imports: import numpy
  import pytest
  from sktime.annotation.igts import IGTS, InformationGainSegmentation, entropy
label_desc: Use image node_id 6 for calling a global function with example usage: test_InformationGainSegmentation(multivariate_mean_shift) without return types