Coverage for tests/test_extension.py: 100.00%

1"""Test extension.""" 

2 

3import griffe 

4 

5 

6def test_extension() -> None: 

7 """Wildcard imported objects are marked as public.""" 

8 with griffe.temporary_visited_package( 

9 "package", 

10 { 

11 "__init__.py": "from package.module import *", 

12 "module.py": "def f(): ...\nclass C: ...", 

13 }, 

14 resolve_aliases=True, 

15 extensions=griffe.load_extensions("griffe_public_wildcard_imports"), 

16 ) as package: 

17 assert package["f"].public 

18 assert package["f"].is_public 

19 assert package["C"].public 

20 assert package["C"].is_public