code: def test_transition_model_bad_matrix_dimension(self):
      model = MC(["var"], [2])
      transition_model = np.array([0.3, 0.7])
      # check for square dimension of the matrix
      self.assertRaises(
          ValueError,
          model.add_transition_model,
          "var",
          transition_model,
      )
      transition_model = np.array([[0.3, 0.6, 0.1], [0.3, 0.3, 0.4]])
      self.assertRaises(
          ValueError,
          model.add_transition_model,
          "var",
          transition_model,
      )
  
imports: import sys
  import unittest
  import numpy
  from mock import call, patch
  from pandas import DataFrame
  from pgmpy.factors.discrete import State
  from pgmpy.models import MarkovChain
label_desc: Use image node_id 22 for calling the TestMarkovChain obj's underlying member method code with example usage: obj.test_transition_model_bad_matrix_dimension() without return types