code: def test_video_io_dataset(video_data, io_dataset_func):
      """test_video_io_dataset"""
      video_path, video_value = video_data
  
      video_dataset = io_dataset_func(video_path)
  
      i = 0
      for value in video_dataset:
          assert video_value[i].shape == value.shape
          i += 1
      assert i == 166
  
      video_dataset = io_dataset_func(video_path).batch(2)
  
      i = 0
      for value in video_dataset:
          assert video_value[i : i + 2].shape == value.shape
          i += 2
      assert i == 166
  
file_path: io/tests/test_video.py
imports: import os
  import sys
  import pytest
  import tensorflow
  import tensorflow_io
label_desc: Use image node_id 2 for calling a global function with example usage: test_video_io_dataset(video_data, io_dataset_func) without return types