code: def test_io_dataset_basic_operation(
      fixture_lookup, io_dataset_fixture
  ):
      """test_io_dataset_basic_operation"""
      args, func, expected = fixture_lookup(io_dataset_fixture)
  
      dataset = func(args)
  
      # Test of take
      expected_taken = expected[:5]
      entries_taken = list(dataset.take(5))
  
      assert len(entries_taken) == len(expected_taken)
      assert all(
          [
              element_equal(a, b)
              for (a, b) in zip(entries_taken, expected_taken)
          ]
      )
  
      # Test of batch
      indices = list(range(0, len(expected), 3))
      indices = list(zip(indices, indices[1:] + [len(expected)]))
  
      expected_batched = [
          element_slice(expected, i, j) for i, j in indices
      ]
      entries_batched = list(dataset.batch(3))
  
      assert len(entries_batched) == len(expected_batched)
      assert all(
          [
              all([element_equal(i, j) for (i, j) in zip(a, b)])
              for (a, b) in zip(entries_batched, expected_batched)
          ]
      )
  
imports: import os
  import sys
  import time
  import shutil
  import tempfile
  import numpy
  import pytest
  import tensorflow
  import tensorflow_io
label_desc: Use image node_id 35 for calling a global function with example usage: test_io_dataset_basic_operation(fixture_lookup, io_dataset_fixture) without return types