code: def testDistribution(self):
      def f(s):
          return tfd.Normal(loc=np.arange(3.0), scale=s).sample()
  
      f(1.0)
      f(np.array([1, 2, 3.0]))
  
imports: import numpy
  from tensorflow_probability.substrates import numpy
  from absl.testing import absltest
label_desc: Use image node_id 2 for calling the NumpyIntegrationTest obj's underlying member method code with example usage: obj.testDistribution() and returns: tfd