code: def testSampleLikeArgsGetDistDType(self):
      dist = johnson_su.JohnsonSU(1.0, 2.0, 0.0, 1.0)
      self.assertEqual(tf.float32, dist.dtype)
      for method in (
          "log_prob",
          "prob",
          "log_cdf",
          "cdf",
          "log_survival_function",
          "survival_function",
          "quantile",
      ):
          self.assertEqual(
              tf.float32, getattr(dist, method)(1).dtype, method
          )
  
imports: import math
  import numpy
  from scipy import stats
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import johnson_su
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 4 for calling the JohnsonSUTest obj's underlying member method code with example usage: obj.testSampleLikeArgsGetDistDType() without return types