code: def test_generate_choice_distribution():
      choice_array = distribution.generate_choice_distribution(
          n=10, probs=None, size=10
      )
  
      assert choice_array.shape == (10,)
  
label_desc: Use image node_id 2 for calling a global function with example usage: test_generate_choice_distribution() without return types