code: def test_document_datasets():
      all_docs = list(
          document_datasets.iter_documentation_builders(
              # Builder with and without config, as well as config-based builder.
              datasets=["mnist", "coco", "pass"],
              doc_util_paths=doc_utils.DocUtilPaths(
                  fig_base_path=None,
                  df_base_path=None,
                  nightly_path=None,
              ),
          )
      )
      assert {d.name for d in all_docs} == {"mnist", "coco", "pass"}
  
file_path: datasets/tensorflow_datasets/scripts/documentation/document_datasets_test.py
imports: import functools
  from typing import Set
  from unittest import mock
  import pytest
  import tensorflow_datasets
  from tensorflow_datasets.scripts.documentation import doc_utils
  from tensorflow_datasets.scripts.documentation import document_datasets
label_desc: Use image node_id 2 for calling a global function with example usage: test_document_datasets() without return types