code: def test_add_transition_model_bad_states(self):
      model = MC(["var"], [2])
      # transition for state=1 not defined
      transition_model = {0: {0: 0.1, 1: 0.9}}
      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 17 for calling the TestMarkovChain obj's underlying member method code with example usage: obj.test_add_transition_model_bad_states() without return types