Added test that should verify if #80 is fixed, will work on fixing it

tests/test_enums.py

@@ -132,6 +132,23 @@ def test_enum_name_as_type(self):
         self.assertEqual(dom.test, 1)
         self.assertEqual(dom.test.enum_name, "BLAH2")
 
+    def test_enum_word_type(self):
+        dom = self._test_parse_build(
+            "\x01",
+            """
+                enum <WORD> tagID {
+                    M_TAG0, // should be 0
+                    M_TAG1 = 0xff01,
+                    M_TAG2,
+                    M_TAG3,
+                };
+            """,
+        )
+        self.assertEqual(dom.M_TAG0.__class__.__name__, "UShort")
+        self.assertEqual(dom.M_TAG1.__class__.__name__, "UShort")
+        self.assertEqual(dom.M_TAG2.__class__.__name__, "UShort")
+        self.assertEqual(dom.M_TAG3.__class__.__name__, "UShort")
+
     def test_enum_with_bitfield(self):
         dom = self._test_parse_build(
             "\x31",