code: def setUp(self) -> None:
      self._tensor_distribution = (
          TestDistribution.TestIntKeyDistribution(
              torch.tensor([1.0, 2.0, 3.0, 4.0])
          )
      )
      self._array_distribution = (
          TestDistribution.TestIntKeyDistribution(
              np.array([1.0, 2.0, 3.0, 4.0])
          )
      )
      self._list_distribution = TestDistribution.TestIntKeyDistribution(
          [1.0, 2.0, 3.0, 4.0]
      )
      self._map_distribution = TestDistribution.TestIntKeyDistribution(
          {0: 1.0, 1: 2.0, 2: 3.0, 3: 4.0}
      )
  
imports: import unittest
  from typing import Tuple, Union
  import numpy
  import torch
  from reagent.ope.estimators.types import ActionDistribution, TypeWrapper, Values
label_desc: Use image node_id 1 for calling the TestDistribution obj's underlying member method code with example usage: obj.setUp() without return types