code: def test_known_values_small_dimension(self):
      # The first five elements of the non-randomized Sobol sequence
      # with dimension 2
      for dtype in [np.float16, np.float32, np.float64]:
          sample = random.sobol.sample(2, 5, dtype=dtype)
          # These are in the original order, not Gray code order.
          expected = np.array(
              [
                  [0.5, 0.5],
                  [0.25, 0.75],
                  [0.75, 0.25],
                  [0.125, 0.625],
                  [0.625, 0.125],
              ],
              dtype=dtype,
          )
          self.assertAllClose(
              expected, self.evaluate(sample), rtol=1e-6
          )
          self.assertEqual(sample.dtype.as_numpy_dtype, dtype)
  
file_path: tf-quant-finance/tf_quant_finance/math/random_ops/sobol/sobol_test.py
imports: import numpy
  import tensorflow.compat.v2
  import tensorflow_probability
  from tensorflow.python.framework import test_util
  from tf_quant_finance.math import random
label_desc: Use image node_id 1 for calling the SampleSobolSequenceTest obj's underlying member method code with example usage: obj.test_known_values_small_dimension() without return types