code: def test_high_probability_value_selected_as_expected(self):
      a = np.zeros([2, 2, 2, 2], dtype=np.float)
      a[0, 1, 1, 1] = 1.0
  
      pmf = PMF(
          {
              "Flu": [False, True],
              "Meningitis": [False, True],
              "Light Sensitivity": [False, True],
              "Fever": [False, True],
          },
          a,
      )
      choice = pmf.select()
      expected_choice = {
          "Flu": False,
          "Meningitis": True,
          "Light Sensitivity": True,
          "Fever": True,
      }
      self.assertDictEqual(expected_choice, choice)
  
file_path: typedb-ml/examples/diagnosis/dataset/pmf_test.py
imports: import unittest
  import numpy
  import pandas
  from pmf import PMF
label_desc: Use image node_id 1 for calling the TestPMF obj's underlying member method code with example usage: obj.test_high_probability_value_selected_as_expected() without return types