code: def test_initialization_raises(probs):
      assert_raises(ValueError, JointCategorical, 0.3)
      assert_raises(ValueError, JointCategorical, probs, inertia=-0.4)
      assert_raises(ValueError, JointCategorical, probs, inertia=1.2)
      assert_raises(
          ValueError,
          JointCategorical,
          probs,
          inertia=1.2,
          frozen="true",
      )
      assert_raises(
          ValueError, JointCategorical, probs, inertia=1.2, frozen=3
      )
  
      assert_raises(ValueError, JointCategorical, inertia=-0.4)
      assert_raises(ValueError, JointCategorical, inertia=1.2)
      assert_raises(
          ValueError, JointCategorical, inertia=1.2, frozen="true"
      )
      assert_raises(ValueError, JointCategorical, inertia=1.2, frozen=3)
  
      assert_raises(
          ValueError, JointCategorical, numpy.array(probs) + 0.001
      )
      assert_raises(
          ValueError, JointCategorical, numpy.array(probs) - 0.001
      )
  
      p = numpy.array(probs)
      p[0, 0] = -0.03
      assert_raises(ValueError, JointCategorical, p)
  
      p = numpy.array(probs)
      p[0, 0] = 1.03
      assert_raises(ValueError, JointCategorical, p)
  
imports: import numpy
  import torch
  import pytest
  from pomegranate.distributions import JointCategorical
  from ._utils import _test_initialization_raises_one_parameter
  from ._utils import _test_initialization
  from ._utils import _test_predictions
  from ._utils import _test_raises
  from nose.tools import assert_raises
  from numpy.testing import assert_array_almost_equal
label_desc: Use image node_id 6 for calling a global function with example usage: test_initialization_raises(probs) without return types