docstr: Test that all public exports have __module__ set correctly.
code: def test_exported_names_match_module(
      self, module_name, include, exclude
  ):
      """Test that all public exports have __module__ set correctly."""
      module = importlib.import_module(module_name)
      self.assertEqual(module.__name__, module_name)
      for name in dir(module):
          if name not in include and (
              name.startswith("_") or name in exclude
          ):
              continue
          obj = getattr(module, name)
          if isinstance(obj, types.ModuleType):
              continue
          self.assertEqual(obj.__module__, module_name)
  
file_path: jax/tests/package_structure_test.py
imports: from collections.abc import Sequence
  import importlib
  import types
  from absl.testing import absltest, parameterized
  from jax._src import test_util
label_desc: Use image node_id 1 for calling the PackageStructureTest obj's underlying member method code with example usage: obj.test_exported_names_match_module(module_name, include, exclude) without return types