code: def test_generate(self):
      # Use parquet_files_count to speed up the test
      imagenet_directory_to_petastorm_dataset(
          TestGenerate.mock_imagenet_dir,
          "file://" + TestGenerate.mock_output_dir,
          spark_master="local[3]",
          parquet_files_count=3,
          noun_id_to_text=TestGenerate.noun_id_to_text,
      )
  
file_path: petastorm/examples/imagenet/tests/test_generate_imagenet_dataset.py
imports: import os
  import shutil
  import tempfile
  import unittest
  import cv2
  import numpy
  from examples.imagenet.generate_petastorm_imagenet import download_nouns_mapping, imagenet_directory_to_petastorm_dataset
label_desc: Use image node_id 4 for calling the TestGenerate obj's underlying member method code with example usage: obj.test_generate() without return types