docstr: Check that the output is correct for a input.
docstr_tok: ["Check","that","the","output","is","correct","for","a","input","."]
code: def assert_output_shapes_are_correct(self, dict_layer, outputs):
      """Check that the output is correct for a input."""
      self.assertListEqual(
          list(dict_layer.output_keys), list(outputs.keys())
      )
      for v in outputs.values():
          self.assertEqual(v.shape[-1], self.n_out)
  
imports: from absl.testing import parameterized
  from ddsp.training import nn
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 2 for calling the DictLayerTest obj's underlying member method code with example usage: obj.assert_output_shapes_are_correct(dict_layer, outputs) without return types