From b70e664ff050b30d2fe40ee041b102fa240827eb Mon Sep 17 00:00:00 2001 From: Bob Halley Date: Tue, 4 Aug 2020 06:05:00 -0700 Subject: [PATCH] add enumeration checking helpers --- tests/util.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/util.py b/tests/util.py index df736df0..644a88a1 100644 --- a/tests/util.py +++ b/tests/util.py @@ -15,7 +15,26 @@ # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. +import enum +import inspect import os.path def here(filename): return os.path.join(os.path.dirname(__file__), filename) + +def enumerate_module(module, super_class): + """Yield module attributes which are subclasses of given class""" + for attr_name in dir(module): + attr = getattr(module, attr_name) + if inspect.isclass(attr) and issubclass(attr, super_class): + yield attr + +def check_enum_exports(module, eq_callback, only=None): + """Make sure module exports all mnemonics from enums""" + for attr in enumerate_module(module, enum.Enum): + if only is not None and attr not in only: + print('SKIP', attr) + continue + for flag, value in attr.__members__.items(): + print(module, flag, value) + eq_callback(getattr(module, flag), value) -- 2.47.3