code: def test_sobol_generating_matrices_with_dtype(self):
      dim = 5
      num_results = 31
      num_digits = 5  # ceil(log2(num_results))
  
      for dtype in [tf.int32, tf.int64]:
          expected = tf.constant(
              [
                  [16, 8, 4, 2, 1],
                  [16, 24, 20, 30, 17],
                  [16, 24, 12, 18, 29],
                  [16, 24, 4, 10, 31],
                  [16, 8, 4, 22, 31],
              ],
              dtype=dtype,
          )
  
          actual = qmc.sobol_generating_matrices(
              dim,
              num_results,
              num_digits,
              validate_args=True,
              dtype=dtype,
          )
  
          self.assertAllEqual(
              self.evaluate(actual), self.evaluate(expected)
          )
          self.assertEqual(actual.dtype, 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 7 for calling the SobolTest obj's underlying member method code with example usage: obj.test_sobol_generating_matrices_with_dtype() without return types