code: def testScipySpecialFunBernoulli(self, n):
      dtype = jax.numpy.zeros(0).dtype  # default float dtype.
      scipy_op = lambda: osp_special.bernoulli(n).astype(dtype)
      lax_op = functools.partial(lsp_special.bernoulli, n)
      args_maker = lambda: []
      self._CheckAgainstNumpy(
          scipy_op, lax_op, args_maker, atol=0, rtol=1e-5
      )
      self._CompileAndCheck(lax_op, args_maker, atol=0, rtol=1e-5)
  
file_path: jax/tests/lax_scipy_special_functions_test.py
imports: import collections
  import functools
  import itertools
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import scipy.special
  import jax
  from jax._src import test_util
  from jax.scipy import special
  from jax import config
label_desc: Use image node_id 3 for calling the LaxScipySpcialFunctionsTest obj's underlying member method code with example usage: obj.testScipySpecialFunBernoulli(n) without return types