code: def test_get_states(self):
      states_expected = {
          "VisitToAsia": ["NoVisit", "Visit"],
          "Tuberculosis": ["Absent", "Present"],
          "Smoking": ["NonSmoker", "Smoker"],
          "LungCancer": ["Absent", "Present"],
          "TbOrCa": ["Nothing", "CancerORTuberculosis"],
          "XRay": ["Normal", "Abnormal"],
          "Bronchitis": ["Absent", "Present"],
          "Dyspnea": ["Absent", "Present"],
      }
      states = self.reader.get_states()
      for variable in states_expected:
          self.assertListEqual(
              states_expected[variable], states[variable]
          )
  
imports: import unittest
  import numpy
  from pgmpy import config
  from pgmpy.readwrite import NETReader, NETWriter
  from pgmpy.utils import compat_fns, get_example_model
label_desc: Use image node_id 3 for calling the TestNETReader obj's underlying member method code with example usage: obj.test_get_states() without return types