commit d0e49f01ca40154348d2deaf229541e2f9ee2988
author Kevin Wojniak <kainjow@users.noreply.github.com> Sat Jan 22 21:04:30 2022 -0800
committer Jens Geyer <Jens-G@users.noreply.github.com> Sat Mar 05 09:58:27 2022 +0100
tree 50db29e967683ca5c30c1c5d6f3039655b98e82a
parent 8d6694d60787e2c5e16378e3738a4b31382b6834

add tests

test/py/SerializationTest.py

diff --git a/test/py/SerializationTest.py b/test/py/SerializationTest.py
index ef79835..f47c3d4 100755
--- a/test/py/SerializationTest.py
+++ b/test/py/SerializationTest.py
@@ -432,7 +432,19 @@
         self.assertEquals(obj, objcopy)
 
         # test enums
-        for num, name in Numberz._VALUES_TO_NAMES.items():
+        def _enumerate_enum(enum_class):
+            if hasattr(enum_class, '_VALUES_TO_NAMES'):
+                # old-style enums
+                for num, name in enum_class._VALUES_TO_NAMES.items():
+                    yield (num, name)
+            else:
+                # assume Python 3.4+ IntEnum-based
+                from enum import IntEnum
+                self.assertTrue((issubclass(enum_class, IntEnum)))
+                for num in enum_class:
+                    yield (num.value, num.name)
+
+        for num, name in _enumerate_enum(Numberz):
             obj = Bonk(message='enum Numberz value %d is string %s' % (num, name), type=num)
             objcopy = Bonk()
             deserialize(objcopy, serialize(obj))