code: def testSampleGPU(self):
      if not tf.test.is_gpu_available():
          self.skipTest("no GPU")
      with tf.device("GPU"):
          _, runtime = self.evaluate(
              poisson_lib.random_poisson(
                  shape=tf.constant([], dtype=tf.int32),
                  rates=tf.constant(10.0),
                  seed=test_util.test_seed(),
              )
          )
      self.assertEqual(
          implementation_selection._RUNTIME_DEFAULT, runtime
      )
  
imports: import numpy
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import poisson
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import implementation_selection
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 2 for calling the PoissonSamplingTest obj's underlying member method code with example usage: obj.testSampleGPU() without return types