add enumeration checking helpers

diff --git a/tests/util.py b/tests/util.py
index df736df05e69761df7eccd59bbc8e0267a2569d2..644a88a10a25504ea09ad4e9096052d737a722f7 100644 (file)
--- 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)