code: def test_sampler(self, sampler, kwargs):
      if FLAGS.test_tfp_jax_prng == "rbg" and sampler == samplers.gamma:
          self.skipTest("gamma sampler not implemented for rbg PRNG.")
      seed = test_util.test_seed(sampler_type="stateless")
      s1 = sampler(seed=seed, **kwargs)
      s2 = sampler(seed=seed, **kwargs)
      self.assertAllEqual(s1, s2)
  
      # We don't test these scenarios for numpy, jax, where we don't support
      # stateful sampling.
      if not JAX_MODE and not NUMPY_MODE:
          self.verify_tf_behavior_match(sampler, kwargs)
  
file_path: probability/tensorflow_probability/python/internal/samplers_test.py
imports: from absl import flags
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 9 for calling the RandomTest obj's underlying member method code with example usage: obj.test_sampler(sampler, kwargs) without return types