class_docstr: Test public repo.
class_docstr_tok: ["Test","public","repo","."]
code: def test_input_pipeline(self):
      crop_size = 128
      hific = FakeHiFiC(
          configs.get_config("hific"), mode=helpers.ModelMode.TRAINING
      )
      ds = hific.build_input(
          batch_size=2,
          crop_size=crop_size,
          tfds_arguments=helpers.TFDSArguments(
              dataset_name="", features_key="", downloads_dir=""
          ),
      )
      iterator = tf.data.make_initializable_iterator(ds)
      ds_next = iterator.get_next()
      with tf.Session() as sess:
          sess.run(iterator.initializer)
          image = sess.run(ds_next)
      self.assertEqual(
          image["input_image"].shape[1:3], (crop_size, crop_size)
      )
  
file_path: compression/models/hific/hific_test.py
imports: import contextlib
  import tempfile
  import numpy
  import tensorflow.compat.v1
  from hific import configs
  from hific import helpers
  from hific import model
  from hific import train
label_desc: Use image node_id 2 for calling the HiFiCTest obj's underlying member method code with example usage: obj.test_input_pipeline() without return types