code: def test_mocking_imagenet():
      ds = tfds.load("imagenet2012", split="train")
      assert ds.element_spec == {
          "file_name": tf.TensorSpec(shape=(), dtype=tf.string),
          "image": tf.TensorSpec(shape=(None, None, 3), dtype=tf.uint8),
          "label": tf.TensorSpec(shape=(), dtype=tf.int64),
      }
      list(ds.take(3))
  
imports: import functools
  import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 3 for calling a global function with example usage: test_mocking_imagenet() without return types