code: def test_generate_bernoulli_distribution():
      bernoulli_array = distribution.generate_bernoulli_distribution(
          prob=0.5, size=10
      )
  
      assert bernoulli_array.shape == (10,)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_generate_bernoulli_distribution() without return types