code: def test_raw_benchmark(
      ds, num_iter, batch_size, expected_num_examples
  ):
      result = benchmark.raw_benchmark(
          ds, num_iter=num_iter, batch_size=batch_size
      )
      assert isinstance(result, benchmark.RawBenchmarkResult)
      assert result.num_examples == expected_num_examples
      assert result.end_time > result.start_time
      assert result.end_time >= result.first_batch_time
  
imports: import pytest
  import tensorflow
  from tensorflow_datasets.core.utils import benchmark
label_desc: Use image node_id 1 for calling a global function with example usage: test_raw_benchmark(ds, num_iter, batch_size, expected_num_examples) without return types