code: def test_calculate_storage_specific_sizes(self) -> None:
      for inputs in calculate_storage_specific_size_data_provider():
          (
              sharding_type,
              optimizer_class,
              expected_storage,
          ) = inputs.values()
          estimates = _calculate_storage_specific_sizes(
              storage=100,
              shape=torch.Size((10, 5, 3)),
              shard_sizes=[[5, 5, 3], [5, 5, 3]],
              sharding_type=sharding_type.value,
              optimizer_class=optimizer_class,
          )
  
          self.assertEqual(estimates, expected_storage)
  
imports: import unittest
  from typing import cast
  import torch
  import torchrec.optim
  from torchrec.distributed.embedding import EmbeddingCollectionSharder
  from torchrec.distributed.embedding_types import EmbeddingComputeKernel
  from torchrec.distributed.embeddingbag import EmbeddingBagCollectionSharder
  from torchrec.distributed.fbgemm_qcomm_codec import CommType, get_qcomm_codecs_registry, QCommsConfig
  from torchrec.distributed.planner.constants import BATCH_SIZE
  from torchrec.distributed.planner.enumerators import EmbeddingEnumerator
  from torchrec.distributed.planner.shard_estimators import _calculate_storage_specific_sizes, EmbeddingOffloadStats, EmbeddingPerfEstimator
  from torchrec.distributed.planner.types import ParameterConstraints, Perf, Topology
  from torchrec.distributed.quant_embeddingbag import QuantEmbeddingBagCollectionSharder
  from torchrec.distributed.test_utils.test_model import TestSparseNN
  from torchrec.distributed.tests.test_quant_model_parallel import _quantize
  from torchrec.distributed.tests.test_sequence_model import TestSequenceSparseNN
  from torchrec.distributed.types import CacheParams, CacheStatistics, ModuleSharder, ShardingType
  from torchrec.modules.embedding_configs import EmbeddingBagConfig, EmbeddingConfig
label_desc: Use image node_id 1 for calling the TestEmbeddingStorageEstimator obj's underlying member method code with example usage: obj.test_calculate_storage_specific_sizes() without return types