code: def test_mocking_partial_decoding():
      ds = tfds.load(
          "mnist",
          split="train",
          decoders=tfds.decode.PartialDecoding(
              {
                  "image": tfds.features.Image(shape=(None, None, 1)),
              }
          ),
      )
      assert ds.element_spec == {
          "image": tf.TensorSpec(shape=(28, 28, 1), dtype=tf.uint8),
      }
      list(ds.take(3))
  
imports: import functools
  import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 5 for calling a global function with example usage: test_mocking_partial_decoding() without return types