code: def _testParamStaticShapes(self, sample_shape, expected):
      param_shapes = johnson_su.JohnsonSU.param_static_shapes(
          sample_shape
      )
      mu_shape, sigma_shape = param_shapes["loc"], param_shapes["scale"]
      self.assertEqual(expected, mu_shape)
      self.assertEqual(expected, sigma_shape)
  
imports: import math
  import numpy
  from scipy import stats
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import johnson_su
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 3 for calling the JohnsonSUTest obj's underlying member method code with example usage: obj._testParamStaticShapes(sample_shape, expected) without return types