code: def testClassImportIsAdded(self):
      imports, _, _ = create_python_api.get_api_init_text(
          packages=[create_python_api._DEFAULT_PACKAGE],
          packages_to_ignore=[],
          output_package="tensorflow",
          api_name="tensorflow",
          api_version=2,
      )
      if create_python_api._LAZY_LOADING:
          expected_import = (
              "'NewTestClass':"
              " ('tensorflow.python.test_module',"
              " 'TestClass')"
          )
      else:
          expected_import = (
              "from tensorflow.python.test_module " "import TestClass"
          )
      self.assertTrue(
          "TestClass" in str(imports),
          msg="%s not in %s" % (expected_import, str(imports)),
      )
  
file_path: tensorflow/tensorflow/python/tools/api/generator/create_python_api_test.py
imports: import os
  import sys
  import types
  from tensorflow.python.platform import test
  from tensorflow.python.tools.api.generator import create_python_api
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 4 for calling the CreatePythonApiTest obj's underlying member method code with example usage: obj.testClassImportIsAdded() without return types