code: def test_probability(X, model, model2):
      prob = model.probability(X)
      assert_array_almost_equal(
          prob,
          [
              1.663520e-04,
              2.242500e-04,
              5.004999e-04,
              2.102100e-04,
              2.935625e-04,
              4.384369e-05,
              4.968956e-05,
              1.751750e-04,
              2.935625e-04,
              5.979991e-05,
              8.768743e-05,
          ],
      )
  
      prob = model2.probability(X)
      assert_array_almost_equal(
          prob,
          [
              5.133332e-05,
              9.199999e-05,
              7.186660e-05,
              2.587201e-04,
              3.079999e-04,
              9.774998e-06,
              2.606666e-05,
              2.156001e-04,
              3.079999e-04,
              2.299997e-05,
              2.932498e-05,
          ],
      )
  
imports: import numpy
  import torch
  import pytest
  from pomegranate.factor_graph import FactorGraph
  from pomegranate.distributions import Exponential
  from pomegranate.distributions import Categorical
  from pomegranate.distributions import JointCategorical
  from nose.tools import assert_raises
  from numpy.testing import assert_array_almost_equal
label_desc: Use image node_id 14 for calling a global function with example usage: test_probability(X, model, model2) without return types