code: def test_probability_dtypes(X, probs):
      X = numpy.array(X)
      X_int = numpy.array(X, dtype=numpy.int32)
  
      y = Bernoulli(probs).probability(X)
      assert y.dtype == torch.float32
  
      y = Bernoulli(probs).probability(X_int)
      assert y.dtype == torch.float32
  
      X = X.astype(numpy.float64)
      X_int = X.astype("int32")
  
      y = Bernoulli(
          numpy.array(probs, dtype=numpy.float64)
      ).probability(X)
      assert y.dtype == torch.float64
  
      y = Bernoulli(
          numpy.array(probs, dtype=numpy.float64)
      ).probability(X_int)
      assert y.dtype == torch.float64
  
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 13 for calling a global function with example usage: test_probability_dtypes(X, probs) without return types