code: def test_class_balance_make_cylinder_bell_funnel(
      params, class_balance_desired
  ):
      """Test that the class balance is the expected one."""
      X, y = make_cylinder_bell_funnel(**params)
      class_balance_actual = np.bincount(y)
      np.testing.assert_array_equal(
          class_balance_actual, class_balance_desired
      )
  
imports: import numpy
  import pytest
  import re
  from pyts.datasets import make_cylinder_bell_funnel
label_desc: Use image node_id 2 for calling a global function with example usage: test_class_balance_make_cylinder_bell_funnel(params, class_balance_desired) without return types