code: def test_split_row_set(self):
      os.environ["BIGTABLE_EMULATOR_HOST"] = self.emulator.get_addr()
      self.emulator.create_table(
          "fake_project",
          "fake_instance",
          "test-table",
          ["fam1", "fam2"],
          splits=[
              "row005",
              "row010",
              "row015",
              "row020",
              "row025",
              "row030",
          ],
      )
  
      values = [[f"[{i,j}]" for j in range(2)] for i in range(40)]
  
      ten = tf.constant(values)
  
      client = BigtableClient("fake_project", "fake_instance")
  
      self.emulator.write_tensor(
          "fake_project",
          "fake_instance",
          "test-table",
          ten,
          ["row" + str(i).rjust(3, "0") for i in range(40)],
          ["fam1:col1", "fam2:col2"],
      )
  
      rs = row_set.from_rows_or_ranges(row_range.infinite())
  
      num_parallel_calls = 2
      samples = [
          s
          for s in core_ops.bigtable_split_row_set_evenly(
              client._client_resource,
              rs._impl,
              "test-table",
              num_parallel_calls,
          )
      ]
      self.assertEqual(len(samples), num_parallel_calls)
  
      num_parallel_calls = 6
      samples = [
          s
          for s in core_ops.bigtable_split_row_set_evenly(
              client._client_resource,
              rs._impl,
              "test-table",
              num_parallel_calls,
          )
      ]
  
      # The emulator may return different samples each time, so we can't
      # expect an exact number, but it must be no more than num_parallel_calls
      self.assertLessEqual(len(samples), num_parallel_calls)
  
      num_parallel_calls = 1
      samples = [
          s
          for s in core_ops.bigtable_split_row_set_evenly(
              client._client_resource,
              rs._impl,
              "test-table",
              num_parallel_calls,
          )
      ]
      self.assertEqual(len(samples), num_parallel_calls)
  
file_path: io/tests/test_bigtable/test_parallel_read_rows.py
imports: import os
  from re import escape
  from .bigtable_emulator import BigtableEmulator
  from tensorflow_io.python.ops import core_ops
  from tensorflow_io.python.ops.bigtable.bigtable_dataset_ops import BigtableClient
  import tensorflow_io.python.ops.bigtable.bigtable_row_range
  import tensorflow_io.python.ops.bigtable.bigtable_row_set
  import tensorflow
  from tensorflow import test
  import pytest
  import sys
label_desc: Use image node_id 5 for calling the BigtableParallelReadTest obj's underlying member method code with example usage: obj.test_split_row_set() without return types