code: def test_initialize(X):
      d = 4
      model = KMeans(d)
  
      assert model.d is None
      assert model._initialized == False
      assert_raises(AttributeError, getattr, model, "_w_sum")
      assert_raises(AttributeError, getattr, model, "_xw_sum")
  
      model._initialize(X)
      assert model._initialized == True
      assert model.centroids.shape == (4, 3)
      assert model.d == 3
      assert_array_almost_equal(
          model._w_sum,
          [
              [0.0, 0.0, 0.0],
              [0.0, 0.0, 0.0],
              [0.0, 0.0, 0.0],
              [0.0, 0.0, 0.0],
          ],
      )
      assert_array_almost_equal(
          model._xw_sum,
          [
              [0.0, 0.0, 0.0],
              [0.0, 0.0, 0.0],
              [0.0, 0.0, 0.0],
              [0.0, 0.0, 0.0],
          ],
      )
  
imports: import os
  import numpy
  import torch
  import pytest
  from pomegranate.kmeans import KMeans
  from .distributions._utils import _test_initialization_raises_one_parameter
  from .distributions._utils import _test_initialization
  from .distributions._utils import _test_predictions
  from .distributions._utils import _test_efd_from_summaries
  from .distributions._utils import _test_raises
  from nose.tools import assert_raises
  from numpy.testing import assert_array_almost_equal
label_desc: Use image node_id 8 for calling a global function with example usage: test_initialize(X) without return types