code: def test_sample(probs):
      torch.manual_seed(0)
  
      X = Bernoulli(probs).sample(1)
      assert_array_almost_equal(X, [[0, 1, 1]])
  
      X = Bernoulli(probs).sample(5)
      assert_array_almost_equal(
          X,
          [
              [0.0, 1.0, 0.0],
              [0.0, 0.0, 0.0],
              [0.0, 1.0, 0.0],
              [1.0, 1.0, 1.0],
              [0.0, 1.0, 0.0],
          ],
      )
  
imports: import os
  import numpy
  import torch
  import pytest
  from pomegranate.distributions import Bernoulli
  from ._utils import _test_initialization_raises_one_parameter
  from ._utils import _test_initialization
  from ._utils import _test_predictions
  from ._utils import _test_efd_from_summaries
  from ._utils import _test_raises
  from nose.tools import assert_raises
  from numpy.testing import assert_array_almost_equal
label_desc: Use image node_id 11 for calling a global function with example usage: test_sample(probs) without return types