File tree Expand file tree Collapse file tree 2 files changed +13
-1
lines changed
Expand file tree Collapse file tree 2 files changed +13
-1
lines changed Original file line number Diff line number Diff line change @@ -166,6 +166,10 @@ def __getitem__(self, name) -> EntryPoint:
166166 except Exception :
167167 raise KeyError (name )
168168
169+ @property
170+ def names (self ):
171+ return set (ep .name for ep in self )
172+
169173
170174class GroupedEntryPoints (tuple ):
171175 """
@@ -175,6 +179,10 @@ class GroupedEntryPoints(tuple):
175179 def __getitem__ (self , group ) -> EntryPoints :
176180 return EntryPoints (ep for ep in self if ep .group == group )
177181
182+ @property
183+ def groups (self ):
184+ return set (ep .group for ep in self )
185+
178186
179187class PackagePath (pathlib .PurePosixPath ):
180188 """A reference to a path in a package"""
Original file line number Diff line number Diff line change @@ -65,7 +65,11 @@ def test_read_text(self):
6565 self .assertEqual (top_level .read_text (), 'mod\n ' )
6666
6767 def test_entry_points (self ):
68- ep = entry_points ()['entries' ]['main' ]
68+ eps = entry_points ()
69+ assert 'entries' in eps .groups
70+ entries = eps ['entries' ]
71+ assert 'main' in entries .names
72+ ep = entries ['main' ]
6973 self .assertEqual (ep .value , 'mod:main' )
7074 self .assertEqual (ep .extras , [])
7175
You can’t perform that action at this time.
0 commit comments