class_docstr: Tests of input_fn, doing end-to-end I/O.
  
  These tests instantiate an input stream and then check it in various ways,
  in increasing complexity.
class_docstr_tok: ["Tests","of","input_fn",",","doing","end-to-end","I\/O",".","These","tests","instantiate","an","input","stream","and","then","check","it","in","various","ways",",","in","increasing","complexity","."]
code: def testInputStream(self, use_tpu):
      # Read batch_feed one at a time, check the shape of each, and the
      # total count.
      batch_size = 1
      batch_feed = self.get_batch_feed(
          batch_size=batch_size, use_tpu=use_tpu
      )
      expected_n_batches = math.ceil(
          float(testdata.N_GOLDEN_CALLING_EXAMPLES) / batch_size
      )
      self.check_batch_feed(
          batch_feed, use_tpu, batch_size, expected_n_batches
      )
  
imports: import math
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow import estimator
  from deepvariant import data_providers
  from deepvariant import dv_config
  from deepvariant import dv_constants
  from deepvariant import dv_utils
  from deepvariant import testdata
  from deepvariant.protos import deepvariant_pb2
  from third_party.nucleus.io import tfrecord
  from third_party.nucleus.testing import test_utils
  from third_party.nucleus.util import variant_utils
  from tensorflow.core.example import example_pb2
label_desc: Use image node_id 3 for calling the InputTest obj's underlying member method code with example usage: obj.testInputStream(use_tpu) without return types