code: def test_document_collection():
      all_docs = list(
          document_datasets.iter_collections_documentation(
              collection_names=["xtreme", "longt5"]
          )
      )
      assert {d.name for d in all_docs} == {"xtreme", "longt5"}
  
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 3 for calling a global function with example usage: test_document_collection() without return types