code: def test_summarize(X, probs):
      for probs_ in probs, None:
          d = ConditionalCategorical(probs_)
          d.summarize(X[:4])
          assert_array_almost_equal(d._w_sum[0], [0.0, 4.0])
          assert_array_almost_equal(d._w_sum[1], [0.0, 0.0, 4.0])
          assert_array_almost_equal(d._w_sum[2], [3.0, 1.0])
  
          assert_array_almost_equal(
              d._xw_sum[0], [[0.0, 0.0], [3.0, 1.0]]
          )
          assert_array_almost_equal(
              d._xw_sum[1], [[0.0, 0.0], [0.0, 0.0], [2.0, 2.0]]
          )
          assert_array_almost_equal(
              d._xw_sum[2], [[3.0, 0.0], [0.0, 1.0]]
          )
  
          d.summarize(X[4:])
          assert_array_almost_equal(d._w_sum[0], [1.0, 6.0])
          assert_array_almost_equal(d._w_sum[1], [0.0, 3.0, 4.0])
          assert_array_almost_equal(d._w_sum[2], [5.0, 2.0])
  
          assert_array_almost_equal(
              d._xw_sum[0], [[1.0, 0.0], [4.0, 2.0]]
          )
          assert_array_almost_equal(
              d._xw_sum[1], [[0.0, 0.0], [2.0, 1.0], [2.0, 2.0]]
          )
          assert_array_almost_equal(
              d._xw_sum[2], [[5.0, 0.0], [0.0, 2.0]]
          )
  
          d = ConditionalCategorical(probs_)
          d.summarize(X)
          assert_array_almost_equal(d._w_sum[0], [1.0, 6.0])
          assert_array_almost_equal(d._w_sum[1], [0.0, 3.0, 4.0])
          assert_array_almost_equal(d._w_sum[2], [5.0, 2.0])
  
          assert_array_almost_equal(
              d._xw_sum[0], [[1.0, 0.0], [4.0, 2.0]]
          )
          assert_array_almost_equal(
              d._xw_sum[1], [[0.0, 0.0], [2.0, 1.0], [2.0, 2.0]]
          )
          assert_array_almost_equal(
              d._xw_sum[2], [[5.0, 0.0], [0.0, 2.0]]
          )
  
imports: import numpy
  import torch
  import pytest
  from pomegranate.distributions import ConditionalCategorical
  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 17 for calling a global function with example usage: test_summarize(X, probs) without return types