code: def mock_cwd(path: epath.PathLike) -> Iterator[None]:
      """Mock the current directory."""
      path = pathlib.Path(path)
      assert (
          path.exists() and path.is_dir()
      )  # Check given path is valid cwd dir
      with mock.patch("os.getcwd", return_value=os.fspath(path)):
          yield
  
imports: import contextlib
  import os
  import pathlib
  from typing import Dict, Iterator, List, Optional
  from unittest import mock
  from absl.testing import parameterized
  from etils import epath
  import pytest
  import tensorflow_datasets
  from tensorflow_datasets import testing
  from tensorflow_datasets.core import download
  from tensorflow_datasets.core import utils
  from tensorflow_datasets.core.utils import file_utils
  from tensorflow_datasets.scripts.cli import build
  from tensorflow_datasets.scripts.cli import main
label_desc: Use image node_id 2 for calling a global function with example usage: mock_cwd(path) without return types