class_docstr: Base class for setting up and testing SqlDataset.
class_docstr_tok: ["Base","class","for","setting","up","and","testing","SqlDataset","."]
code: def _createSqlDataset(
      self, query, output_types, driver_name="sqlite", num_repeats=1
  ):
      dataset = readers.SqlDataset(
          driver_name, self.data_source_name, query, output_types
      ).repeat(num_repeats)
      return dataset
  
imports: import os
  from absl.testing import parameterized
  import sqlite3
  from tensorflow.python.data.experimental.ops import readers
  from tensorflow.python.data.kernel_tests import checkpoint_test_base
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the SqlDatasetTestBase obj's underlying member method code with example usage: obj._createSqlDataset(query, output_types, driver_name, num_repeats) and returns: dataset