code: def test_kmeans():
      x = np.random.uniform(0, 1, (20, 2, 1))
  
      y = general.kmeans(x, n_clusters=3)
  
      assert y.shape[0] == 20
  
imports: import numpy
  from opytimizer.math import general
label_desc: Use image node_id 2 for calling a global function with example usage: test_kmeans() without return types