code: def div2k_datafolder(tmp_path: Path):
      rng = numpy.random.default_rng(0xFEEEFEEE)
  
      directory = tmp_path.joinpath("div2k").joinpath("DIV2K_valid_HR")
  
      directory.mkdir(parents=True)
  
      n = rng.integers(1, 16)
  
      images = []
  
      for index in range(n):
          path = directory.joinpath(f"{index:03}.png")
  
          images.append(create_random_image((3, 224, 224), rng))
          write_image_to_file(images[-1], path)
  
      return directory.parent, images
  
file_path: NeuralCompression/tests/data/test_div2k.py
imports: from pathlib import Path
  import numpy
  import pytest
  import torch
  from torchvision.transforms import ToTensor
  from utils import create_random_image, write_image_to_file
  from neuralcompression.data import DIV2KDataset
label_desc: Use image node_id 1 for calling a global function with example usage: div2k_datafolder(tmp_path) and returns: directory,  images