Workaround single locations test

Author: johnslavik

packages/griffelib/src/griffe/__init__.py

@@ -593,21 +593,25 @@
     "vtree",
 ]
 
+# Re-export griffecli for backward compatibility.
 try:
-    from griffecli import DEFAULT_LOG_LEVEL, check, dump, get_parser, main  # noqa: F401
+    from griffecli import *
+    from griffecli import __all__ as __cli_all__
 except ImportError:
-    _CLI_MISSING_FEATURES = {
+    # Keep this in sync with the exported members of griffecli.
+    _MISSING_FROM_GRIFFECLI = {
         "DEFAULT_LOG_LEVEL",
         "check",
         "dump",
         "get_parser",
         "main",
     }
-else:
-    _CLI_MISSING_FEATURES = set()
-
 
-def __getattr__(attr: str) -> object:
-    if attr in _CLI_MISSING_FEATURES:
-        raise ImportError(f"Please install `griffecli` to use {'griffe.' + attr!r}")
-    raise AttributeError(attr)
+    def __getattr__(attr: str) -> object:
+        if attr in _MISSING_FROM_GRIFFECLI:
+            raise ImportError(f"Please install `griffecli` to use {'griffe.' + attr!r}")
+        raise AttributeError(attr)
+else:
+    __all__ += __cli_all__
+    # Ignore this namespace in internal API tests.
+    _SINGLE_LOCATIONS_IGNORE = set(__cli_all__)

tests/test_api.py

@@ -143,7 +143,7 @@ def _public_path(obj: griffe.Object | griffe.Alias) -> bool:
 
     public_api = _get_public_api(tested_module)
     multiple_locations = {}
-    for obj_name in tested_module.__all__:
+    for obj_name in set(tested_module.__all__) - getattr(tested_module, "_SINGLE_LOCATIONS_IGNORE", set()):
         obj = public_api[obj_name]
         if obj.aliases and (
             public_aliases := [path for path, alias in obj.aliases.items() if path != obj.path and _public_path(alias)]