code: def test_max_values(mock_data):
      with mock_data(num_examples=50):
          ds = tfds.load("mnist", split="train")
          assert ds.element_spec == {
              "image": tf.TensorSpec(shape=(28, 28, 1), dtype=tf.uint8),
              "label": tf.TensorSpec(shape=(), dtype=tf.int64),
          }
          for ex in ds.take(50):
              assert tf.math.reduce_max(ex["label"]).numpy() < 10
          # Test determinism (iterating twice should yield the same samples)
          assert [ex["label"].numpy() for ex in ds.take(5)] == [
              8,
              2,
              9,
              3,
              1,
          ]
          assert [ex["label"].numpy() for ex in ds.take(5)] == [
              8,
              2,
              9,
              3,
              1,
          ]
  
imports: import functools
  import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 10 for calling a global function with example usage: test_max_values(mock_data) without return types