1 files changed, 82 insertions, 0 deletions
diff --git a/_test/test_yamlobject.py b/_test/test_yamlobject.py
new file mode 100644
index 0000000..0f9c48c
--- /dev/null
+++ b/_test/test_yamlobject.py
@@ -0,0 +1,82 @@
+# coding: utf-8
+
+import pytest  # NOQA
+
+from .roundtrip import save_and_run  # NOQA
+
+
+def test_monster(tmpdir):
+    program_src = '''\
+    import ruyaml
+    from textwrap import dedent
+
+    class Monster(ruyaml.YAMLObject):
+        yaml_tag = '!Monster'
+
+        def __init__(self, name, hp, ac, attacks):
+            self.name = name
+            self.hp = hp
+            self.ac = ac
+            self.attacks = attacks
+
+        def __repr__(self):
+            return "%s(name=%r, hp=%r, ac=%r, attacks=%r)" % (
+                self.__class__.__name__, self.name, self.hp, self.ac, self.attacks)
+
+    data = ruyaml.load(dedent("""\\
+        --- !Monster
+        name: Cave spider
+        hp: [2,6]    # 2d6
+        ac: 16
+        attacks: [BITE, HURT]
+    """), Loader=ruyaml.Loader)
+    # normal dump, keys will be sorted
+    assert ruyaml.dump(data) == dedent("""\\
+        !Monster
+        ac: 16
+        attacks: [BITE, HURT]
+        hp: [2, 6]
+        name: Cave spider
+    """)
+    '''
+    assert save_and_run(program_src, tmpdir) == 0
+
+
+def test_qualified_name00(tmpdir):
+    """issue 214"""
+    program_src = """\
+    import ruyaml
+    from io import StringIO
+
+    class A:
+        def f(self):
+            pass
+
+    yaml = ruyaml.YAML(typ='unsafe', pure=True)
+    yaml.explicit_end = True
+    buf = StringIO()
+    yaml.dump(A.f, buf)
+    res = buf.getvalue()
+    print('res', repr(res))
+    assert res == "!!python/name:__main__.A.f ''\\n...\\n"
+    x = ruyaml.load(res)
+    assert x == A.f
+    """
+    assert save_and_run(program_src, tmpdir) == 0
+
+
+def test_qualified_name01(tmpdir):
+    """issue 214"""
+    from io import StringIO
+
+    import ruyaml.comments
+    from ruyaml import YAML
+
+    yaml = YAML(typ='unsafe', pure=True)
+    yaml.explicit_end = True
+    buf = StringIO()
+    yaml.dump(ruyaml.comments.CommentedBase.yaml_anchor, buf)
+    res = buf.getvalue()
+    assert res == "!!python/name:ruyaml.comments.CommentedBase.yaml_anchor ''\n...\n"
+    x = yaml.load(res)
+    assert x == ruyaml.comments.CommentedBase.yaml_anchor