code: def testSampleLikeArgsGetDistDType(self):
      dist = self.make_two_piece_normal()
  
      self.assertEqual(self.dtype, dist.dtype)
  
      seed = test_util.test_seed()
      self.assertEqual(self.dtype, dist.sample(1, seed=seed).dtype)
  
      for method in (
          "prob",
          "cdf",
          "survival_function",
          "quantile",
          "log_prob",
          "log_cdf",
          "log_survival_function",
      ):
          self.assertEqual(
              self.dtype, getattr(dist, method)(1).dtype, msg=method
          )
  
      for method in ("mean", "variance", "mode"):
          self.assertEqual(
              self.dtype, getattr(dist, method)().dtype, msg=method
          )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import two_piece_normal
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 7 for calling the _TwoPieceNormalTest obj's underlying member method code with example usage: obj.testSampleLikeArgsGetDistDType() without return types