code: def test_sobol_sample_with_dtype(self):
      for dtype in [tf.float32, tf.float64]:
          expected = tf.constant(
              [
                  [0.000, 0.000, 0.000, 0.000, 0.000, 0.000],
                  [0.500, 0.500, 0.500, 0.500, 0.500, 0.500],
                  [0.250, 0.750, 0.750, 0.750, 0.250, 0.250],
                  [0.750, 0.250, 0.250, 0.250, 0.750, 0.750],
                  [0.125, 0.625, 0.375, 0.125, 0.125, 0.375],
                  [0.625, 0.125, 0.875, 0.625, 0.625, 0.875],
                  [0.375, 0.375, 0.625, 0.875, 0.375, 0.125],
                  [0.875, 0.875, 0.125, 0.375, 0.875, 0.625],
              ],
              dtype=dtype,
          )
  
          actual = qmc.sobol_sample(
              6, 8, validate_args=True, dtype=dtype
          )
  
          self.assertAllClose(
              self.evaluate(actual), self.evaluate(expected), rtol=1e-6
          )
          self.assertEqual(actual.dtype, expected.dtype)
  
file_path: tf-quant-finance/tf_quant_finance/math/qmc/sobol_test.py
imports: import tensorflow.compat.v2
  import tensorflow_probability
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 5 for calling the SobolTest obj's underlying member method code with example usage: obj.test_sobol_sample_with_dtype() without return types