Fix issue with representing Enum types

yaml · Sep 22, 2021 · 34e764f · 34e764f
1 parent 56644b3
commit 34e764f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/lib/yaml/representer.py b/lib/yaml/representer.py
@@ -369,7 +369,7 @@ def represent_ordered_dict(self, data):
 Representer.add_representer(tuple,
         Representer.represent_tuple)
 
-Representer.add_representer(type,
+Representer.add_multi_representer(type,
         Representer.represent_name)
 
 Representer.add_representer(collections.OrderedDict,

diff --git a/tests/data/construct-python-name-module.code b/tests/data/construct-python-name-module.code
@@ -1 +1 @@
-[str, yaml.Loader, yaml.dump, abs, yaml.tokens]
+[str, yaml.Loader, yaml.dump, abs, yaml.tokens, signal.Handlers]
diff --git a/tests/data/construct-python-name-module.data b/tests/data/construct-python-name-module.data
@@ -3,3 +3,4 @@
 - !!python/name:yaml.dump
 - !!python/name:abs
 - !!python/module:yaml.tokens
+- !!python/name:signal.Handlers
diff --git a/tests/lib/test_constructor.py b/tests/lib/test_constructor.py
@@ -5,6 +5,9 @@
 import datetime
 import yaml.tokens
 
+# Import any packages here that need to be referenced in .code files.
+import signal
+
 def execute(code):
     global value
     exec(code)