code: def test_load_directory(
      karate_gnn: Tuple[
          FeatureStore,
          Dict[Tuple[str, str, str], np.ndarray],
          Dict[str, int],
      ],
      directory: str,
  ):
      if directory == "local":
          local_dir = tempfile.TemporaryDirectory(dir=".")
  
      cugraph_store = CuGraphStore(*karate_gnn)
      cugraph_loader = CuGraphNeighborLoader(
          (cugraph_store, cugraph_store),
          torch.arange(8, dtype=torch.int64),
          2,
          num_neighbors=[8, 4, 2],
          random_state=62,
          replace=False,
          directory=None if directory == "temp" else local_dir.name,
          batches_per_partition=1,
      )
  
      it = iter(cugraph_loader)
      next_batch = next(it)
      assert next_batch is not None
  
      if directory == "local":
          assert len(os.listdir(local_dir.name)) == 4
  
      count = 1
      while next(it, None) is not None:
          count += 1
  
      assert count == 4
  
imports: import pytest
  import tempfile
  import os
  import cudf
  import cupy
  import numpy
  from cugraph_pyg.loader import CuGraphNeighborLoader
  from cugraph_pyg.loader import BulkSampleLoader
  from cugraph_pyg.data import CuGraphStore
  from cugraph_pyg.nn import SAGEConv
  from cugraph.gnn import FeatureStore
  from cugraph.utilities.utils import import_optional, MissingModule
  from typing import Dict, Tuple
label_desc: Use image node_id 9 for calling a global function with example usage: test_load_directory(karate_gnn, directory) without return types