code: def test_get_module_docstring(self):
      docs_by_module = {
          "io": "io docs",
      }
      self.assertEqual(
          "io docs",
          generator._get_module_docstring(docs_by_module, "io"),
      )
      self.assertEqual(
          "Public API for math namespace",
          generator._get_module_docstring(docs_by_module, "math"),
      )
  
file_path: tensorflow/tensorflow/python/tools/api/generator2/generator/generator_test.py
imports: import collections
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  from tensorflow.python.tools.api.generator2.generator import generator
  from tensorflow.python.tools.api.generator2.shared import exported_api
label_desc: Use image node_id 6 for calling the GeneratorTest obj's underlying member method code with example usage: obj.test_get_module_docstring() without return types