class_docstr: Base test class.
class_docstr_tok: ["Base","test","class","."]
code: def assertDistShape(self, dist, event_shape, batch_shape):
      self.assertEqual(dist.event_shape, event_shape)
      self.assertEqual(dist.batch_shape, batch_shape)
  
imports: import numpy
  import tensorflow.compat.v1
  from tensorflow_probability.examples import disentangled_vae
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 2 for calling the _DisentangledVAETest obj's underlying member method code with example usage: obj.assertDistShape(dist, event_shape, batch_shape) without return types