def METHOD_NAME(self, num_dims: int, ghost_batch_size: int, batch_size: int) -> None:
    module = ModuleWithBatchnorm(num_dims=num_dims)
    ghostbn.apply_ghost_batchnorm(module, ghost_batch_size=ghost_batch_size)
    self._assert_ghost_batches_normalized(module=module, ghost_batch_size=ghost_batch_size, batch_size=batch_size)