class_docstr: Test MVTec AD Data Module.
class_docstr_tok: ["Test","MVTec","AD","Data","Module","."]
docstr: Test if the batch returned by __getitem__ contains the required keys.
code: def test_keys_in_batch_dict(self, make_data_module, dataset, task):
      """Test if the batch returned by __getitem__ contains the required keys."""
      data_module = make_data_module(dataset=dataset, task=task)
      _, val_data = next(enumerate(data_module.val_dataloader()))
      _, test_data = next(enumerate(data_module.test_dataloader()))
  
      assert set(EXPECTED_KEYS_PER_TASK_TYPE[task]).issubset(
          set(val_data.keys())
      )
      assert set(EXPECTED_KEYS_PER_TASK_TYPE[task]).issubset(
          set(test_data.keys())
      )
  
imports: import os
  import numpy
  import pytest
  from anomalib.config import update_input_size_config
  from anomalib.data import Avenue, BTech, Folder, MVTec, ShanghaiTech, UCSDped, Visa, get_datamodule
  from anomalib.data.utils import DirType
  from anomalib.pre_processing.transforms import Denormalize, ToNumpy
  from tests.helpers.config import get_test_configurable_parameters
  from tests.helpers.dataset import TestDataset, get_dataset_path
label_desc: Use image node_id 2 for calling the TestDataModule obj's underlying member method code with example usage: obj.test_keys_in_batch_dict(make_data_module, dataset, task) without return types