class_docstr: Test MVTec AD Data Module.
class_docstr_tok: ["Test","MVTec","AD","Data","Module","."]
docstr: Test if both single and multiple batch size returns outputs with the right shape.
code: def test_batch_size(self, make_data_module, dataset, batch_size):
      """Test if both single and multiple batch size returns outputs with the right shape."""
      data_module = make_data_module(
          dataset=dataset, batch_size=batch_size
      )
      _, train_data_sample = next(
          enumerate(data_module.train_dataloader())
      )
      _, val_data_sample = next(enumerate(data_module.val_dataloader()))
      assert train_data_sample["image"].shape[0] == batch_size
      assert val_data_sample["image"].shape[0] == batch_size
  
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 1 for calling the TestDataModule obj's underlying member method code with example usage: obj.test_batch_size(make_data_module, dataset, batch_size) without return types