code: def test_benchmark():
      # Works with tf.data.Dataset
      ds = tf.data.Dataset.range(10)
      result = benchmark.benchmark(ds)
      assert isinstance(result, benchmark.BenchmarkResult)
  
      # Works with other iterators
      result = benchmark.benchmark(range(10))
      assert isinstance(result, benchmark.BenchmarkResult)
  
imports: import pytest
  import tensorflow
  from tensorflow_datasets.core.utils import benchmark
label_desc: Use image node_id 2 for calling a global function with example usage: test_benchmark() without return types