code: def test_dataset_yields_correct_sequence(self):
      ds = y4m_dataset.Y4MDataset(
          [self.tempfile_1.full_path, self.tempfile_2.full_path]
      )
      it = iter(ds)
  
      y, cbcr = next(it)
      self.assertEqual(tf.uint8, y.dtype)
      self.assertEqual(tf.uint8, cbcr.dtype)
      cb, cr = tf.unstack(cbcr, axis=-1)
      self.assertAllEqual(shaped_uint8(b"ABCDEFGH", (2, 4, 1)), y)
      self.assertAllEqual(shaped_uint8(b"IJ", (1, 2)), cb)
      self.assertAllEqual(shaped_uint8(b"KL", (1, 2)), cr)
  
      y, cbcr = next(it)
      self.assertEqual(tf.uint8, y.dtype)
      self.assertEqual(tf.uint8, cbcr.dtype)
      self.assertAllEqual(shaped_uint8(b"a", (1, 1, 1)), y)
      self.assertAllEqual(shaped_uint8(b"bc", (1, 1, 2)), cbcr)
  
      y, cbcr = next(it)
      self.assertEqual(tf.uint8, y.dtype)
      self.assertEqual(tf.uint8, cbcr.dtype)
      self.assertAllEqual(shaped_uint8(b"d", (1, 1, 1)), y)
      self.assertAllEqual(shaped_uint8(b"ef", (1, 1, 2)), cbcr)
  
      with self.assertRaises(StopIteration):
          next(it)
  
imports: import tensorflow
  from tensorflow_compression.python.datasets import y4m_dataset
label_desc: Use image node_id 2 for calling the Y4MDatasetTest obj's underlying member method code with example usage: obj.test_dataset_yields_correct_sequence() without return types